echo_server.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
void error_handling(char *message);
int main(int argc, char *argv[])
{
int serv_sock;
int clnt_sock;
struct sockaddr_in serv_addr;
struct sockaddr_in clnt_addr;
socklen_t clnt_addr_size;
int str_len;
char message[1024];
if(argc!=2)
{
exit(1);
}
//TCP socket
//1.安装电话机
serv_sock=socket(PF_INET, SOCK_STREAM, 0);
if(serv_sock == -1)
error_handling("socket error!");
//2.调用bind分配电话号码
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET; //IPV4协议族
//INADDR_ANY表示自动获取服务器端IP地址,
//如果计算机只有一个NIC(网络接口卡),则可以这么配置,如果有多个,则需要手动配置
//服务器IP地址的数量与NIC数量有关
serv_addr.sin_addr.s_addr = htonl(INADDR_ANY); //主机字节序(host)转换成网络字节序(net)(大端序)
serv_addr.sin_port = htons(atoi(argv[1])); //端口号
//表示请把进入IP...、9190端口的数据传给我
if(bind(serv_sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)
error_handling("bind error");
//3.调用listen连接电话线(进入等待连接请求状态)
//5表示连接请求等待队列为5
if(listen(serv_sock, 5) == -1)
error_handling("listen error");
//4.调用accept拿起话筒(受理连接请求)
clnt_addr_size = sizeof(clnt_addr);
int i;
for(i=0;i<5;i++)
{
//服务器调用accept函数时进入阻塞状态,直到有请求进入
//第二个参数保存发起连接请求的客户端地址
clnt_sock=accept(serv_sock, (struct sockaddr*) &clnt_addr, &clnt_addr_size);
if(clnt_sock == -1)
error_handling("accept error");
else
printf("connect client: %d \n", i+1);
//read函数返回接受的字节数(成功时),遇到文件结尾返回0,失败时返回-1
//对于一个套接字的read和write默认是阻塞的,有数据时才会有返回
while((str_len=read(clnt_sock, message, 1024))!=0)
{
//由于线程阻塞,所以在while循环中不能进行printf显示,跳出循环后才会显示
//printf("%d", str_len);
write(clnt_sock, message, str_len);
}
close(clnt_sock);
}
close(serv_sock);
return 0;
}
void error_handling(char *message)
{
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}
echo_client.c
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
#define BUF_SIZE 1024
void error_handling(char *message);
int main(int argc, char *argv[])
{
int sock;
struct sockaddr_in serv_addr;
int str_len;
char message[BUF_SIZE];
if(argc!=3)
{
exit(1);
}
sock=socket(PF_INET, SOCK_STREAM, 0);
if(sock == -1)
error_handling("socket error!");
memset(&serv_addr, 0, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
//基于字符串的IP地址初始化
//也可以使用 int inet_aton(const char* string, struct in_addr* addr);
//反向转换 char* inet_ntoa(struct in_addr adr);
serv_addr.sin_addr.s_addr = inet_addr(argv[1]);
serv_addr.sin_port = htons(atoi(argv[2]));
//向服务器发起连接请求
//注意:客户端IP地址在调用connect时会自动分配
if(connect(sock, (struct sockaddr*) &serv_addr, sizeof(serv_addr)) == -1)
error_handling("connect error");
else
puts("Connected.....");
while(1)
{
fputs("Input Message: ", stdout);
fgets(message, BUF_SIZE, stdin);
if(!strcmp(message, "q\n") || !strcmp(message, "Q\n"))
break;
// //存在问题:多次调用write函数传递的字符串有可能一次性传输到服务器
// //客户端有可能在尚未受到全部数据包时就调用read函数
// write(sock, message, strlen(message));
// str_len=read(sock, message, BUF_SIZE-1);
//记录接收字符的大小有效解决一次读取不全的问题
str_len=write(sock, message, strlen(message));
int recv_len=0;
while(recv_len<str_len)
{
int recv_cnt = read(sock, &message[recv_len], BUF_SIZE-1);
if(recv_cnt==-1)
error_handling("read error");
recv_len+=recv_cnt;
}
message[str_len]=0;
printf("Message from server: %s", message);
}
close(sock);
return 0;
}
void error_handling(char *message)
{
fputs(message, stderr);
fputc('\n', stderr);
exit(1);
}