【原创】TCP Socket 简单练习 --- 新进程辅助通信 分类： Linux --- 应用程序设计 2014-12-23 10:27 64人阅读 评论(0) 收藏

【原创】TCP Socket 简单练习 --- 新进程辅助通信

补充：

在进行地址转换时用inet_pton

//将ip地址转换为32位网络地址 inet_addr inet_aton或者inet_pton
//但为了安全，最好用inet_pton代替inet_addr和inet_aton这两个函数
//三种方式使用方法如下，其中argv[1]为从命令行获得的IP地址
1. inet_aton(argv[1], (struct in_addr *) &dest.sin_addr.s_addr);
2. server_addr.sin_addr.s_addr = inet_addr(argv[1]);
3. inet_pton(AF_INET, argv[1], &server_addr.sin_addr);//最安全

运行方式

服务器端，可以直接运行命令，也可以带参数运行，如果不带参数运行，则程序自主获取主机ip，然后默认设定port和lisnum的值。其中port默认为6666，lisnum默认为5.

./socket_select_server IP地址（可选） 端口号（可选）监听队列大小（可选）

客户端，必须指定服务器的IP地址和端口号，例如：

./socket_select_client 172.18.229.60 6666

服务器端代码

/*************************************************************************
> File Name: tcp_server.c
> Author: genglut
> Mail: genglut@163.com
> Created Time: 2014年12月22日 星期一 10时14分53秒
************************************************************************/

/*
struct sockaddr_in
{
short int sin_family;                // 地址协议
in_port_t sin_port;    　　　		   // 端口号
struct in_addr sin_addr;         　　// IP地址
unsigned char sin_zero[8];     　　　// 预留位
};

struct in_addr
{
_u32 s_addr;           　　// 32位地址
};
*/

#include <stdio.h>
#include <stdlib.h>
#include <errno.h>
#include <string.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <sys/socket.h>
#include <sys/wait.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <sys/ioctl.h>

#define MAXBUF 1024

void get_ip(char * str, char *ip);//获取本地IP地址

