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;
}