Socket客户端例子[C语言]

Socket客户端例子[C语言]

#include    < sys / stat.h >    
#include    < sys / types.h >    
#include    < sys / socket.h >    
#include    < stdio.h >    
#include    < malloc.h >    
#include    < netdb.h >    
#include    < fcntl.h >
#include    < unistd.h >
#include    < netinet / in .h >
#include    < arpa / inet.h >

#define     RES_LENGTH  10240  // 接受字符的最大长度

int      connect_socket( char   *  server, int  serverPort);
int      send_msg( int  sockfd, char   *  sendBuff);
char   *   recv_msg( int  sockfd);
int      close_socket( int  sockfd);

int  main( int  argc, char   **  argv)
{
     int     sockfd = 0 ;
     char      sendMsg[ 30 ] = " abc.org\r\n\r " ;
     char      *  res;
    sockfd = connect_socket( " whois.pir.org " , 43 );
    send_msg(sockfd,sendMsg);
    res = recv_msg(sockfd);
    printf(res);
    free(res);
    close_socket(sockfd);
     return   0 ;
}
/* ***********************************************************
 * 连接SOCKET服务器，如果出错返回-1，否则返回socket处理代码
 * server：服务器地址(域名或者IP),serverport：端口
 * ******************************************************* */
int     connect_socket( char   *  server, int  serverPort){
     int     sockfd = 0 ;
     struct     sockaddr_in    addr;
     struct     hostent         *  phost;
     // 向系统注册，通知系统建立一个通信端口
     // AF_INET表示使用IPv4协议
     // SOCK_STREAM表示使用TCP协议
     if ((sockfd = socket(AF_INET,SOCK_STREAM, 0 )) < 0 ){
        herror( " Init socket error! " );
         return   - 1 ;
    }

    bzero( & addr, sizeof (addr));
    addr.sin_family = AF_INET;
    addr.sin_port = htons(serverPort);
    addr.sin_addr.s_addr = inet_addr(server); // 按IP初始化
     if (addr.sin_addr.s_addr  ==  INADDR_NONE){ // 如果输入的是域名
        phost  =  ( struct  hostent * )gethostbyname(server);
         if (phost == NULL){
            herror( " Init socket s_addr error! " );
             return   - 1 ;
        }
        addr.sin_addr.s_addr  = (( struct  in_addr * )phost -> h_addr) -> s_addr;
    }
     if (connect(sockfd,( struct  sockaddr * ) & addr, sizeof (addr)) < 0 )
         return   - 1 ; // 0表示成功，-1表示失败
     else
         return  sockfd;

}

/* *************************************************************
 * 发送消息，如果出错返回-1，否则返回发送的字符长度
 * sockfd：socket标识，sendBuff：发送的字符串
 * ******************************************************** */
int     send_msg( int  sockfd, char   *  sendBuff){
     int     sendSize = 0 ;
     if ((sendSize = send(sockfd,sendBuff,strlen(sendBuff), 0 )) <= 0 ){
        herror( " Send msg error! " );
         return   - 1 ;
    } else
         return  sendSize;
}

/* ***************************************************************
 *接受消息，如果出错返回NULL，否则返回接受字符串的指针(动态分配，注意释放)
 *sockfd：socket标识
 * ******************************************************** */
char   *     recv_msg( int  sockfd){
     char   *  response;
     int   flag = 0 ,recLenth = 0 ;
    response = ( char   * )malloc(RES_LENGTH);
    memset(response, 0 ,RES_LENGTH);
     for (flag = 0 ;;){
         if ((recLenth = recv(sockfd,response + flag,RES_LENGTH - flag, 0 )) ==- 1 ){
            free(response);
            herror( " Recv msg error! " );
             return  NULL;
        } else   if (recLenth == 0 )
             break ;
         else {
            flag += recLenth;
            recLenth = 0 ;
        }
    }
    response[flag] = ' \0 ' ;
     return  response;
}
/* *************************************************
 *关闭连接
 * ********************************************* */
int     close_socket( int  sockfd){
    close(sockfd);
     return   0 ;
}