int main(int argc, char *argv[])
{

//用于测试命令行输入的值的情况
//printf("argv[0] = %s\n", argv[0]);
//printf("argv[1] = %s\n", argv[1]);
//printf("argv[2] = %s\n", argv[2]);
//printf("argv[3] = %s\n", argv[3]);

pid_t pid;
int sockfd, newfd;
socklen_t len;
struct sockaddr_in server_addr, client_addr;//结构体sockaddr_in
unsigned int server_port, lisnum;//用int也可以
char buf[MAXBUF + 1];

if(argv[1] && argv[2])//要根据argv[1]的情况来判断argv[2]的情况，否则会出错
server_port = atoi(argv[2]);
else
server_port = 6666;

if(argv[1] && argv[2] && argv[3])//与上面道理相同
lisnum = atoi(argv[3]);
else
lisnum = 5;

bzero(&server_addr, sizeof(server_addr));//也可以用memset
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(server_port);//转换为网络字节序

if(argv[1])
server_addr.sin_addr.s_addr = inet_addr(argv[1]);
else
{
char ip[128];
get_ip("eno16777736", ip);
server_addr.sin_addr.s_addr = inet_addr(ip);
}

//建立sockfd
if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
{
perror("socket");
exit(EXIT_FAILURE);
}

//输出ip和port信息，用于测试
printf("server_ip = %s\nserver_port = %d\nlisnum = %d\n", inet_ntoa(server_addr.sin_addr), server_port, lisnum);

//绑定sockfd和服务器的IP地址server_addr
if(bind(sockfd, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
{
perror("bind");
exit(EXIT_FAILURE);
}

//监听sockfd
if(listen(sockfd, lisnum) == -1)
{
perror("listen");
exit(EXIT_FAILURE);
}

printf("wait client connect ...\n");

//等待接收客户端的连接
//连接成功后，客户端地址信息存储在client_addr中
//新建立的socket描述符存储在newfd中
len = sizeof(struct sockaddr);
if((newfd = accept(sockfd, (struct sockaddr *)&client_addr, &len)) == -1)
{
perror("accept");
exit(EXIT_FAILURE);
}

//创建子进程
if((pid = fork()) == -1)
{
perror("fork");
exit(EXIT_FAILURE);
}
else if(pid == 0) //子进程用于向客户端发送消息
{
close(sockfd);
while(1)
{
bzero(buf, sizeof(buf));//清空buf
printf("pls input the message to send:  ");
fgets(buf, sizeof(buf)-1, stdin);//从终端接收输入

if(!strncasecmp(buf, "quit", 4))//判断是否为退出
{
printf("i will close the connect!\n");
break;
}

if(send(newfd, buf, strlen(buf)-1, 0) < 0 )//向客户端发送消息
{
printf("message '%s' send failure !\n", buf);
printf("errno code is %d, errno message is '%s'\n", errno, strerror(errno));
break;
}
}
close(newfd);

}
else //父进程用于从客户端接收消息
{
close(newfd);
while(1)
{
bzero(buf, sizeof(buf));
len = recv(newfd, buf, sizeof(buf)-1, 0);//从客户端接收消息
if(len > 0 )
printf("message recv successful : '%s', %d Byte recv\n", buf, len);
else if(len < 0)
{
printf("recv failure !\nerrno code is %d, errno message is '%s'\n", errno, strerror(errno));
break;
}
else//如果客户端已关闭
{
printf("the other one close quit\n");
break;
}
}
}

close(sockfd);

return 0;
}

void get_ip(char * str, char *ip)
{
int inet_sock;
struct ifreq ifr;
inet_sock = socket(AF_INET, SOCK_DGRAM, 0);
strcpy(ifr.ifr_name, str);//#include <net/if.h>
if (ioctl(inet_sock, SIOCGIFADDR, &ifr) < 0)//#include <sys/ioctl.h>
{
perror("ioctl");
exit(EXIT_FAILURE);
}
sprintf(ip,"%s", inet_ntoa(((struct sockaddr_in*)&(ifr.ifr_addr))->sin_addr));
close(inet_sock);
}

客户端代码

/*************************************************************************
> File Name: tcp_client.c
> Author: genglut
> Mail: genglut@163.com
> Created Time: 2014年12月22日 星期一 11时31分38秒
************************************************************************/

#include <stdio.h>
#include <string.h>
#include <errno.h>
#include <sys/socket.h>
#include <resolv.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>

#define MAXBUF 1024

int main(int argc, char *argv[])
{
pid_t pid;
int sockfd;
socklen_t len;
struct sockaddr_in server_addr;
char buf[MAXBUF + 1];

if(argc != 3)
{
printf("error failure, it must be:\n\t\t%s IP port \n", argv[0]);
exit(EXIT_FAILURE);
}

if((sockfd = socket(AF_INET, SOCK_STREAM, 0)) == -1)
{
perror("socket");
exit(EXIT_FAILURE);
}

bzero(&server_addr, sizeof(server_addr));
server_addr.sin_family = AF_INET;
server_addr.sin_port = htons(atoi(argv[2]));
server_addr.sin_addr.s_addr = inet_addr(argv[1]);

if(connect(sockfd, (struct sockaddr *)&server_addr, sizeof(struct sockaddr)) == -1)
{
perror("connect");
exit(EXIT_FAILURE);
}

printf("already connected to server %s\n", argv[1]);

if((pid = fork()) == -1)
{
perror("fork");
exit(EXIT_FAILURE);
}
else if(pid == 0)
{
while(1)
{
bzero(buf, sizeof(buf));
len = recv(sockfd, buf, sizeof(buf)-1, 0);
if(len >0)
printf("recv successful:'%s', %d byte recv\n", buf, len);
else
{
printf("the server close, quit\n");
break;
}
}
}
else
{
while(1)
{
bzero(buf, sizeof(buf));
printf("pls input message to send: ");
fgets(buf, sizeof(buf)-1, stdin);
if(!strncasecmp(buf, "quit", 4))
{
printf("i will quit!\n");
break;
}
len = send(sockfd, buf, strlen(buf)-1, 0);
if(len < 0)
{
perror("send");
break;
}
}
}

close(sockfd);
printf("i quited!\n");
return 0;
}

原文链接

/article/1574908.html