VxWorks中网络编程实现TCP协议的Socket通信

本示例代码演示了在VxWorks系统中通过使用Socket API实现TCP协议的网络编程，包括建立连接、收发数据等基本操作，为后续网络编程开发提供基础。

#include <stdio.h>
#include <string.h>
#include <ioLib.h>
#include <sockLib.h>
#include <inetLib.h>

#define SERVER_PORT_NUM 1234
#define SERVER_HOST_ADDR "192.168.1.2"
#define MAX_BUF_SIZE 256

void tcpServer()
{
    int sockFd, newSockFd, sockAddrSize, recvLen, sendLen;
    struct sockaddr_in serverAddr, clientAddr;
    char recvBuf[MAX_BUF_SIZE];
    char sendBuf[MAX_BUF_SIZE];

    // 创建Socket
    if ((sockFd = socket(AF_INET, SOCK_STREAM, 0)) == ERROR)
    {
        perror("socket");
        return;
    }

    // 设置服务器地址信息
    bzero((char *)&serverAddr, sizeof(serverAddr));
    serverAddr.sin_family = AF_INET;
    serverAddr.sin_addr.s_addr = inet_addr(SERVER_HOST_ADDR);
    serverAddr.sin_port = htons(SERVER_PORT_NUM);

    // 绑定Socket
    if (bind(sockFd, (struct sockaddr *)&serverAddr, sizeof(serverAddr)) == ERROR)
    {
        perror("bind");
        close(sockFd);
        return;
    }

    // 监听连接请求
    if (listen(sockFd, 5) == ERROR)
    {
        perror("listen");
        close(sockFd);
        return;
    }

    sockAddrSize = sizeof(clientAddr);

    while (1)
    {
        // 接受连接请求
        if ((newSockFd = accept(sockFd, (struct sockaddr *)&clientAddr, &sockAddrSize)) == ERROR)
        {
            perror("accept");
            close(sockFd);
            return;
        }

        printf("Connection from %s, port %d\n", inet_ntoa(clientAddr.sin_addr), ntohs(clientAddr.sin_port));

        // 收取数据
        if ((recvLen = recv(newSockFd, recvBuf, sizeof(recvBuf), 0)) == ERROR)
        {
            perror("recv");
            close(newSockFd);
            continue;
        }
        recvBuf[recvLen] = '\0';
        printf("Received %d bytes: %s\n", recvLen, recvBuf);

        // 发送数据
        snprintf(sendBuf, sizeof(sendBuf), "Hello, %s", recvBuf);
        sendLen = strlen(sendBuf);
        if ((send(newSockFd, sendBuf, sendLen, 0)) != sendLen)
        {
            perror("send");
        }

        // 关闭Socket
        close(newSockFd);
    }
}

void tcpClient()
{
    int sockFd, sockAddrSize, connectStatus, recvLen, sendLen;
    struct sockaddr_in serverAddr;
    char recvBuf[MAX_BUF_SIZE];
    char sendBuf[MAX_BUF_SIZE];
    char *serverHostName = "192.168.1.2";

    // 创建Socket
    if ((sockFd = socket(AF_INET, SOCK_STREAM, 0)) == ERROR)
    {
        perror("socket");
        return;
    }

    // 设置服务器地址信息
    bzero((char *)&serverAddr, sizeof(serverAddr));
    serverAddr.sin_family = AF_INET;
    serverAddr.sin_addr.s_addr = inet_addr(serverHostName);
    serverAddr.sin_port = htons(SERVER_PORT_NUM);

    // 连接服务器
    if ((connectStatus = connect(sockFd, (struct sockaddr *)&serverAddr, sizeof(serverAddr))) == ERROR)
    {
        perror("connect");
        close(sockFd);
        return;
    }

    // 发送数据
    printf("Enter message to send: ");
    fgets(sendBuf, MAX_BUF_SIZE, stdin);
    sendLen = strlen(sendBuf);
    if ((send(sockFd, sendBuf, sendLen, 0)) != sendLen)
    {
        perror("send");
        close(sockFd);
        return;
    }

    // 接收数据
    if ((recvLen = recv(sockFd, recvBuf, sizeof(recvBuf), 0)) == ERROR)
    {
        perror("recv");
        close(sockFd);
        return;
    }
    recvBuf[recvLen] = '\0';
    printf("%s\n", recvBuf);

    // 关闭Socket
    close(sockFd);
}

void testTcp()
{
    int choice;

    while (1)
    {
        printf("1. Server\n");
        printf("2. Client\n");
        printf("0. Quit\n");
        printf("Enter choice: ");
        scanf("%d", &choice);
        getchar();

        switch (choice)
        {
        case 1:
            tcpServer();
            break;

        case 2:
            tcpClient();
            break;

        case 0:
            return;

        default:
            printf("Invalid choice\n");
        }
    }
}

免责申明：文章和图片全部来源于公开网络，如有侵权，请通知删除 server@dude6.com