mirror of
https://github.com/riba2534/TCP-IP-NetworkNote.git
synced 2026-04-09 13:49:59 +08:00
补充课后作业
This commit is contained in:
caixiongjiang
143
ch17/homework/char_EPETserv.c
Normal file
143
ch17/homework/char_EPETserv.c
Normal file
@@ -0,0 +1,143 @@
|
||||
//边缘触发
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <errno.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/epoll.h>
|
||||
|
||||
#define BUF_SIZE 100
|
||||
#define MAX_CLNT 256
|
||||
#define EPOLL_SIZE 50
|
||||
|
||||
void setnonblockingmode(int fd);
|
||||
void error_handling(char *buf);
|
||||
void send_msg(char * msg, int len);
|
||||
|
||||
int clnt_cnt = 0;
|
||||
int clnt_socks[MAX_CLNT];
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
int serv_sock, clnt_sock;
|
||||
struct sockaddr_in serv_adr, clnt_adr;
|
||||
socklen_t adr_sz;
|
||||
int str_len, i;
|
||||
char buf[BUF_SIZE];
|
||||
|
||||
struct epoll_event *ep_events;
|
||||
struct epoll_event event;
|
||||
int epfd, event_cnt;
|
||||
|
||||
if(argc!=2) {
|
||||
printf("Usage : %s <port>\n", argv[0]);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
|
||||
memset(&serv_adr, 0, sizeof(serv_adr));
|
||||
serv_adr.sin_family = AF_INET;
|
||||
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||||
serv_adr.sin_port = htons(atoi(argv[1]));
|
||||
|
||||
if(bind(serv_sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr)) == -1)
|
||||
error_handling("bind() error");
|
||||
if(listen(serv_sock, 5) == -1)
|
||||
error_handling("listen() error");
|
||||
|
||||
epfd = epoll_create(EPOLL_SIZE);
|
||||
ep_events = malloc(sizeof(struct epoll_event)*EPOLL_SIZE);
|
||||
|
||||
setnonblockingmode(serv_sock);
|
||||
event.events = EPOLLIN;
|
||||
event.data.fd = serv_sock;
|
||||
epoll_ctl(epfd, EPOLL_CTL_ADD, serv_sock, &event);
|
||||
|
||||
while(1)
|
||||
{
|
||||
event_cnt = epoll_wait(epfd, ep_events, EPOLL_SIZE, -1);
|
||||
if(event_cnt == -1)
|
||||
{
|
||||
puts("epoll_wait() error");
|
||||
break;
|
||||
}
|
||||
|
||||
puts("return epoll_wait");
|
||||
for(i = 0; i < event_cnt; ++i)
|
||||
{
|
||||
if(ep_events[i].data.fd == serv_sock)
|
||||
{
|
||||
adr_sz = sizeof(clnt_adr);
|
||||
clnt_sock = accept(serv_sock, (struct sockaddr*)&clnt_adr, &adr_sz);
|
||||
setnonblockingmode(clnt_sock);
|
||||
event.events = EPOLLIN|EPOLLET;
|
||||
event.data.fd = clnt_sock;
|
||||
epoll_ctl(epfd, EPOLL_CTL_ADD, clnt_sock, &event);
|
||||
clnt_socks[clnt_cnt++] = clnt_sock;
|
||||
printf("connected client: %d \n", clnt_sock);
|
||||
}
|
||||
else
|
||||
{
|
||||
while(1)
|
||||
{
|
||||
str_len = read(ep_events[i].data.fd, buf, BUF_SIZE);
|
||||
if(str_len == 0) // close request!
|
||||
{
|
||||
epoll_ctl(epfd, EPOLL_CTL_DEL, ep_events[i].data.fd, NULL);
|
||||
close(ep_events[i].data.fd);
|
||||
|
||||
for(i = 0; i < clnt_cnt; ++i)
|
||||
{
|
||||
if(ep_events[i].data.fd == clnt_socks[i])
|
||||
{
|
||||
while(i++ < clnt_cnt-1)
|
||||
clnt_socks[i] = clnt_socks[i+1];
|
||||
break;
|
||||
}
|
||||
}
|
||||
--clnt_cnt;
|
||||
printf("closed client: %d \n", ep_events[i].data.fd);
|
||||
break;
|
||||
}
|
||||
else if(str_len < 0)
|
||||
{
|
||||
if(errno == EAGAIN)
|
||||
break;
|
||||
}
|
||||
else
|
||||
{
|
||||
send_msg(buf, str_len);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
close(serv_sock);
|
||||
close(epfd);
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
void send_msg(char * msg, int len) // send to all
|
||||
{
|
||||
int i;
|
||||
for(i = 0; i < clnt_cnt; ++i)
|
||||
write(clnt_socks[i], msg, len);
|
||||
}
|
||||
|
||||
//设置为非阻塞模式
|
||||
void setnonblockingmode(int fd)
|
||||
{
|
||||
int flag = fcntl(fd, F_GETFL, 0);
|
||||
fcntl(fd, F_SETFL, flag|O_NONBLOCK);
|
||||
}
|
||||
|
||||
void error_handling(char *buf)
|
||||
{
|
||||
fputs(buf, stderr);
|
||||
fputc('\n', stderr);
|
||||
exit(1);
|
||||
}
|
||||
124
ch17/homework/char_EPLTserv.c
Normal file
124
ch17/homework/char_EPLTserv.c
Normal file
@@ -0,0 +1,124 @@
|
||||
//条件触发
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <unistd.h>
|
||||
#include <fcntl.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <sys/socket.h>
|
||||
#include <sys/epoll.h>
|
||||
|
||||
#define BUF_SIZE 100
|
||||
#define MAX_CLNT 256
|
||||
#define EPOLL_SIZE 50
|
||||
void error_handling(char *buf);
|
||||
void send_msg(char *msg, int len);
|
||||
|
||||
int clnt_cnt = 0;
|
||||
int clnt_socks[MAX_CLNT];
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
int serv_sock, clnt_sock;
|
||||
struct sockaddr_in serv_adr, clnt_adr;
|
||||
socklen_t adr_sz;
|
||||
int str_len, i;
|
||||
char buf[BUF_SIZE];
|
||||
|
||||
struct epoll_event *ep_events;
|
||||
struct epoll_event event;
|
||||
int epfd, event_cnt;
|
||||
|
||||
if(argc != 2) {
|
||||
printf("Usage: %s <port>\n", argv[0]);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
serv_sock = socket(PF_INET, SOCK_STREAM, 0);
|
||||
memset(&serv_adr, 0, sizeof(serv_adr));
|
||||
serv_adr.sin_family = AF_INET;
|
||||
serv_adr.sin_addr.s_addr = htonl(INADDR_ANY);
|
||||
serv_adr.sin_port = htons(atoi(argv[1]));
|
||||
|
||||
if(bind(serv_sock, (struct sockaddr*) &serv_adr, sizeof(serv_adr)) == -1)
|
||||
error_handling("bind() error");
|
||||
if(listen(serv_sock, 5) == -1)
|
||||
error_handling("listen() error");
|
||||
|
||||
//创建epoll例程
|
||||
epfd = epoll_create(EPOLL_SIZE);
|
||||
//动态分配内存
|
||||
ep_events = malloc(sizeof(struct epoll_event)*EPOLL_SIZE);
|
||||
|
||||
event.events = EPOLLIN;//事件类型为需要读取的情况
|
||||
event.data.fd = serv_sock;
|
||||
//在epoll例程中注册服务端套接字
|
||||
epoll_ctl(epfd, EPOLL_CTL_ADD, serv_sock, &event);
|
||||
|
||||
while(1)
|
||||
{
|
||||
//获取发生改变的文件描述符数量
|
||||
event_cnt = epoll_wait(epfd, ep_events, EPOLL_SIZE, -1);
|
||||
if(event_cnt == -1)
|
||||
break;
|
||||
|
||||
for(i = 0; i < event_cnt; ++i)
|
||||
{
|
||||
if(ep_events[i].data.fd == serv_sock)//serv_sock
|
||||
{
|
||||
adr_sz = sizeof(clnt_adr);
|
||||
clnt_sock = accept(serv_sock, (struct sockaddr*) &clnt_adr, &adr_sz);
|
||||
event.events = EPOLLIN;
|
||||
event.data.fd = clnt_sock;
|
||||
//在epoll例程中注册clnt_sock套接字
|
||||
epoll_ctl(epfd, EPOLL_CTL_ADD, clnt_sock, &event);
|
||||
clnt_socks[clnt_cnt++] = clnt_sock;
|
||||
printf("connected client: %d \n", clnt_sock);
|
||||
}
|
||||
else //clnt_sock
|
||||
{
|
||||
str_len = read(ep_events[i].data.fd, buf, BUF_SIZE);
|
||||
if(str_len == 0)
|
||||
{
|
||||
//在例程中删除clnt_sock套接字
|
||||
epoll_ctl(epfd, EPOLL_CTL_DEL, ep_events[i].data.fd, NULL);
|
||||
//关闭clnt_sock套接字
|
||||
close(ep_events[i].data.fd);
|
||||
printf("closed client: %d \n", ep_events[i].data.fd);
|
||||
|
||||
for(i = 0; i < clnt_cnt; ++i)
|
||||
{
|
||||
if(clnt_sock == clnt_socks[i])
|
||||
{
|
||||
while(i++ < clnt_cnt - 1)
|
||||
clnt_socks[i] = clnt_socks[i + 1];
|
||||
break;
|
||||
}
|
||||
}
|
||||
--clnt_cnt;
|
||||
}
|
||||
else
|
||||
{
|
||||
send_msg(buf, str_len);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
close(serv_sock);
|
||||
close(epfd);
|
||||
return 0;
|
||||
}
|
||||
|
||||
void send_msg(char * msg, int len)
|
||||
{
|
||||
int i;
|
||||
for(i = 0; i < clnt_cnt; ++i)
|
||||
write(clnt_socks[i], msg, len);
|
||||
}
|
||||
|
||||
void error_handling(char *buf)
|
||||
{
|
||||
fputs(buf, stderr);
|
||||
fputc('\n', stderr);
|
||||
exit(1);
|
||||
}
|
||||
93
ch17/homework/chat_clnt.c
Normal file
93
ch17/homework/chat_clnt.c
Normal file
@@ -0,0 +1,93 @@
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <unistd.h>
|
||||
#include <string.h>
|
||||
#include <arpa/inet.h>
|
||||
#include <sys/socket.h>
|
||||
#include <pthread.h>
|
||||
|
||||
#define BUF_SIZE 100
|
||||
#define NAME_SIZE 20
|
||||
|
||||
void * send_msg(void * arg);
|
||||
void * recv_msg(void * arg);
|
||||
void error_handling(char * msg);
|
||||
|
||||
char name[NAME_SIZE] = "[DEFAULT]";
|
||||
char msg[BUF_SIZE];
|
||||
|
||||
int main(int argc, char *argv[])
|
||||
{
|
||||
int sock;
|
||||
struct sockaddr_in adr;
|
||||
pthread_t send_thread, recv_thread;
|
||||
void * thread_return;//线程返回值
|
||||
if(argc != 4) {
|
||||
printf("Usage: %s <IP> <port> <name> \n", argv[0]);
|
||||
exit(1);
|
||||
}
|
||||
|
||||
sprintf(name, "[%s]", argv[3]);//把名字写入name数组
|
||||
sock = socket(PF_INET, SOCK_STREAM, 0);
|
||||
|
||||
memset(&adr, 0, sizeof(adr));
|
||||
adr.sin_family = AF_INET;
|
||||
adr.sin_addr.s_addr = inet_addr(argv[1]);
|
||||
adr.sin_port = htons(atoi(argv[2]));
|
||||
|
||||
if(connect(sock, (struct sockaddr*) &adr, sizeof(adr)) == -1)
|
||||
error_handling("connect() error");
|
||||
/*线程创建*/
|
||||
pthread_create(&send_thread, NULL, send_msg, (void *)&sock);
|
||||
pthread_create(&recv_thread, NULL, recv_msg, (void *)&sock);
|
||||
/*进程阻塞，进入发送消息的线程*/
|
||||
pthread_join(send_thread, &thread_return);
|
||||
/*进程阻塞，进入接收消息的线程*/
|
||||
pthread_join(recv_thread, &thread_return);
|
||||
close(sock);
|
||||
return 0;
|
||||
}
|
||||
|
||||
void * send_msg(void * arg)
|
||||
{
|
||||
int sock = *((int *)arg);
|
||||
char name_msg[NAME_SIZE + BUF_SIZE];
|
||||
|
||||
while(1)
|
||||
{
|
||||
fputs("Input message(q to quit):\n", stdout);
|
||||
fgets(msg, BUF_SIZE, stdin);
|
||||
if(!(strcmp(msg, "q\n")) || !strcmp(msg, "Q\n"))
|
||||
{
|
||||
close(sock);
|
||||
exit(0);
|
||||
}
|
||||
sprintf(name_msg, "%s %s", name, msg);
|
||||
write(sock, name_msg, strlen(name_msg));//strlen()函数碰到'\0'会结束
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void * recv_msg(void * arg)
|
||||
{
|
||||
int sock = *((int *)arg);
|
||||
char name_msg[NAME_SIZE + BUF_SIZE];
|
||||
int str_len;
|
||||
|
||||
while(1)
|
||||
{
|
||||
str_len = read(sock, name_msg, NAME_SIZE + BUF_SIZE - 1);
|
||||
if(str_len == -1)
|
||||
return (void *)-1;
|
||||
name_msg[str_len] = 0;
|
||||
fputs(name_msg, stdout);
|
||||
}
|
||||
return NULL;
|
||||
}
|
||||
|
||||
void error_handling(char *msg)
|
||||
{
|
||||
fputs(msg, stderr);
|
||||
fputc('\n', stderr);
|
||||
exit(1);
|
||||
}
|
||||
Reference in New Issue
Block a user