TCP服务器模型

TCP服务器模型

一、TCP服务器的模型

1、循环服务器

socket(...);
bind(...);
listen(...);
accept(...);
while(1){
read(...);
...
write(...);
}
TCP循环服务器用的相对较少

2、并发服务器

并发服务器实现原理是多次调用accept函数

socket(...);

bind(...);

listen(...);
while(1){
accept(...);
pthread_create(...); //创建线程处理新的链接
}

多线程实现并发

#ifndef _SER_H_
#define _SER_H_
#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <stdlib.h>

#define SIZE 32

int tcp_server_init(char *ip, int port, int backlog);
int tcp_server_wait(int fd);

#endif

//服务端

#include "ser.h"

int tcp_server_init(char *ip, int port, int backlog)
{
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == sfd){
        perror("socket");
        return -1;
    }
    
    struct sockaddr_in sddr;
    
    sddr.sin_family      = AF_INET;
    sddr.sin_port        = htons(port);
    sddr.sin_addr.s_addr = inet_addr(ip);
    
    if(-1 == bind(sfd, (void *)&sddr, sizeof sddr)){
        perror("bind");
        return -1;
    }

    if(-1 == listen(sfd, backlog)){
        perror("listen");
        return -1;
    }
    puts("Server-------OK");
    return sfd;
}

int tcp_server_wait(int fd)
{
    struct sockaddr_in cddr;
    socklen_t size = sizeof cddr;
    int nfd = accept(fd, (void *)&cddr, &size);
    if(-1 == nfd){
        perror("accept");
        return -1;
    }
    printf("IP:%s PORT:%d connected!\n", inet_ntoa(cddr.sin_addr),ntohs(cddr.sin_port));
    return nfd;
}


//main函数

#include <stdio.h>
#include "ser.h"
#include <pthread.h>

//线程执行函数
void *server_thread_handle(void *arg);

int main(int argc, char *argv[])
{
    int fd = tcp_server_init("0.0.0.0", 6666, 10);
    while(1){
        pthread_t tid;
        
        int nfd = tcp_server_wait(fd);
        
        if(0 != pthread_create(&tid, NULL, server_thread_handle, &nfd)){
            close(nfd);
            continue;
        }
    }
    return 0;
}

void *server_thread_handle(void *arg)
{
    int fd = *(int *)arg;
    char buf[SIZE];
    int ret = -1;
    while(1){
        ret = read(fd, buf, SIZE);
        if(ret <= 0){
            close(fd);
            break;
        }
        printf("Recv: %s", buf);
        write(fd, buf, SIZE);
    }
}

多进程实现并发


socket(...);
bind(...);
listen(...);
while(1){
accept(...);
fork(...); //创建子进程处理新的链接
}
记得回收子进程,否则会出现僵尸进程



//ser.h

#ifndef _SER_H_
#define _SER_H_
#include <stdio.h>
#include <sys/types.h>          /* See NOTES */
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <stdlib.h>

#define SIZE 32

int tcp_server_init(char *ip, int port, int backlog);
int tcp_server_wait(int fd);

#endif


/*===============================================
*   文件名称:ser.c
*   创 建 者:     
*   创建日期:2022年08月01日
*   描    述:
================================================*/
#include "ser.h"

int tcp_server_init(char *ip, int port, int backlog)
{
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == sfd){
        perror("socket");
        return -1;
    }
    
    struct sockaddr_in sddr;
    
    sddr.sin_family      = AF_INET;
    sddr.sin_port        = htons(port);
    sddr.sin_addr.s_addr = inet_addr(ip);
    
    if(-1 == bind(sfd, (void *)&sddr, sizeof sddr)){
        perror("bind");
        return -1;
    }

    if(-1 == listen(sfd, backlog)){
        perror("listen");
        return -1;
    }
    puts("Server-------OK");
    return sfd;
}

int tcp_server_wait(int fd)
{
    struct sockaddr_in cddr;
    socklen_t size = sizeof cddr;
    int nfd = accept(fd, (void *)&cddr, &size);
    if(-1 == nfd){
        perror("accept");
        return -1;
    }
    printf("IP:%s PORT:%d connected!\n", inet_ntoa(cddr.sin_addr),ntohs(cddr.sin_port));
    return nfd;
}


/*===============================================
*   文件名称:main.c
*   创 建 者:     
*   创建日期:2022年08月01日
*   描    述:
================================================*/
#include <stdio.h>
#include "ser.h"
#include <signal.h>
#include <sys/wait.h>

//子进程的处理函数
void server_child_handle(int fd);

//信号处理函数
void wait_child_handle(int no);

int main(int argc, char *argv[])
{
    int fd = tcp_server_init("0.0.0.0", 6666, 10);
    if(-1 == fd){
        return -1;
    }
    
    //将SIGCHILD信号和信号处理函数关联
    signal(SIGCHLD, wait_child_handle);
    while(1){
        int nfd = tcp_server_wait(fd);
        if(-1 == nfd){
            continue;
        }
        
        pid_t pid = fork();
        if(-1 == pid){   //err
            continue;
        }else if(0 == pid){ //子进程
            close(fd);  //关掉的是子进程的fd
            server_child_handle(nfd);
            break;
        }else{
            close(nfd); //关掉的是父进程的nfd,避免资源浪费
        }
    }
    
    return 0;
}

