UNIXネットワークプログラミング(第3版)テストコード7.2
3828 ワード
#ifndef UNP_H_
#define UNP_H_
/**
* @file unp.h
* @brief global structure for unp
* @author xiyanxiyan10
* @TODO ,
*
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <signal.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netinet/tcp.h>
#include <arpa/inet.h>
#include <signal.h>
#include <errno.h>
#include <sys/mman.h>
#include <sys/ioctl.h>
#include <pthread.h>
#include <errno.h>
#define SERVER_PORT 8899
#define MAX_LISTEN 10
#define MAXN 1024
typedef unsigned int size_t;
typedef signed int ssize_t;
#endif
server
#include "unp.h"
void str_echo(int sockfd){
ssize_t n;
char buf[MAXN];
again:
while( (n = read(sockfd, buf, MAXN) )> 0)
write(sockfd, buf, MAXN);
if(n < 0 && errno == EINTR)
goto again;
if(n < 0)
strerror(errno);
if(n == 0)
return;
}
int main(int argc, char **argv)
{
int listen_fd;
int conn_fd;
pid_t child_fd;
socklen_t client_len;
struct sockaddr_in server_addr, client_addr;
char client_str[INET_ADDRSTRLEN];
memset( (void *)&server_addr, 0, sizeof(struct sockaddr_in));
memset( (void *)&client_addr, 0, sizeof(struct sockaddr_in));
memset( (void *)client_str, 0, sizeof(client_str));
listen_fd = socket(AF_INET, SOCK_STREAM, 0);
server_addr.sin_family = AF_INET;
server_addr.sin_addr.s_addr = htons(INADDR_ANY);
server_addr.sin_port = htons(SERVER_PORT);
bind(listen_fd, (struct sockaddr *)&server_addr, sizeof(server_addr));
listen(listen_fd, MAX_LISTEN);
while(1){
conn_fd = accept( listen_fd, (struct sockaddr*)&client_addr, &client_len);
child_fd = fork();
if(child_fd == 0){
close(listen_fd);
str_echo(conn_fd);
exit(0);
}
inet_ntop(AF_INET, (void *)&client_addr, client_str, INET_ADDRSTRLEN);
fprintf(stderr, "conn_fd from %s
", client_str);
close(conn_fd);
memset( (void *)&client_addr, 0, sizeof(struct sockaddr_in));
memset( (void *)client_str, 0, sizeof(client_str));
}
return 0;
}
client
#include "unp.h"
void print_mss(int connfd){
int mss;
socklen_t size = sizeof(mss);
if( getsockopt(connfd, IPPROTO_TCP, TCP_MAXSEG, &mss, &size) < 0){
fprintf(stderr, "get mss fail
");
}else{
fprintf(stderr, "get mss success %u
", mss);
}
}
void str_cli(FILE *fp, int conn_fd){
char buf[MAXN];
while( fgets(buf, MAXN, fp) != NULL){
write(conn_fd,buf, MAXN);
read(conn_fd, buf, MAXN);
fputs(buf, stdout);
print_mss(conn_fd);
}
}
int main(int argc, char **argv)
{
int conn_fd;
struct sockaddr_in server_addr;
if(argc != 2){
fprintf(stderr, "server address needed
");
return -1;
}
memset( (void *)&server_addr, 0, sizeof(struct sockaddr_in));
conn_fd = socket(AF_INET, SOCK_STREAM, 0);
inet_pton(AF_INET, argv[1], (void*)&server_addr);
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(SERVER_PORT);
if ( connect(conn_fd, (struct sockaddr *)&server_addr, sizeof(server_addr)) < 0){
fprintf(stderr, "connect fail to server %s
", argv[1]);
exit(0);
}
str_cli(stdin, conn_fd);
exit(0);
}