TCP的socket连接示例

转载 2006年06月23日 13:13:00

/*
 * on Unix:
 *    cc -c connector.c
 *    cc -o connector connector.o
 *
 * on Windows NT:
 *    open connector.c in Visual Studio
 *    press 'F7' to link -- a project to be created
 *    add wsock32.lib to the link section under project setting
 *    press 'F7' again
 *
 * running:
 *    type 'connector' for usage
 */


#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <stdarg.h>
#include <errno.h>
#include <fcntl.h>
#ifdef WIN32
#include <winsock2.h>
#else
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <sys/ioctl.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <netdb.h>
#endif

#ifndef INADDR_NONE
#define INADDR_NONE     0xffffffff
#endif
#define MAX_STRING_LEN  1024
#define BUFSIZE  2048

#ifndef WIN32
#define SOCKET int
#else
#define errno WSAGetLastError()
#define close(a) closesocket(a)
#define write(a, b, c) send(a, b, c, 0)
#define read(a, b, c) recv(a, b, c, 0)
#endif

char buf[BUFSIZE];

static char i_host[MAX_STRING_LEN];  /* site name */
static char i_port[MAX_STRING_LEN];  /* port number */

void err_doit(int errnoflag, const char *fmt, va_list ap);
void err_quit(const char *fmt, ...);
int tcp_connect(const char *host, const unsigned short port);
void print_usage();

//xnet_select x defines
#define READ_STATUS  0
#define WRITE_STATUS 1
#define EXCPT_STATUS 2

/*
s    - SOCKET
sec  - timeout seconds
usec - timeout microseconds
x    - select status
*/

SOCKET xnet_select(SOCKET s, int sec, int usec, short x)
{
 int st = errno;
 struct timeval to;
 fd_set fs;
 to.tv_sec = sec;
 to.tv_usec = usec;
 FD_ZERO(&fs);
 FD_SET(s, &fs);
 switch(x){
  case READ_STATUS:
  st = select(s+1, &fs, 0, 0, &to);
  break;
  case WRITE_STATUS:
  st = select(s+1, 0, &fs, 0, &to);
  break;
  case EXCPT_STATUS:
  st = select(s+1, 0, 0, &fs, &to);
  break;
 }
 return(st);
}

int tcp_connect(const char *host, const unsigned short port)
{
    unsigned long non_blocking = 1;
    unsigned long blocking = 0;
    int ret = 0;
    char * transport = "tcp";
    struct hostent      *phe;   /* pointer to host information entry    */
    struct protoent *ppe;       /* pointer to protocol information entry*/
    struct sockaddr_in sin;     /* an Internet endpoint address  */
    SOCKET s;                    /* socket descriptor and socket type    */
    int error;

#ifdef WIN32
    {
        WORD wVersionRequested;
        WSADATA wsaData;
        int err;
 
        wVersionRequested = MAKEWORD( 2, 0 );
 
        err = WSAStartup( wVersionRequested, &wsaData );
        if ( err != 0 ) {
            /* Tell the user that we couldn't find a usable */
            /* WinSock DLL.                               */
            printf("can't initialize socket library/n");
            exit(0);
        }
    }
#endif    
    
    memset(&sin, 0, sizeof(sin));
    sin.sin_family = AF_INET;
    
    if ((sin.sin_port = htons(port)) == 0)
        err_quit("invalid port /"%d/"/n", port);
    
    /* Map host name to IP address, allowing for dotted decimal */
    if ( phe = gethostbyname(host) )
        memcpy(&sin.sin_addr, phe->h_addr, phe->h_length);
    else if ( (sin.sin_addr.s_addr = inet_addr(host)) == INADDR_NONE )
        err_quit("can't get /"%s/" host entry/n", host);
    
    /* Map transport protocol name to protocol number */
    if ( (ppe = getprotobyname(transport)) == 0)
        err_quit("can't get /"%s/" protocol entry/n", transport);
    
    /* Allocate a socket */
    s = socket(PF_INET, SOCK_STREAM, ppe->p_proto);
    if (s < 0)
        err_quit("can't create socket: %s/n", strerror(errno));
    
    /* Connect the socket with timeout */
#ifdef WIN32
    ioctlsocket(s,FIONBIO,&non_blocking);
#else
    ioctl(s,FIONBIO,&non_blocking);
#endif
    //fcntl(s,F_SETFL, O_NONBLOCK);
    if (connect(s, (struct sockaddr *)&sin, sizeof(sin)) == -1){
        struct timeval tv; 
        fd_set writefds;
        // 设置连接超时时间
        tv.tv_sec = 10; // 秒数
        tv.tv_usec = 0; // 毫秒
        FD_ZERO(&writefds); 
        FD_SET(s, &writefds); 
        if(select(s+1,NULL,&writefds,NULL,&tv) != 0){ 
            if(FD_ISSET(s,&writefds)){
                int len=sizeof(error); 
                //下面的一句一定要,主要针对防火墙 
                if(getsockopt(s, SOL_SOCKET, SO_ERROR,  (char *)&error, &len) < 0)
                    goto error_ret; 
                if(error != 0) 
                    goto error_ret; 
            }
            else
                goto error_ret; //timeout or error happen 
        }
        else goto error_ret; ; 

#ifdef WIN32
        ioctlsocket(s,FIONBIO,&blocking);
#else
        ioctl(s,FIONBIO,&blocking);
#endif

    }
    else{
error_ret:
        close(s);
        err_quit("can't connect to %s:%d/n", host, port);
    }
    return s;
}

