打算利用所学知识写一个基于UDP的聊天小项目,大致分为几个阶段:
1、网络通信模块封装;
2、客户端发送消息模块封装;
3、服务端接受消息模块封装;
服务端代码:
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <stdlib.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#define PORT 1234
#define MAXDATASIZE 100
int main(void)
{
int sockfd;
struct sockaddr_in server;
struct sockaddr_in client;
socklen_t len;
int num;
char buf[MAXDATASIZE];
if((sockfd = socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
perror("Creating socket failed.\n");
exit(1);
}
bzero(&server, sizeof(server));
server.sin_family = AF_INET;
server.sin_port = htons(PORT);
server.sin_addr.s_addr = htonl(INADDR_ANY);
if(bind(sockfd, (struct sockaddr *)&server, sizeof(server)) == -1)
{
perror("Bind() error.\n");
exit(1);
}
len = sizeof(client);
while(1)
{
num = recvfrom(sockfd, buf, MAXDATASIZE, 0, (struct sockaddr *)&client, &len);
if(num < 0)
{
perror("recvfrom() error.\n");
exit(1);
}
buf[num] = '\0';
printf("You got a message <%s> from client. \nIt's ip is %s, port is %d. \n", buf, inet_ntoa(client.sin_addr),htons(client.sin_port));
sendto(sockfd, "Welcome\n", 8, 0, (struct sockaddr *)&client, len);
if ( !strcmp(buf, "bye") ){
break;
}
}
close(sockfd);
return 0;
}
客户端代码:
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <arpa/inet.h>
#define PORT 1234
#define MAXDATASIZE 100
int main(int argc, char *argv[])
{
int sockfd, num;
char buf[MAXDATASIZE];
struct sockaddr_in server, peer;
if(argc != 3)
{
printf("Usage: %s <IP address> <message>\n", argv[0]);
exit(1);
}
if((sockfd=socket(AF_INET, SOCK_DGRAM, 0)) == -1)
{
printf("socket() error\n");
exit(1);
}
bzero(&server, sizeof(server));
server.sin_family = AF_INET;
server.sin_port = htons(PORT);
server.sin_addr.s_addr = inet_addr(argv[1]);
//server.sin_addr.s_addr = inet_addr(argv[1]);
if(connect(sockfd, (struct sockaddr *)&server, sizeof(server)) == -1)
{
printf("connect() error.\n");
exit(1);
}
send(sockfd, argv[2], strlen(argv[2]), 0);
while(1)
{
if((num = recv(sockfd, buf, MAXDATASIZE, 0)) == -1)
{
printf("recv() error.\n");
exit(1);
}
buf[num] = '\0';
printf("Server Message: %s.\n", buf);
break;
}
close(sockfd);
return 0;
}
代码来源于网络,若有侵权,请联系作者删除。
代码分析:
sockaddr_in(在netinet/in.h中定义):
struct sockaddr_in
{
short sin_family;/*Address family一般来说AF_INET(地址族)PF_INET(协议族)*/
unsigned short sin_port;/*Port number(必须要采用网络数据格式,普通数字可以用htons()函数转换成网络数据格式的数字)*/
struct in_addr sin_addr;/*IP address in network byte order(Internet address)*/
unsigned char sin_zero[8];/*Same size as struct sockaddr没有实际意义,只是为了 跟SOCKADDR结构在内存中对齐*/
};
另一种结构:
struct sockaddr
{
unsigned short sa_family; /* address family, AF_xxx */
char sa_data[14]; /* 14 bytes of protocol address */
};
两者之间的区别于联系:
用于存储参与(IP)Windows/linux套接字通信的计算机上的一个internet协议(IP)地址。为了统一地址结构的表示方法 ,统一接口函数,使得不同的地址结构可以被bind()、connect()、recvfrom()、sendto()等函数调用。但一般的编程中并不直接对此数据结构进行操作,而使用另一个与之等价的数据结构sockaddr_in。这是由于Microsoft TCP/IP套接字开发人员的工具箱仅支持internet地址字段,而实际填充字段的每一部分则遵循sockaddr_in数据结构,两者大小都是16字节,所以二者之间可以进行切换。
bind(sockfd, (struct sockaddr *) &serv, sizeof(serv));绑定套接字