服务器端:
/*
* File Name: socket_udp_server.c
* Author: eagle
* mail: zhaocheng121@163.com
* Created Time: 2015-10-08 18:43
*/
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>
char Name[100][40];
int findclient(struct sockaddr_in client_address, struct sockaddr_in client[], int *client_num, char name[]);
int main(int argc, char *argv[])
{
int server_socketfd, client_socketfd;
int server_len,client_len;
struct sockaddr_in server_address;
struct sockaddr_in client_address;
struct sockaddr_in client[100];//记录客户端信息
int i, byte, client_num = 0;
char recv[100], send[100];
int flag;
<pre name="code" class="cpp"> /*hostip 本机ip地址*/
int inet_sock;
struct ifreq ifr;
char hostip[15];
inet_sock = socket(AF_INET, SOCK_DGRAM,0);
strcpy(ifr.ifr_name, "wlan0");
if (ioctl(inet_sock, SIOCGIFADDR, &ifr) < 0)
perror("ioctl");
strcpy(hostip, inet_ntoa(((struct sockaddr_in *)&(ifr.ifr_addr))->sin_addr));
/*上面部分为获取本机ip*/
/*create socket and bind*/
server_socketfd = socket(AF_INET, SOCK_DGRAM, 0); //创建socket对象
server_address.sin_family = AF_INET;
server_address.sin_addr.s_addr = inet_addr("192.168.99.184"); //主机ip
server_address.sin_port = 9734; //主机端口
server_len = sizeof(server_address);
bind(server_socketfd, (struct sockaddr *) & server_address, server_len);
client_len = sizeof(struct sockaddr_in);
memset(&client_address, 0, sizeof(struct sockaddr_in));
while(1){
if((byte = recvfrom(server_socketfd, &recv, sizeof(recv), 0, //从客户端接受消息
(struct sockaddr *)&client_address, //存储客户端地址信息
(socklen_t *)&client_len)) == -1)
{
perror("recvfrom");
exit(EXIT_FAILURE);
}
flag = findclient(client_address, client, &client_num, recv);
// printf("%d\n",flag);
if(flag == -1) {
strcpy(send, "Welcome to chating room");
if((byte = sendto(server_socketfd, &send, sizeof(send), 0,
(struct sockaddr *)&client_address,
(socklen_t)client_len)) == -1)
{
perror("send");
exit(EXIT_FAILURE);
}
} //给刚加入的客户端发送确认信息
//printf("recv:%s\n", recv);
//printf("send:");
//scanf("%s", send);
else {
strcpy(send, Name[flag]);
strcat(send, ":");
strcat(send, recv);
for(i = 0; i < client_num; i++){
// if(client[i].sin_port != client_address.sin_port) {
if((client[i].sin_addr.s_addr != client_address.sin_addr.s_addr) || (client[i].sin_port != client_address.sin_port)) {
client_len = sizeof(client[i]);
if((byte = sendto(server_socketfd, &send, sizeof(send), 0,
(struct sockaddr *)&client[i],
(socklen_t)client_len)) == -1)
{
perror("send");
exit(EXIT_FAILURE);
}
}
}
}
}
// shutdown(client_socketfd, 2);
// shutdown(server_socketfd, 2);
return 0;
}
int findclient(struct sockaddr_in client_address, struct sockaddr_in client[], int *client_num, char name[])
{
int i;
if((*client_num) == 0) {
// client[0].sin_family = client_address.sin_family;
// client[0].sin_port = client_address.sin_port;
client[0] = client_address;
strcpy(Name[0], name);
(*client_num)++;
return -1; //没有客户端信息
}
else {
for(i = 0; i < (*client_num); i++) {
if(client[i].sin_addr.s_addr == client_address.sin_addr.s_addr && client[i].sin_port == client_address.sin_port)
return i; //列表中有客户端的信息
}
client[i] = client_address;
strcpy(Name[i], name);
(*client_num)++;
return -1; //没有客户信息
}
}
<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);">客户端:(本机测试时需将hostip改为127.0.0.1</span>
/*
* File Name: socket_udp_server.c
* Author: eagle
* mail: zhaocheng121@163.com
* Created Time: 2015-10-08 18:43
*/
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <net/if.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/ioctl.h>
char name[40];
void first_message(int socketfd, struct sockaddr_in server_address, int server_len);
int main(int argc, char *argv[])
{
int socketfd;
int server_len,client_len;
struct sockaddr_in server_address;
struct sockaddr_in client_address;
int i,byte;
char recv[100], send[100];
int port;
srand(time(0));
port = 9000 + rand() % 1000;
//char recv, send;
/*hostip 本机ip地址*/
int inet_sock;
struct ifreq ifr;
char hostip[15];
inet_sock = socket(AF_INET, SOCK_DGRAM,0);
strcpy(ifr.ifr_name, "wlan0");
if (ioctl(inet_sock, SIOCGIFADDR, &ifr) < 0)
perror("ioctl");
strcpy(hostip, inet_ntoa(((struct sockaddr_in *)&(ifr.ifr_addr))->sin_addr));
/*上面部分为获取本机ip*/
socketfd = socket(AF_INET, SOCK_DGRAM, 0); //创建socket对象
server_address.sin_family = AF_INET;
server_address.sin_addr.s_addr = inet_addr("192.168.99.184"); //主机ip(需要自己设置)
server_address.sin_port = 9734; //主机端口
server_len = sizeof(server_address);
client_address.sin_family = AF_INET;
client_address.sin_addr.s_addr = inet_addr(hostip);
client_address.sin_port = htons(port);
client_len = sizeof(client_address);
bind(socketfd, (struct sockaddr *) & client_address, client_len);
/*第一次连接时向服务器发送识别信息*/
first_message(socketfd, server_address, server_len);
if(fork()==0) {
while(1) {
// printf("send:");
//scanf("%s", send);
gets(send);
if((byte = sendto(socketfd, &send, sizeof(send), 0,
(struct sockaddr *)&server_address,
(socklen_t)server_len)) == -1)
{
perror("send");
exit(EXIT_FAILURE);
}
}
}
else {
while(1) {
if((byte = recvfrom(socketfd, &recv, sizeof(recv), 0, //从客户端接受消息
(struct sockaddr *)&server_address, //存储客户端地址信息
(socklen_t *)&server_len)) == -1)
{
perror("recvprom");
exit(EXIT_FAILURE);
}
printf("recv:");
puts(recv);
printf("\n");
}
}
shutdown(socketfd, 2);
return 0;
}
void first_message(int socketfd, struct sockaddr_in server_address, int server_len)
{
char name[40];
//struct message mess;
printf("please input your name: ");
gets(name);
//puts(mess.ID);
if((sendto(socketfd, &name, sizeof(name), 0,
(struct sockaddr *)&server_address,
(socklen_t)server_len)) == -1)
{
perror("send");
exit(EXIT_FAILURE);
}
}