void err_doit(int errnoflag, const char *fmt, va_list ap)
{
    int errno_save;
    char buf[MAX_STRING_LEN];

    errno_save = errno; 
    vsprintf(buf, fmt, ap);
    if (errnoflag)
        sprintf(buf + strlen(buf), ": %s", strerror(errno_save));
    strcat(buf, "/n");
    fflush(stdout);
    fputs(buf, stderr);
    fflush(NULL);
    return;
}

/* Print a message and terminate. */
void err_quit(const char *fmt, ...)
{
    va_list ap;
    va_start(ap, fmt);
    err_doit(0, fmt, ap);
    va_end(ap);
    exit(1);
}

#ifdef WIN32
char *optarg;

char getopt(int c, char *v[], char *opts)
{
    static int now = 1;
    char *p;

    if (now >= c) return EOF;

    if (v[now][0] == '-' && (p = strchr(opts, v[now][1]))) {
        optarg = v[now+1];
        now +=2;
        return *p;
    }

    return EOF;
}

#else
extern char *optarg;
#endif

#define required(a) if (!a) { return -1; }

int init(int argc, char *argv[])
{
    char c;
    //int i,optlen;
    //int slashcnt;

    i_host[0]  =  '/0';
    i_port[0]  =  '/0';

    while ((c = getopt(argc, argv, "h:p:?")) != EOF) {
        if (c == '?')
            return -1;
        switch (c) { 
        case 'h':
            required(optarg);
            strcpy(i_host, optarg);
            break;
        case 'p':
            required(optarg);
            strcpy(i_port, optarg);
            break;
        default:
            return -1;
        }
    }

    /* 
     * there is no default value for hostname, port number, 
     * password or uri
     */

    if (i_host[0] == '/0' || i_port[0] == '/0')
        return -1;

    return 1;
}

void print_usage()
{
    char *usage[] =
    {
        "Usage:",
        "    -h    host name",
        "    -p    port",
        "example:",
        "    -h 127.0.0.1 -p 4001",
    };   
    int i;

    for (i = 0; i < sizeof(usage) / sizeof(char*); i++)
        printf("%s/n", usage[i]);
    
    return;
}

int main(int argc, char *argv[])
{
    SOCKET fd;
    int n;

    /* parse command line etc ... */
    if (init(argc, argv) < 0) {
        print_usage();
        exit(1);
    }

    buf[0] = '/0';

    /* pack the info into the buffer */     
    strcpy(buf, "HelloWorld");

    /* make connection to the server */
    fd = tcp_connect(i_host, (unsigned short)atoi(i_port));

    if(xnet_select(fd, 0, 500, WRITE_STATUS)>0){
        /* send off the message */
        write(fd, buf, strlen(buf));
    }
    else{
        err_quit("Socket I/O Write Timeout %s:%s/n", i_host, i_port);
    }

    if(xnet_select(fd, 3, 0, READ_STATUS)>0){
        /* display the server response */
        printf("Server response:/n");
        n = read(fd, buf, BUFSIZE);
        buf[n] = '/0';
        printf("%s/n", buf);
    }
    else{
        err_quit("Socket I/O Read Timeout %s:%s/n", i_host, i_port);
    }
    close(fd);

#ifdef WIN32
    WSACleanup();
#endif

    return 0;
}

