网络编程2 - TCP套接口编程

网络编程2 - TCP套接口程序实例

有了网络编程1 - socket基本函数的基础之后，就可用开始写TCP套接口的C/S程序实例了。
程序分为服务端server_tcp.c和客户端client_tcp.c两部分。
两部分的基本步骤如下：
服务端：socket -> bind -> listen -> accept -> write
客户端：socket -> connect -> read

1.服务端程序
server_tcp.c代码如下：
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <errno.h>
#include <string.h>
#define MAXSIZE 1024



int main(int argc, char *argv[])
{
    int sockfd, newsockfd;
    struct sockaddr_in server_addr, client_addr; //定义server和client套接口
    char buf[MAXSIZE];
    if(argc <= 1)
    {
        printf("please input the port number like 5566");
        exit(1);
    }
    int PORT = atoi(argv[1]);

    if((sockfd=socket(AF_INET,SOCK_STREAM,0))<0)
    {
        printf("socket error:%s\n",strerror(errno));
        exit(1);
    }
    bzero(&server_addr,sizeof(struct sockaddr_in));
    server_addr.sin_family=AF_INET;
    server_addr.sin_addr.s_addr=htonl(INADDR_ANY);
    server_addr.sin_port=htons(PORT);
    if(bind(sockfd,(struct sockaddr*)(&server_addr),sizeof(struct sockaddr))<0)
    {
        printf("bind error:%s\n",strerror(errno));
        exit(1);
    }
    const int max_connection = 5;
    if(listen(sockfd,max_connection)<0)
    {
        printf("listen error:%s\n",strerror(errno));
        exit(1);
    }
    else
    {
        printf("listenning....\n");
    }

    socklen_t sin_size=sizeof(struct sockaddr_in);
    while(1)
    {
        if((newsockfd=accept(sockfd,(struct sockaddr*)(&client_addr),&sin_size))<0)
        {
            printf("accept error:%s\n",strerror(errno));
            exit(1);
        }
        printf("server get connection from %s\n",inet_ntoa(client_addr.sin_addr));
        printf("connected successfully! pelase input the message[<1024 bytes]\n");
        if(fgets(buf,sizeof(buf),stdin) == NULL)
        {
            printf("fgets error\n");
            exit(1);
        }
        if(write(newsockfd,buf,strlen(buf))<0)
        {
            printf("write error:%s\n",strerror(errno));
            exit(1);
        }
        close(newsockfd);
    }
    close(sockfd);

    return 0;
}


2.客户端程序
client_tcp.c代码如下：

#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#include <fcntl.h>
#include <sys/stat.h>
#include <errno.h>
#include <string.h>
#include <netdb.h>
#define MAXSIZE 1024

int main(int argc, char * argv[])
{
    int sockfd;
    struct sockaddr_in server_addr; //定义server和client套接口
    char buf[MAXSIZE];
    struct hostent * host;
    char * HOST=argv[1];
    int PORT=atoi(argv[2]);

    if(argc < 3)
    {
        printf("please input the host AND port number like: 127.0.0.1 5566");
        exit(1);
    }

    if( (host=gethostbyname(HOST))==NULL)
    {
        printf("get host name error:%s\n",strerror(errno));
        exit(1);
    }

    if((sockfd=socket(AF_INET,SOCK_STREAM,0))<0)
    {
        printf("socket error:%s\n",strerror(errno));
        exit(1);
    }
    bzero(&server_addr,sizeof(struct sockaddr_in));
    server_addr.sin_family=AF_INET;
    server_addr.sin_addr=*((struct in_addr *)host->h_addr);
    server_addr.sin_port=htons(PORT);
    if(connect(sockfd,(struct sockaddr*)(&server_addr),sizeof(struct sockaddr))<0)
    {
        printf("connect error:%s\n",strerror(errno));
        exit(1);
    }
    int nbytes;
    if((nbytes=read(sockfd,buf,MAXSIZE))<0)
    {
        printf("read error:%s\n",strerror(errno));
        exit(1);
    }
    buf[nbytes]='\0';
    printf("I have receive:%s\n",buf);
    close(sockfd);

    return 0;
}

3.运行方法
打开两个shell终端程序，一个用于server，一个用于client，都要先进入.c源文件所在目录（使用cd命令）。

服务端输入：
gcc server_tcp.c -o server_tcp
./server_tcp 5678

客户端输入：
gcc client_tcp.c -o client_tcp
./client_tcp 127.0.0.1 5678

服务端输出：
server get connection from 127.0.0.1
connected successfully! pelase input the message[<1024 bytes]
服务端输入：
hello world! linux programming!

服务端输出：
I have receive:hello world! linux programming!