Linux下TCP/IP协议的Socket编程

server.c

# include<stdio.h>
# include<arpa/inet.h>
# include<stdlib.h>
# include<unistd.h>
# include<sys/socket.h>
# include<sys/mman.h>
int main(int argc, const char *argv[])
{
    int sfd=socket(AF_INET,SOCK_STREAM,0),cfd,n,*count;                                        //count记录登陆的用户数量
    count=(int *)mmap(NULL,sizeof(int),PROT_READ|PROT_WRITE,MAP_SHARED|MAP_ANONYMOUS,-1,0);    //开辟一个共享内存空间，把地址返回给count
    *count=0;
    struct sockaddr_in s,c;
    char buf[128];
    socklen_t len;
    s.sin_family=AF_INET;                                                                    //后三行初始化一个socket地址。inet_addr 将字符串强制转化ip地址，htons将整数转化为网络可识别的
    s.sin_addr.s_addr=inet_addr(argv[1]);
    s.sin_port=htons(atoi(argv[2]));
    bind(sfd,(struct sockaddr *)&s,sizeof(s));                                                //绑定服务器接口
    listen(sfd,5);                                                                            //打开监听
    while(1)
    {
        puts("Waitting...............");
        len=sizeof(c);
        cfd=accept(sfd,(struct sockaddr *)&c,&len);                                            //接收来自客户端的连接
        printf("%s login, now user is %d
",inet_ntoa(c.sin_addr),++(*count));                //inet_ntoa将ip地址转化为字符串
        while(fork()==0)                                                                    //开辟多进程
        {
            close(sfd);
            while((n=recv(cfd,buf,sizeof(buf),0))!=0)                                        //接收客户端发来的数据再发回去
            {
                buf[n]='';
                printf("recv %s: %s
",inet_ntoa(c.sin_addr),buf);
                send(cfd,buf,n,0);
            }
            close(cfd);                                                                        //关闭cfd
            printf("%s logout, now user is %d
",inet_ntoa(c.sin_addr),--(*count));
            exit(0);
        }
        close(cfd);
    }
    return 0;
}

client.c

# include<stdio.h>
# include<stdlib.h>
# include<unistd.h>
# include<sys/socket.h>
# include<arpa/inet.h>
# include<string.h>
int main(int argc, const char *argv[])
{
    int cfd=socket(AF_INET,SOCK_STREAM,0);                    //建立一个套接字
    struct sockaddr_in s;
    char buf[128];
    s.sin_family=AF_INET;                                    //初始化一个socket
    s.sin_port=htons(atoi(argv[2]));
    s.sin_addr.s_addr=inet_addr(argv[1]);
    connect(cfd,(struct sockaddr *)&s,sizeof(s));            //连接服务器
    while(1)
    {
        gets(buf);
        send(cfd,buf,strlen(buf),0);                        //向服务器发送数据
        recv(cfd,buf,sizeof(buf),0);                        //从服务器接收数据
        puts(buf);
    }
    close(cfd);
    return 0;
}

然后开始编译：

gcc server.c -o server -Wall

gcc client.c -o client -Wall

之后会得到一个名为server和一个名为client可执行文件。

然后先执行server，再执行client如下图

client登陆之后服务器可以看到谁连接了它，并且显示当前连接的用户数量

连接之后我向服务器发送一个“hello”，服务器会显示客户端发来的内容，并发回客户端，客户端会显示自己刚刚发过去的内容。客户端退出之后，服务器会显示退出信息，并且显示当前的登陆用户数量