123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129 |
- /* vim:expandtab:shiftwidth=2:tabstop=2:smarttab:
- *
- * Data Differential's libhostle
- *
- * Copyright (C) 2012 Data Differential, http://datadifferential.com/
- *
- * Redistribution and use in source and binary forms, with or without
- * modification, are permitted provided that the following conditions are
- * met:
- *
- * * Redistributions of source code must retain the above copyright
- * notice, this list of conditions and the following disclaimer.
- *
- * * Redistributions in binary form must reproduce the above
- * copyright notice, this list of conditions and the following disclaimer
- * in the documentation and/or other materials provided with the
- * distribution.
- *
- * * The names of its contributors may not be used to endorse or
- * promote products derived from this software without specific prior
- * written permission.
- *
- * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
- * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
- * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
- * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
- * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
- * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
- * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
- * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
- * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
- * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
- * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
- *
- */
- #include "gear_config.h"
- #include <errno.h>
- #include <pthread.h>
- #include <stdio.h>
- #include <stdlib.h>
- #include <sys/socket.h>
- #include <sys/types.h>
- #include <time.h>
- #include <unistd.h>
- #include <libhostile/initialize.h>
- #include <libhostile/function.h>
- static int not_until= 500;
- static struct function_st __function;
- static pthread_once_t function_lookup_once = PTHREAD_ONCE_INIT;
- static void set_local(void)
- {
- __function= set_function("connect", "HOSTILE_CONNECT");
- }
- void set_connect_close(bool arg, int frequency, int not_until_arg)
- {
- if (arg)
- {
- __function.frequency= frequency;
- not_until= not_until_arg;
- }
- else
- {
- __function.frequency= 0;
- not_until= 0;
- }
- }
- int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen)
- {
- hostile_initialize();
- (void) pthread_once(&function_lookup_once, set_local);
- if (is_called() == false)
- {
- if (__function.frequency)
- {
- if (--not_until < 0 && rand() % __function.frequency)
- {
- int current_errno= 0;
- switch (rand() % 5)
- {
- case 0:
- errno= EINPROGRESS;
- return -1;
- case 1:
- errno= EINTR;
- return -1;
- case 2:
- current_errno= ECONNABORTED;
- break;
- case 3:
- current_errno= EMFILE;
- break;
- case 4:
- current_errno= ECONNRESET;
- break;
- default:
- case 5:
- current_errno= ETIMEDOUT;
- break;
- }
- shutdown(sockfd, SHUT_RDWR);
- close(sockfd);
- errno= current_errno;
- return -1;
- }
- }
- }
- set_called();
- int ret= __function.function.connect(sockfd, addr, addrlen);
- reset_called();
- return ret;
- }
|