void server_child_handle(int fd)
{
    char buf[SIZE];
    int ret = -1;
    while(1){
        ret = read(fd, buf, SIZE);
        if(ret <= 0){
            close(fd);
            break;
        }
        printf("Recv: %s", buf);
        write(fd, buf, SIZE);
    }
}

void wait_child_handle(int no)
{
    waitpid(-1, NULL, WNOHANG);
}

tcp实现传图片

客户端:

#include <stdio.h>
#include <sys/types.h>          
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <fcntl.h>
#include <string.h>

int main(int argc, char *argv[])
{
    if(3 > argc){
        printf("arc:%d argv[0]:%s IP PORT\n", argc, argv[0]);
        return -1;
    }
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == sfd){
        perror("socket");
        return -1;
    }

    //设置服务器的IP PORT
    struct sockaddr_in sddr = {0};
    sddr.sin_family      = AF_INET;
    sddr.sin_port        = htons(atoi(argv[2]));
    sddr.sin_addr.s_addr = inet_addr(argv[1]);

    //连接服务器
    if(-1 == connect(sfd, (void *)&sddr, sizeof sddr)){
        perror("connect");
        return -1;
    }
    char fname[20];
    printf("input get filename: ");
    fgets(fname, 20, stdin);
    int n = strlen(fname);
    //取消换行符
    fname[n-1] = 0;
    //发送文件名
    write(sfd, fname, 20);
    char stat[3];
    read(sfd, stat, 3);
    if(0 == strcmp(stat, "NO")){
        printf("服务器没有 %s 文件\n", fname);
        return 0;
    }
    int fd = open(fname, O_WRONLY|O_CREAT, 0777);
    if(-1 == fd){
        perror("open");
        return -1;
    }
    int len = -1;
    char buf[100];
    while(1){
        //接收图片数据的长度
        read(sfd, &len, 4);
        if(0 == len){
            puts("Recv OK");
            break;
        }
        //接收对应长度的图片数据
        read(sfd, buf, len);
        //将图片数据写入到文件
        write(fd, buf, len);
    }
    close(sfd);
    close(fd);
    return 0;
}

服务端:

#include <stdio.h>
#include <sys/types.h> 
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/ip.h> 
#include <string.h>
#include <strings.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <unistd.h>
#include <sys/stat.h>
#include <fcntl.h>

#define PSIZE 100

int main(int argc, char *argv[])
{
    //应用层到底层的通道
    /*************************
     *AF_INET:     IPv4协议簇
     *SOCK_STREAM: TCP连接
     *0:          默认为0
     *retrun:      新的文件描述符
     *************************/
    int sfd = socket(AF_INET, SOCK_STREAM, 0);
    if(-1 == sfd){
        perror("socket");
        return -1;
    }

    //设置IP PORT
    struct sockaddr_in sddr, cddr;
    //使用memset清空sddr
    memset(&sddr, 0, sizeof sddr);
    //bzero清空cddr
    bzero(&cddr, sizeof cddr);
    //填充结构体的内容
    sddr.sin_family      = AF_INET;
    //htons将本机字节序转成网络字节序
    sddr.sin_port        = htons(8888);
    //将点分十进制的IP转成网络字节序
    //"0.0.0.0" == INADDR_ANY :任意IP(自己主机网卡的IP)
    sddr.sin_addr.s_addr = htonl(INADDR_ANY);

    /****************************************
     *bind:          将IP PORT和socket套接字绑定
     *sfd:           目标套接字的文件描述符
     *(void *)&sddr: 设置好的IP PORT
     * sizeof sddr : 设置的IP PORT的大小
     * return:       0或-1,表示成功或者失败
     ****************************************/
    if(-1 == bind(sfd, (void *)&sddr, sizeof sddr)){
        perror("bind");
        return -1;
    }

    /*************************************
     *listen: 监听套节字有无连接进来
     *sfd:    目标套接字的文件描述符
     *10:    监听队列的长度
     *return: 0或-1,表示是否成功
     *************************************/
    if(-1 == listen(sfd, 10)){
        perror("listen");
        return -1;
    }
    puts("listen...");

    /****************************************
     *accept: 和客户端建立新的连接
     *sfd:    要建立连接的套接字
     *cddr:  存储连接进来客户端的信息(IP PORT)
     *len:    上一个参数的大小
     *return: 表示新的连接
     *****************************************/
    socklen_t size = sizeof cddr;
    int nfd = accept(sfd, (void *)&cddr, &size);
    if(-1 == nfd){
        perror("accept");
        return -1;
    }

    //解析连接客户端的IP PORT
    //ntohs:     将网络字节序转换为本机字节序
    //inet_ntoa: 将网络字节序转换成点分十进制
    printf("IP:%s PORT:%d connected!\n", inet_ntoa(cddr.sin_addr),ntohs(cddr.sin_port));

    char buf[PSIZE]; //图片的数据
    char fname[20];  //要读取的图片名字
    int len = -1;    //读取图片的数据长度
    int fd = -1;

    /*先接收文件名,然后打开对应文件,成功给客户端发送OK 失败
     * 发送NO*/
    read(nfd, fname, 20);
    fd = open(fname, O_RDONLY);
    if(-1 == fd){
        write(nfd, "NO", 3);
        return -1;
    }else{
        write(nfd, "OK", 3);
    }
    while(1){
        //读取图片数据
        len = read(fd, buf, PSIZE);
        //先发送长度
        write(nfd, &len, 4);
        if(0 == len){
            break;
        }
        //发送图片数据
        write(nfd, buf, len);
    }
    close(nfd);
    close(fd);
    close(sfd);
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值