简单实现Linux与Windows之间的UDP通信
如图所示:
在Linux下使用Makefile进行编译,Makefile代码如下:
CC = g++
SRCS = main.cpp udp.cpp
OBJS = $(SRCS:.cpp=.o)
EXEC = myapp
start:$(OBJS)
$(CC) -o $(EXEC) $(OBJS)
.cpp.o:
$(CC) -o $@ -c $< -DMYLINUX //这里在Linux下参加编译定义一个宏以区别
clean:
rm -rf $(OBJS)
由于Windows系统下使用UDP调用Windows特有的库函数,而在Linux下使用的却是Linux内嵌的udp相关函数,使用在各自系统编译的时候所用到的函数及代码会有一定的区别,并且C/C++混合编译在Windows下需要特别说明使用extern关键字,而在Linux下却不需要。所以通过定义一个宏在一套代码中进行区别,这样在不同平台下都能够进行编译使用。
具体代码如下:
main.cpp代码是没有区别的:
#include <iostream>
#include "udp.h"
using namespace std;
int main(int argc, char* argv[])
{
if(argc > 2)
{
Socket_send(argv[2]);
}
else
{
Socket_recv();
}
return 0;
}
接下来是udp.h头文件,这里就能看出区分:
#ifndef UDP_H
#define UDP_H
#ifndef MY_LINUX
extern "C"
{
#endif
int Socket_send(char *ip);//udp发送
int Socket_recv();//接收端
#ifndef MY_LINUX
}
#endif
#endif // UDP_H
最后就是udp.c源文件代码:
#include <stdio.h>
#ifdef MY_LINUX //Linux下编译需要包含以下头文件,具体通过man可查到
#include <sys/types.h>
#include <sys/socket.h>
#include <string.h>
#include <unistd.h>
#include <arpa/inet.h>
#define SOCKET int //在Linux下SOCKET数据类型就是int类型,所以在这里替换
#else
#include <winsock2.h>
#endif
#ifndef MY_LINUX
int Socket_send(char *ip)
{
size_t vc = 0;
//初始化socket
char buf[1024] = {0};//传输使用的字符串
DWORD ver;
WSADATA wsaData;
ver = MAKEWORD(1,1);//调用wsastartup时告诉Windows使用的sock版本
WSAStartup(ver, &wsaData);
//创建socket结构体
SOCKET st = socket(AF_INET, SOCK_DGRAM, 0);//建立一个socket
//创建结构体
struct sockaddr_in addr;
memset(&addr, 0, sizeof(addr));//初始化结构体
addr.sin_family = AF_INET;//代表使用的是TCP/IP的地址
addr.sin_port = htons(8080);//设置使用的端口号
addr.sin_addr.s_addr = inet_addr(ip);//自检查网口IP
while(1)
{
memset(buf, 0, sizeof(buf));
gets(buf);
if(buf[0] == '0')
{//循环退出条件
break;
}
vc = sendto(st, buf, strlen(buf), 0, (struct sockaddr *)&addr, sizeof(addr));//发送函数
}
closesocket(st);//使用完socket要将其关闭
WSACleanup();//释放win下socket内部的相关资源
return vc;
}
int Socket_recv()
{
//初始化socket
char buf[1024] = {0};//传输使用的字符串
size_t rc = 0;
DWORD ver;
WSADATA wsaData;
ver = MAKEWORD(1,1);//调用wsastartup时告诉Windows使用的sock版本
WSAStartup(ver, &wsaData);
//创建socket结构体
SOCKET st = socket(AF_INET, SOCK_DGRAM, 0);//建立一个socket
//创建结构体
struct sockaddr_in recvaddr;
memset(&recvaddr, 0, sizeof(recvaddr));//初始化结构体
recvaddr.sin_family = AF_INET;//代表使用的是TCP/IP的地址
recvaddr.sin_port = htons(8080);//设置使用的端口号
recvaddr.sin_addr.s_addr = htonl(INADDR_ANY);//接收端采用任意的IP地址
if(bind(st, (struct sockaddr *)&recvaddr, sizeof(recvaddr)) > -1)
{
struct sockaddr_in sendaddr;
memset(&sendaddr, 0, sizeof(sendaddr));
int len = sizeof(sendaddr);
while(1)
{
memset(buf, 0, sizeof(buf));
rc = recvfrom(st, buf, sizeof(buf), 0, (struct sockaddr *)&sendaddr, &len);
//rc = recv(st, buf, sizeof(buf), 0);
printf("recvive = %s\n", buf);
}
// printf("IP = %u\n", sendaddr.sin_addr.s_addr);
}
closesocket(st);//使用完socket要将其关闭
WSACleanup();//释放win下socket内部的相关资源
return rc;
}
#else //编译Linux环境下的函数实现
int Socket_send(char *ip)
{
size_t vc = 0;
//初始化socket
char buf[1024] = {0};//传输使用的字符串
//创建socket结构体
SOCKET st = socket(AF_INET, SOCK_DGRAM, 0);//建立一个socket
//创建结构体
struct sockaddr_in addr;
memset(&addr, 0, sizeof(addr));//初始化结构体
addr.sin_family = AF_INET;//代表使用的是TCP/IP的地址
addr.sin_port = htons(8080);//设置使用的端口号
addr.sin_addr.s_addr = inet_addr(ip);//自检查网口IP
while(1)
{
memset(buf, 0, sizeof(buf));
gets(buf);
if(buf[0] == '0')
{//循环退出条件
break;
}
vc = sendto(st, buf, strlen(buf), 0, (struct sockaddr *)&addr, sizeof(addr));//发送函数
}
close(st);//使用完socket要将其关闭
return vc;
}
int Socket_recv()
{
//初始化socket
char buf[1024] = {0};//传输使用的字符串
size_t rc = 0;
//创建socket结构体
SOCKET st = socket(AF_INET, SOCK_DGRAM, 0);//建立一个socket
//创建结构体
struct sockaddr_in recvaddr;
memset(&recvaddr, 0, sizeof(recvaddr));//初始化结构体
recvaddr.sin_family = AF_INET;//代表使用的是TCP/IP的地址
recvaddr.sin_port = htons(8080);//设置使用的端口号
recvaddr.sin_addr.s_addr = htonl(INADDR_ANY);//接收端采用任意的IP地址
if(bind(st, (struct sockaddr *)&recvaddr, sizeof(recvaddr)) > -1)
{
struct sockaddr_in sendaddr;
memset(&sendaddr, 0, sizeof(sendaddr));
socklen_t len = sizeof(sendaddr);
while(1)
{
memset(buf, 0, sizeof(buf));
rc = recvfrom(st, buf, sizeof(buf), 0, (struct sockaddr *)&sendaddr, &len);
//rc = recv(st, buf, sizeof(buf), 0);
printf("recvive = %s\n", buf);
}
// printf("IP = %u\n", sendaddr.sin_addr.s_addr);
}
close(st);//使用完socket要将其关闭
return rc;
}
#endif
最后在Linux下通过make命令就能得到执行程序:
以上就是简单的Windows下UDP实现代码到Linux系统下的移植过程咯。