sender.c

udp编程

// 模块名: Sender.c
//
// 描述:
//    本例向指定的接收方发送UDP数据报，-c选项调用connect()函数将接收方的IP地
//    址和套接字句柄相关联，使用有连接的通信模式，这样就能够调用send()函数来
//    发送数据；否则，在更通常的情况下，使用无连接的通信模式，这时需要调用
//    sendto()函数来发送数据
//
// 编译方法:
//    cl -o Sender Sender.c ws2_32.lib
//
// 命令行选项:
//    sender [-p:int] [-r:IP] [-c] [-n:x] [-b:x] [-d:c]
//           -p:int   远程端口
//           -r:IP    接收方的IP地址或主机名
//           -c       首先连接时的远程IP地址
//           -n:x     发送消息的次数
//           -b:x     发送缓冲区的大小
//           -d:c     填充缓冲区的字符
//
#include <winsock2.h>
#include <stdio.h>
#include <stdlib.h>

#define DEFAULT_PORT            5150
#define DEFAULT_COUNT           25
#define DEFAULT_CHAR            'a'
#define DEFAULT_BUFFER_LENGTH   64

BOOL  bConnect = FALSE;                 // 首先连接到接收方，采用有连接模式
int   iPort    = DEFAULT_PORT;          // 发送数据的端口
char  cChar    = DEFAULT_CHAR;          // 填充缓冲区的字符
DWORD dwCount  = DEFAULT_COUNT,         // 发送的消息数目
      dwLength = DEFAULT_BUFFER_LENGTH; // 发送缓冲区的长度
char  szRecipient[128];                 // 接收方的IP地址或主机名

//
// 函数: usage
//
// 描述:
//    打印使用方法并退出程序
//
void usage()
{
    printf("usage: sender [-p:int] [-r:IP] "
		   "[-c] [-n:x] [-b:x] [-d:c]\n\n");
    printf("       -p:int   远程端口\n");
    printf("       -r:IP    接收方的IP地址或主机名\n");
    printf("       -c       首先连接时接收方的IP地址\n");
    printf("       -n:x     发送的消息数目\n");
    printf("       -b:x     发送缓冲区的长度\n");
    printf("       -d:c     填充缓冲区的字符\n\n");
    ExitProcess(1);
}

//
// 函数: ValidateArgs
//
// 描述:
//    解析命令行参数，设置一些全局的标志，以便指示要完成的动作
//
void ValidateArgs(int argc, char **argv)
{
    int i;

    for(i = 1; i < argc; i++)
    {
        if ((argv[i][0] == '-') || (argv[i][0] == '/'))
        {
            switch (tolower(argv[i][1]))
            {
                case 'p':        // 本地端口
                    if (strlen(argv[i]) > 3)iPort = atoi(&argv[i][3]);
                    break;
                case 'r':        // 接收方的IP地址
                    if (strlen(argv[i]) > 3)strcpy(szRecipient, &argv[i][3]);
                    break;
                case 'c':        // 连接到接收方的IP地址
                    bConnect = TRUE;
                    break;
                case 'n':        // 发送消息的次数
                    if (strlen(argv[i]) > 3)dwCount = atol(&argv[i][3]);
                    break;
                case 'b':        // 缓冲区尺寸
                    if (strlen(argv[i]) > 3)dwLength = atol(&argv[i][3]);
                    break;
                case 'd':		//填充缓冲区的字符
                    cChar = argv[i][3];
                    break;
                default:
                    usage();
                    break;
            }
        }
    }
}

//
// 主函数: main
//
// 描述:
// 描述:
//    执行主线程，初始化Winsock，解析命令行参数，创建套接字，将套接字
//    与本地网卡和端口绑定, 然后想接收方发送数据报
//
int main(int argc, char **argv)
{
    WSADATA        wsd;
    SOCKET         s;
    char          *sendbuf = NULL;
    int            ret,
                   i;
    SOCKADDR_IN    recipient;

    // 解析命令行参数，装载Winsock
    //
    ValidateArgs(argc, argv);

    if (WSAStartup(MAKEWORD(2, 2), &wsd) != 0)
    {
        printf("加载Winsock失败!\n");
        return 1;
    }
    // 创建socket
    //
    s = socket(AF_INET, SOCK_DGRAM, 0);
    if (s == INVALID_SOCKET)
    {
        printf("创建socket失败; %d\n", WSAGetLastError());
        return 1;
    }
    // 解析接收方的IP地址或主机名
    //
    recipient.sin_family = AF_INET;
    recipient.sin_port = htons((short)iPort);
    if ((recipient.sin_addr.s_addr = inet_addr(szRecipient))== INADDR_NONE)
    {
        struct hostent *host=NULL;

        host = gethostbyname(szRecipient);
        if (host)
            CopyMemory(&recipient.sin_addr, host->h_addr_list[0],host->h_length);
        else
        {
            printf("gethostbyname() failed: %d\n", WSAGetLastError());
            WSACleanup();
            return 1;
        }
    }
    // 分配发送缓冲区
    //
    sendbuf = GlobalAlloc(GMEM_FIXED, dwLength);
    if (!sendbuf)
    {
        printf("分配缓冲区失败: %d\n", GetLastError());
        return 1;
    }
    memset(sendbuf, cChar, dwLength);
    //
    // 如果设置了连接选项,"connect"到接收方，并且调用send()函数发送数据
    //
    if (bConnect)
    {
        if (connect(s, (SOCKADDR *)&recipient,sizeof(recipient)) == SOCKET_ERROR)
        {
            printf("连接失败: %d\n", WSAGetLastError());
            GlobalFree(sendbuf);
            WSACleanup();
            return 1;
        }
        for(i = 0; i < dwCount; i++)
        {
            ret = send(s, sendbuf, dwLength, 0);
            if (ret == SOCKET_ERROR)
            {
                printf("发送数据失败: %d\n", WSAGetLastError());
                break;
            }
            else if (ret == 0)
                break;
            // 发送数据成功!
        }
    }
    else
    {
        // 否则，在无连接模式，使用sendto()函数发送数据
        //
        for(i = 0; i < dwCount; i++)
        {
            ret = sendto(s, sendbuf, dwLength, 0,(SOCKADDR *)&recipient, sizeof(recipient));
            if (ret == SOCKET_ERROR)
            {
                printf("发送数据失败; %d\n", WSAGetLastError());
                break;
            }
            else if (ret == 0)
                break;
            // 发送数据成功!
        }
    }
    closesocket(s);

    GlobalFree(sendbuf);
    WSACleanup();
    return 0;
}