TCP/IP,http,socket,长连接,短连接 总结

网络协议 # http://www.cnblogs.com/liuyong/archive/2011/07/01/2095487.html http://blog.csdn.net/shine0...
  • Siobhan
  • Siobhan
  • 2016年05月18日 09:52
  • 2767

基于Socket的TCP长连接(服务端Java+客户端Android),Service配合AIDL实现

最近公司的项目要求加入消息推送功能,由于项目用户量不是很大,推送需求不是很严格,而且是基于内网的推送,所以我舍弃了使用三方的推送服务,自己使用Socket写了推送功能,剪出一个小Demo来跟大家分享一...
  • qq_29405933
  • qq_29405933
  • 2017年03月30日 16:17
  • 2574

TCP、HTTP与SOCKET连接原理

1、TCP连接 手机能够使用联网功能是因为手机底层实现了TCP/IP协议,可以使手机终端通过无线网络建立TCP连接。TCP协议可以对上层网络提供接口,使上层网络数据的传输建立在“无差别”的网络之...
  • shenggaofei
  • shenggaofei
  • 2016年09月23日 21:16
  • 624

TCP连接、HTTP连接与Socket连接的区别

这会没事,整理了一下HTTP,socket,TCP之间的关系与区别,我们在面试的时候应该会经常问到这方面的东西,那么什么是HTTP呢? HTTP属于老话题了,在项目中我们...
  • min996358312
  • min996358312
  • 2017年04月03日 18:40
  • 1220

socket中TCP建立连接与释放连接

TCP建立连接需要“三次握手” 客户端想服务端发送连接请求,
  • shzxrt
  • shzxrt
  • 2014年05月26日 00:05
  • 1108

[Linux] c 语言tcp socket 示例从简单到复杂

上面例子是最基本的tcp socket 流程,没有错误处理,下面写一个模块 wrap.c 给socket函数加上错误处理代码包装成新的函数; 服务端也不能接收多个客户端请求,使用fork()函数创建一...
  • tzshlyt
  • tzshlyt
  • 2016年11月19日 14:54
  • 1348

基于TCP的socket通信过程及例子

Socket也叫套接字,用来实现网络通讯,通过调用系统提供的API,可以和远程的机子传输数据。Socket有很多种协议,而这篇文章主要讨论TCP部分的内容,文章主要从socket的阻塞模式和非阻塞模式...
  • cwqcwk1
  • cwqcwk1
  • 2013年12月02日 20:21
  • 2749

TCP连接与断开详解(socket通信)

一、TCP数据报结构以及三次握手 TCP(Transmission Control Protocol,传输控制协议)是一种面向连接的、可靠的、基于字节流的通信协议,数据在传输前要建立连接,传输完毕后还...
  • Ctrl_qun
  • Ctrl_qun
  • 2016年09月12日 23:05
  • 8782

socket中的短连接与长连接,心跳包示例详解

 原地址http://blog.chinaunix.net/uid-26000296-id-3758651.html TCP连接简介 当网络通信时采用TCP协议时,在真正的读写操作之前,se...
  • wejoncy
  • wejoncy
  • 2016年04月23日 10:29
  • 646

tcp自连接问题

今天在Segmentfault上有人提问了一个tcp自连接问题,这里记录下。所谓tcp自连接,就是出现源ip和源端口通目的ip和目的端口完全相同的情况,乍看起来不可思议,细细一想情理之中,本文重现下这...
  • secretx
  • secretx
  • 2016年12月29日 18:36
  • 653
内容举报
返回顶部
收藏助手
不良信息举报
您举报文章:TCP的socket连接示例
举报原因:
原因补充:

(最多只允许输入30个字)