服务器:server
#include<stdio.h>
#include<stdlib.h>
#include<pthread.h>
#include<string.h>
#include<sys/socket.h>
#include <sys/epoll.h>
#include <arpa/inet.h>
#define DEFAULT_HEADBEAT_PORT 7090
#define RECV_BUF_SIZE 1024
int epfd_heatbeat = 0;
int heatBeatFd = 0;
int headBeatUdpSocketInit(){
int ret = 0;
int len = 0 , size = 0;
struct epoll_event event;
struct sockaddr_in heatbeadAddr;
epfd_heatbeat = epoll_create(1);
if(epfd_heatbeat == -1){
printf("[headBeatRecv] Epoll Init Failed.\n");
return -1;
}
if ((heatBeatFd = socket(AF_INET, SOCK_DGRAM, 0)) < 0)
{
return -1;
}
memset(&heatbeadAddr, 0, sizeof(heatbeadAddr));
heatbeadAddr.sin_family = AF_INET;
heatbeadAddr.sin_addr.s_addr = htonl(INADDR_ANY); //自动获取IP地址
heatbeadAddr.sin_port = htons(DEFAULT_HEADBEAT_PORT);
if (bind(heatBeatFd, (struct sockaddr *)&heatbeadAddr, sizeof(heatbeadAddr)) < 0)
{
close(heatBeatFd);
return -1;
}
memset(&event, 0, sizeof(event));
if(heatBeatFd > 0){
event.data.fd = heatBeatFd;
event.events = EPOLLIN;
ret = epoll_ctl(epfd_heatbeat, EPOLL_CTL_ADD, heatBeatFd, &event);
if (ret < 0)
{
close(heatBeatFd);
heatBeatFd = -1;
return -1;
}
}
len = sizeof(size);
if (getsockopt(heatBeatFd, SOL_SOCKET, SO_RCVBUF, (void *) &size, &len) == 0)
{
size *= 4;
setsockopt(heatBeatFd, SOL_SOCKET, SO_RCVBUF, (void *) &size, sizeof (size));
}
else
{
}
if (getsockopt(heatBeatFd, SOL_SOCKET, SO_SNDBUF, (void *) &size, &len) == 0)
{
size *= 4;
setsockopt(heatBeatFd, SOL_SOCKET, SO_SNDBUF, (void *) &size, sizeof (size));
}
else
{
}
return 0;
}
void *_headBeatRecvThread(void *data){
int ret = 0, res = 0;
int readfd = 0;
int fdnum = 0;
int recvfd = 0;
int client_addr_len = 0;
struct sockaddr_in clientaddr;
struct epoll_event heatBeatEventArray[1];
char recv_buf[RECV_BUF_SIZE] = {'\0'};
ret = headBeatUdpSocketInit();
if(ret != 0){
pthread_exit(NULL);
}
while(1){
if(heatBeatFd > 0){
readfd = epoll_wait(epfd_heatbeat, heatBeatEventArray, 1, 1000);
if(readfd > 0){
for(fdnum = 0; fdnum < readfd; fdnum++){
bzero(recv_buf, sizeof(recv_buf));
memset(&clientaddr, 0, sizeof(clientaddr));
client_addr_len = sizeof(clientaddr);
recvfd = heatBeatEventArray[fdnum].data.fd;
if(recvfd = heatBeatFd){
if(heatBeatEventArray[fdnum].events & EPOLLIN){
res = recvfrom(heatBeatFd, recv_buf, RECV_BUF_SIZE - 1, 0, (struct sockaddr *)&clientaddr, &client_addr_len);
if (res == -1)
{
}else{
printf("recv value:%s\n",recv_buf);
sendto(heatBeatFd, recv_buf, strlen(recv_buf), 0, (struct sockaddr*)&clientaddr, sizeof(clientaddr));
}
}
}
}
}
}
}
return NULL;
}
int main(){
pthread_t _sipHeatBeatThread = -1;
pthread_attr_t _sipHeatBeatThreadAttr;
pthread_attr_init(&_sipHeatBeatThreadAttr);
pthread_attr_setdetachstate(&_sipHeatBeatThreadAttr, PTHREAD_CREATE_DETACHED);
if(pthread_create(&_sipHeatBeatThread, &_sipHeatBeatThreadAttr, _headBeatRecvThread, NULL) < 0){
printf("pthread create faild\n");
return -1;
}
while(1){
sleep(5);
}
return 0;
}
多线程:编译时加gcc -lpthread
客户端:client
#include<stdio.h>
#include<stdlib.h>
#include<pthread.h>
#include<string.h>
#include<sys/socket.h>
#include<sys/epoll.h>
#include<arpa/inet.h>
#include<time.h>
int main(){
char buffer[]="abc";
char recv_buf[1024] = {'\0'};
int client_addr_len = 0;
int fd = socket(AF_INET, SOCK_DGRAM, 0);
//ssize_t sendto(int sock, void *buf, size_t nbytes, int flags, struct sockaddr *to, socklen_t addrlen);
struct sockaddr_in servAddr;
struct sockaddr_in clientaddr;
memset(&servAddr, 0, sizeof(servAddr));
memset(&clientaddr, 0, sizeof(clientaddr));
servAddr.sin_family = AF_INET;
servAddr.sin_addr.s_addr = inet_addr("127.0.0.1");
servAddr.sin_port = htons(7090);
while(1){
//bzero(buffer, sizeof(buffer));
//printf("输入字符串:");
//scanf("%s",buffer);
bzero(recv_buf, sizeof(recv_buf));
sendto(fd, buffer, strlen(buffer), 0, (struct sockaddr*)&servAddr, sizeof(servAddr));
recvfrom(fd, recv_buf, 1024 - 1, 0, (struct sockaddr *)&clientaddr, &client_addr_len);
printf("recv:%s\n", recv_buf);
sleep(1);
}
close(fd);
return 0;
}
UDP服务器和客户端的编写(原创)(linux系统)
最新推荐文章于 2024-04-29 17:46:09 发布