用写在C套接字编程中发送缓冲区

时间:2019-03-18 14:46:49

标签: c sockets posix system-calls

我正在使用fork()在C中创建一个TCP聊天室。每个客户端消息都应使用缓冲区中的用户名和消息到达服务器,因此我使用strcpy(buffer, name)strcat(buffer, ": ")来组合缓冲区和名称。之后,我使用

n = write (sockfd, buffer, strlen(buffer))

将缓冲区发送到服务器。所以在这里发生了奇怪的事情,我的服务器什么也没显示,但是在那之后,当我使用

n = read(sockfd, buffer, 255);
printf("%s", buffer);

字符串已正确传输。这是我的服务器代码:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>
#include <ctype.h>

int main(){
    int port;
    int i = 1;
    printf("Enter port ->");
    scanf("%d", &port);

    int sockfd, newsockfd, n;
    char buffer[255];
    struct sockaddr_in6 serv_addr, cli_addr;
    int ret, flag;

    sockfd = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP); 
    if (sockfd < 0){
        printf("Error opening socket\n");
    }

    bzero((char *) &serv_addr, sizeof(serv_addr));
    flag = 1;
    ret = setsockopt(sockfd, SOL_SOCKET, SO_REUSEADDR, &flag, sizeof(flag));
    if (ret < 0){
        printf("Set sock opt errror\n");
    }
    serv_addr.sin6_family = AF_INET6;
    serv_addr.sin6_addr = in6addr_any;
    serv_addr.sin6_port = htons(port);

    if (bind(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0){
        printf("Error binding\n");
    }

    listen(sockfd, 5);
    int clilen = sizeof(cli_addr);
    int pid;

    while(1){
        newsockfd = accept(sockfd, (struct sockaddr *) &cli_addr, &clilen);
        if (newsockfd < 0){
            printf("Error on accept\n");
        }
        printf("New connection\n");
        pid = fork();
        if (pid < 0){
            printf("Error on pid\n");
        }
        if (pid == 0){
            close(sockfd);
            while(1){
                bzero(buffer, 255);
                n = read(newsockfd, buffer, 255);
                if (n < 0){
                    printf("Error on reading\n");
                }
                printf("%s", buffer);
                n = write(newsockfd, buffer, strlen(buffer));
                if (n < 0){
                    printf("Error on writing\n");
                }
            }
        } 
    }
    close(newsockfd);
    return 0;
}

和客户代码:

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <unistd.h>

int main(){

    int port;
    printf("Enter port ->");
    scanf("%d", &port);
    char ip[10];
    printf("Enter ip -> ");
    scanf("%s", &ip);
    char name[20];
    printf("Enter name -> ");
    scanf("%s", &name);
    int sockfd, n;
    char buffer[255];
    struct sockaddr_in6 serv_addr;

    sockfd = socket(AF_INET6, SOCK_STREAM, IPPROTO_TCP); 
    if (sockfd < 0){
        printf("Error opening socket\n");
    }
    bzero((char *) &serv_addr, sizeof(serv_addr));
    serv_addr.sin6_family = AF_INET6;
    inet_pton(AF_INET6, "ip", &serv_addr.sin6_addr);
    serv_addr.sin6_port = htons(port);
    if (connect(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0){
        printf("Connection failed\n");
    }

    while(1){

        bzero(buffer, 255);
        fgets(buffer, 255, stdin);
        strcpy(buffer, name);
        strcat(buffer, ": ");
        n = write (sockfd, buffer, strlen(buffer));
        bzero(buffer,255);
        if (n < 0){
            printf("Error on writing\n");
        }
        n = read(sockfd, buffer, 255);
        printf("%s", buffer);
        if (n < 0){
            printf("Error on reading\n");
        }
    }

    close(sockfd);
}

所以也许有人可以看到为什么客户端无法将具有名称和消息的完整缓冲区发送给服务器的问题?我认为问题出在name上。当我仅发送使用

获得的缓冲区时
fgets(buffer, 255, stdin);

服务器工作正常,但我需要客户端用户名。

我尝试测试strlen(name)buffer,一切似乎还不错。我试图做的是手动将名称转换为这样的缓冲区:

 for (int i = 0; i < strlen(name); i++){
        buffer[i] = name[i];
 }

但是如果这是唯一的方法,那么我应该如何将缓冲区推入正确的输入名称而又不丢失消息信息?

1 个答案:

答案 0 :(得分:2)

您的代码有很多错误:)。

首先scanf将指针作为第一个参数,因此,当您想将其用于字符串时,请执行以下操作:

char ip[10];
...
scanf("%s", ip);//THIS IS CORRECT

不是scanf("%s", &ip);

在服务器上,编译器抱怨clilen的类型。在您将该变量声明为简单的accept时,sockelen_t期望有一个unsigned int(或等效的int)。


主要问题

是以下内容。在客户端中,您执行过strcpy(buffer, name);,该操作从缓冲区的开头开始复制name,但是缓冲区的开头实际上包含了您的数据。与下一行一样,您应该使用strcat进行以下操作:

    bzero(buffer, 255);
    fgets(buffer, 255, stdin);
    strcat(buffer, name);
    strcat(buffer, ": ");

其他一般建议

还要注意另一件事:避免将系统调用函数read write getchar与库函数(例如freadfwritefgets)混合使用。我遇到this problem的原因是fgets可能读取的字节数超出了预期。您的情况应该可以,因为您将它们用于不同的目的。