osSocket.h 2.6 KB
Newer Older
S
Shengliang Guan 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
/*
 * Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
 *
 * This program is free software: you can use, redistribute, and/or modify
 * it under the terms of the GNU Affero General Public License, version 3
 * or later ("AGPL"), as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */

#ifndef TDENGINE_OS_SOCKET_H
#define TDENGINE_OS_SOCKET_H

#ifdef __cplusplus
extern "C" {
#endif

S
slguan 已提交
23 24 25 26 27 28 29 30 31 32 33 34
#ifndef TAOS_OS_FUNC_SOCKET_OP
  #define taosSend(sockfd, buf, len, flags) send(sockfd, buf, len, flags)
  #define taosSendto(sockfd, buf, len, flags, dest_addr, addrlen) sendto(sockfd, buf, len, flags, dest_addr, addrlen)
  #define taosReadSocket(fd, buf, len) read(fd, buf, len)
  #define taosWriteSocket(fd, buf, len) write(fd, buf, len)
  #define taosCloseSocket(x) \
    {                        \
      if (FD_VALID(x)) {     \
        close(x);            \
        x = FD_INITIALIZER;  \
      }                      \
    }
S
TD-1057  
Shengliang Guan 已提交
35
  typedef int SOCKET;
S
TD-1057  
Shengliang Guan 已提交
36 37 38 39 40
#endif

#ifndef TAOS_OS_DEF_EPOLL
  #define TAOS_EPOLL_WAIT_TIME -1
#endif  
S
Shengliang Guan 已提交
41 42 43 44

#define taosClose(x) taosCloseSocket(x)

#ifdef TAOS_RANDOM_NETWORK_FAIL
S
Shengliang Guan 已提交
45 46 47 48 49 50 51 52 53 54 55 56 57 58
  #ifdef TAOS_RANDOM_NETWORK_FAIL_TEST
    ssize_t taosSendRandomFail(int sockfd, const void *buf, size_t len, int flags);
    ssize_t taosSendToRandomFail(int sockfd, const void *buf, size_t len, int flags, const struct sockaddr *dest_addr, socklen_t addrlen);
    ssize_t taosReadSocketRandomFail(int fd, void *buf, size_t count);
    ssize_t taosWriteSocketRandomFail(int fd, const void *buf, size_t count);
    #undef taosSend
    #undef taosSendto
    #undef taosReadSocket
    #undef taosWriteSocket
    #define taosSend(sockfd, buf, len, flags) taosSendRandomFail(sockfd, buf, len, flags)
    #define taosSendto(sockfd, buf, len, flags, dest_addr, addrlen) taosSendToRandomFail(sockfd, buf, len, flags, dest_addr, addrlen)
    #define taosReadSocket(fd, buf, len) taosReadSocketRandomFail(fd, buf, len)
    #define taosWriteSocket(fd, buf, len) taosWriteSocketRandomFail(fd, buf, len)
  #endif  
S
Shengliang Guan 已提交
59 60 61
#endif

// TAOS_OS_FUNC_SOCKET
S
TD-1057  
Shengliang Guan 已提交
62
int taosSetNonblocking(SOCKET sock, int on);
S
Shengliang Guan 已提交
63 64 65
void taosBlockSIGPIPE();

// TAOS_OS_FUNC_SOCKET_SETSOCKETOPT
S
TD-1057  
Shengliang Guan 已提交
66
int taosSetSockOpt(SOCKET socketfd, int level, int optname, void *optval, int optlen);
S
Shengliang Guan 已提交
67

68 69 70 71
// TAOS_OS_FUNC_SOCKET_INET
uint32_t taosInetAddr(char *ipAddr);
const char *taosInetNtoa(IN_ADDR ipInt);

S
Shengliang Guan 已提交
72 73 74 75 76
#ifdef __cplusplus
}
#endif

#endif