Linux与Windows的UDP通讯代码实现

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)

#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;
}

#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

#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
", buf);
        }
       // printf("IP = %u
", 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
", buf);
        }
       // printf("IP = %u
", sendaddr.sin_addr.s_addr);
    }
    close(st);//使用完socket要将其关闭
    return rc;
}
#endif