src/network/TCPClient.cxx

   1 /***************************************************************************
   2  *   Copyright (C) 2007 by Prabakaran Thirumalai   *
   3  *   praba_tuty@yahoo.com   *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
  19  ***************************************************************************/
  20 #include <os.h>
  21 #include <CSql.h>
  22 #include <Network.h>
  23 #include <Parser.h>
  24 #if defined(SOLARIS)
  25 #define MSG_NOSIGNAL 0
  26 #endif
  27 TCPClient::TCPClient()
  28 {
  29     isConnectedFlag =false;
  30     cacheClient = false;
  31     sockfd = -1;
  32     respPkt = new ResponsePacket();
  33     pktHdr = new PacketHeader();
  34 }
  35
  36 TCPClient::~TCPClient()
  37 {
  38    if (isConnectedFlag) disconnect();
  39    delete respPkt;
  40    delete pktHdr;
  41 }
  42 DbRetVal TCPClient::send(NetworkPacketType type)
  43 {
  44     DbRetVal rv = OK;
  45     if (sockfd == -1) {
  46         printError(ErrNoConnection, "Connection lost with the peer.");
  47         return ErrNoConnection;
  48     }
  49     pktHdr->packetType = type;
  50     pktHdr->packetLength = 0;
  51     int numbytes=0;
  52     if ((numbytes=os::send(sockfd, pktHdr, sizeof(PacketHeader), MSG_NOSIGNAL)) == -1) {
  53         if (errno == EPIPE) {
  54             printError(ErrNoConnection, "connection not present");
  55             isConnectedFlag = false;
  56             return ErrNoConnection;
  57         }
  58         printError(ErrOS, "Unable to send the packet");
  59         os::closeSocket(sockfd);
  60         sockfd = -1;
  61         isConnectedFlag = false;
  62         return ErrNoConnection;
  63     }
  64     return rv;
  65 }
  66
  67 DbRetVal TCPClient::send(NetworkPacketType type, int stmtid)
  68 {
  69     DbRetVal rv = OK;
  70     if (sockfd == -1) {
  71         printError(ErrNoConnection, "Connection lost with the peer.");
  72         return ErrNoConnection;
  73     }
  74     pktHdr->packetType = type;
  75     pktHdr->packetLength = 0;
  76     pktHdr->stmtID = stmtid;
  77     int numbytes=0;
  78     if ((numbytes=os::send(sockfd, pktHdr, sizeof(PacketHeader), MSG_NOSIGNAL)) == -1) {
  79         if (errno == EPIPE) {
  80             printError(ErrNoConnection, "connection not present");
  81             isConnectedFlag = false;
  82                         os::closeSocket(sockfd);
  83             return ErrNoConnection;
  84         }
  85         printError(ErrOS, "Unable to send the packet");
  86         os::closeSocket(sockfd);
  87         sockfd = -1;
  88         isConnectedFlag = false;
  89         return ErrNoConnection;
  90     }
  91     return rv;
  92 }
  93 DbRetVal TCPClient::send(NetworkPacketType type, char *buf, int len)
  94 {
  95     DbRetVal rv = OK;
  96     if (sockfd == -1) {
  97         printError(ErrNoConnection, "Connection lost with the peer.");
  98         return ErrNoConnection;
  99     }
 100 //    printf("NW:TCP Send\n");
 101     //void* totalBuffer = malloc(sizeof(PacketHeader)+ len);
 102     //PacketHeader *hdr=  new PacketHeader();
 103     pktHdr->packetType = type;
 104     pktHdr->packetLength = len;
 105     pktHdr->srcNetworkID = networkid;
 106     pktHdr->version = 1;
 107     pktHdr->stmtID = 0;
 108     //memcpy(((char*)totalBuffer) + sizeof(PacketHeader) , buf, len);
 109     int numbytes=0;
 110     if ((numbytes=os::send(sockfd, pktHdr, sizeof(PacketHeader), MSG_NOSIGNAL)) == -1) {
 111         if (errno == EPIPE) {
 112             printError(ErrNoConnection, "connection not present");
 113             isConnectedFlag = false;
 114                         os::closeSocket(sockfd);
 115             return ErrNoConnection;
 116         }
 117         printError(ErrOS, "Unable to send the packet");
 118         os::closeSocket(sockfd);
 119         sockfd = -1;
 120         isConnectedFlag = false;
 121         return ErrNoConnection;
 122     }
 123 //    printf("Sent bytes %d\n", numbytes);
 124     if ((numbytes=os::send(sockfd, buf, len, MSG_NOSIGNAL)) == -1) {
 125         if (errno == EPIPE) {
 126             printError(ErrNoConnection, "connection not present");
 127             isConnectedFlag = false;
 128                         os::closeSocket(sockfd);
 129             return ErrNoConnection;
 130         }
 131         printError(ErrOS, "Unable to send the packet");
 132         os::closeSocket(sockfd);
 133         sockfd = -1;
 134         isConnectedFlag = false;
 135         return ErrNoConnection;
 136     }
 137 //    printf("Sent bytes %d\n", numbytes);
 138     //free(totalBuffer);
 139     //delete hdr;
 140     return rv;
 141 }
 142 DbRetVal TCPClient::receive()
 143 {
 144     DbRetVal rv = OK;
 145     printDebug(DM_Network, "NW:TCP receive\n");
 146     fd_set fdset;
 147     FD_ZERO(&fdset);
 148     FD_SET(sockfd, &fdset);
 149     struct timeval timeout;
 150     timeout.tv_sec = Conf::config.getNetworkResponseTimeout();
 151     timeout.tv_usec = 0;
 152     int ret = os::select(sockfd+1, &fdset, 0, 0, &timeout);
 153     if (ret <= 0) {
 154         printError(ErrPeerTimeOut,"Response timeout for peer site");
 155         os::closeSocket(sockfd);
 156         sockfd = -1;
 157         isConnectedFlag = false;
 158         return ErrPeerTimeOut;
 159     }
 160     socklen_t len = sizeof(struct sockaddr);
 161         printDebug(DM_Network, "Sizeof response packet %d\n",  sizeof(ResponsePacket));
 162     int numbytes = os::recv(sockfd, respPkt, sizeof(ResponsePacket), 0);
 163     printDebug(DM_Network, " NUMBYTES packet %d\n",  numbytes);
 164     if (numbytes == -1)
 165     {
 166        printError(ErrOS, "Unable to receive response from peer");
 167         os::closeSocket(sockfd);
 168         sockfd = -1;
 169         isConnectedFlag = false;
 170         return ErrNoConnection;
 171     }
 172     return respPkt->errRetVal;
 173 }
 174
 175 DbRetVal TCPClient::connect()
 176 {
 177     printDebug(DM_Network, "NW:TCP connect %s %d %d\n", hostName, port, networkid);
 178     //TODO::send endian to the peer site
 179     //do not do endian conversion here. it will be done at the server side
 180 #ifdef WINNT
 181         int iResult=0;
 182         iResult = WSAStartup(MAKEWORD(2, 2), &wsaData);
 183     if (iResult != 0) {
 184         printf("WSAStartup failed: %d\n", iResult);
 185         return ErrSysInit;
 186     }
 187 #endif
 188         isConnectedFlag = false;
 189     struct hostent *he;
 190     int numbytes;
 191     if ((he=gethostbyname(hostName)) == NULL) { // get the host info
 192         printError(ErrOS,"Unable to get the peer host name");
 193         return ErrOS;
 194     }
 195     if ((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1) {
 196         printError(ErrOS,"Unable to create socket to peer host name");
 197         return ErrOS;
 198     }
 199     srvAddr.sin_family = AF_INET; // host byte order
 200     srvAddr.sin_port = htons(port); // short, network byte order
 201     srvAddr.sin_addr = *((struct in_addr *)he->h_addr);
 202     memset(&(srvAddr.sin_zero), '\0', 8); // zero the rest of the struct
 203     if (::connect(sockfd, (struct sockaddr*) & srvAddr, sizeof(struct sockaddr))
 204          == -1)
 205     {
 206         printError(ErrOS, "Unable to connect to peer site");
 207         return ErrOS;
 208     }
 209 //    printf("NW:TCP connecting\n");
 210     fd_set fdset;
 211     FD_ZERO(&fdset);
 212     FD_SET(sockfd, &fdset);
 213     struct timeval timeout;
 214     timeout.tv_sec = Conf::config.getNetworkConnectTimeout();
 215     timeout.tv_usec = 0;
 216     int ret = os::select(sockfd+1, &fdset, 0, 0, &timeout);
 217     if (ret <= 0) {
 218         printError(ErrPeerTimeOut,"Response timeout for peer site");
 219         return ErrPeerTimeOut;
 220     }
 221     int response=0;
 222     socklen_t len = sizeof(struct sockaddr);
 223     numbytes = os::recv(sockfd, &response, 4, 0);
 224     if (numbytes !=4)
 225     {
 226        printError(ErrOS, "Unable to receive response from peer");
 227        return ErrOS;
 228     }
 229     //    if (response != 1) return ErrPeerResponse;
 230
 231     //packet header information
 232     pktHdr->srcNetworkID = networkid;
 233     pktHdr->version = 1;
 234     isConnectedFlag = true;
 235     return OK;
 236 }
 237
 238 DbRetVal TCPClient::disconnect()
 239 {
 240     if (isConnectedFlag) {
 241         pktHdr->packetType = SQL_NW_PKT_DISCONNECT;
 242         pktHdr->packetLength = 0;
 243         int numbytes=0;
 244         if ((numbytes=os::send(sockfd, pktHdr, sizeof(PacketHeader), MSG_NOSIGNAL)) == -1) {
 245             if (errno == EPIPE) {
 246                 printError(ErrNoConnection, "connection not present");
 247                                 os::closeSocket(sockfd);
 248 #ifdef WINNT
 249                 WSACleanup();
 250 #endif
 251                 isConnectedFlag = false;
 252                 return ErrNoConnection;
 253             }
 254             printError(ErrOS, "Unable to send the packet");
 255             os::closeSocket(sockfd);
 256 #ifdef WINNT
 257             WSACleanup();
 258 #endif
 259             sockfd = -1;
 260             isConnectedFlag = false;
 261             return ErrNoConnection;
 262         } else {
 263             printDebug(DM_Network,"Sent bytes %d\n", numbytes);
 264             DbRetVal rv = receive();
 265             isConnectedFlag = false;
 266             os::closeSocket(sockfd);
 267 #ifdef WINNT
 268             WSACleanup();
 269 #endif
 270             sockfd = -1;
 271         }
 272     }
 273     return OK;
 274 }