C程序-服务器函数错误

Question

我正在尝试学习 C，但无法让这些应用程序运行。我正在创建 2 个应用程序客户端/服务器，其中客户端通过指定端口连接到服务器，并向服务器发送文件名(文本)。然后，服务器获取文件名，通过字数统计功能运行它，然后用文件名/行/字/字符数响应客户端。我已经让客户端和服务器进行连接和通信，但我的问题是我似乎无法正确地将客户端输入传递给字数统计函数。另外，我不确定我的函数是否会正确返回，因为我还没有找到返回精心设计的字符串的适当方法。任何帮助，将不胜感激。谢谢!

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

void error(const char *msg){
    perror(msg);
    exit(1);
}

char * wordcount(char *cfile){
    int i = 0;
    int ccount = 0;
    int wcount = 0;
    int lcount = 0;
    char *fn = malloc(strlen(cfile+7));
    sprintf(fn, "/Files/%s", cfile);
    FILE *cfilename = fopen (fn, "r");
    while ((i = fgetc(cfilename)) != EOF){
        if (i == '\n') {
            lcount++;
        }
        if (i == '\t' || i == ' '){
            wcount++;
        }
        ccount++;
    }
    printf("%c contains %d words, %d characters and %d lines.\n", cfile, wcount, ccount, lcount);
    return 0;
}

int main(int argc, char *argv[]){
    int sock, newSock, portno, n;
    struct sockaddr_in serv_addr, cli_addr;
    socklen_t clilen;
    char buffer[256];  
    int index = 5;
    int lowPortNum = 2500 + (10 * index);
    int highPortNum = 2500 + (10 * index) + 9;  
/* Check for proper amount of args */
     if (argc < 2){
         fprintf(stderr, "ERROR: No port specified. Exiting...\n");
         printf("NOTE: Port must be between %d & %d.\n",lowPortNum,highPortNum);
         exit(1);
     } 
/* Create socket by using args to form components */
    sock = socket(AF_INET , SOCK_STREAM , 0);
    if (sock < 0){
        error("ERROR: Could not create socket");
    }
    bzero((char *) &serv_addr, sizeof(serv_addr));
    portno = atoi(argv[1]);
    serv_addr.sin_family = AF_INET;
    serv_addr.sin_addr.s_addr = INADDR_ANY;
    serv_addr.sin_port = htons(portno);
    if(bind(sock, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0){
        error("ERROR: Binding failed.");
        return 1;
    }
/* Start listening on socket */
    listen(sock, 5);
    puts("Server is waiting for connection...");
    clilen = sizeof(cli_addr);
/* Accept connection from incoming client */
    newSock = accept(sock, (struct sockaddr *) &cli_addr, &clilen);
    if (newSock < 0){
        error("ERROR: Accept failed.");
    }
/* Read message from the client */
        bzero(buffer, 256);
        if (read(newSock, buffer, 255) < 0){
            error("ERROR: Cannot read from socket.");
        }
//debug 
        printf("client msg read: %s\n",buffer);
/* Send message to the client */
//wordcount(buffer);
        if (write(newSock , wordcount(buffer) , 100) < 0){
            error("ERROR: Cannot write to socket.");
        }
    close(newSock);
    close(sock);
    return 0;
}

Answer 1

首先，我认为你在这里犯了一个错误:

char *fn = malloc(strlen(cfile+7));
sprintf(fn, "/Files/%s", cfile);

你的意思可能是这样的:

char fn[strlen(cfile) + 8];
sprintf(fn, "/Files/%s", cfile);

你可能会注意到我稍微重新安排了它；我认为您想将 7 添加到 strlen 的返回值中，而不是添加到 strlen 的参数中。我写了 8，因为额外的 1 是用于位于字符串末尾的 '\0'；这非常重要。当您制作字符串时，始终记住为 '\0' 留出空间。

此外，我已将您的 malloc 更改为可变长度数组。您确实不需要为此使用动态存储持续时间(例如 malloc)； 尝试选择自动存储持续时间，除非您绝对需要动态存储持续时间。

特别是考虑到您的代码会泄漏内存(这是避免它的确切原因，除非您绝对需要它)。也许 valgrind 在您的开发环境中会是一个有用的工具？ 始终记住释放您malloc所拥有的任何内存。

<小时/>

这里还有另一个错误:

FILE *cfilename = fopen (fn, "r");
while ((i = fgetc(cfilename)) != EOF){

假设fopen返回NULL(可能表明该文件不存在)，那么对它后面的fgetc的调用显然会进行以灾难性的方式失败。我认为你的意思是这样的:

FILE *cfilename = fopen (fn, "r");
if (cfilename == NULL) {
    /* XXX: HANDLE THIS ERROR! We'll get to this later... */
}
while ((i = fgetc(cfilename)) != EOF){

...同样，您忘记fclose那个fopend的文件。 始终记住fclose您fopend的文件。

<小时/>

... my problem is that I can't seem to pass the client input to the wordcount function properly

如果前面提到的错误得到修复，您应该能够像评论中那样安全地将从套接字接收到的消息传递到 wordcount，而不会发生崩溃或资源泄漏:wordcount (缓冲区)；...

如果您愿意，可以尝试一下，但请再耐心等待一下，因为您还有其他要求需要评估。

<小时/>

I'm not sure my function will return properly as I haven't found an appropriate method of returning a crafted string.

考虑一下标准库函数如何处理这个问题。您在这里使用了其中之一:sprintf(fn, "/Files/%s", cfile);。通过接受字符串的目标 (fn) 作为参数，sprintf 允许您使用您喜欢的任何存储持续时间。此外，这允许 sprintf 返回一些其他的 int 值(您可以在自己的时间在 the sprintf manual 中查找)...

如果您将函数设计为写入参数指向的目的地，就像 sprintf(和其他)那样，您将能够按照您喜欢的方式使用您的函数(例如自动< em>或动态存储持续时间？)。您还可以返回一个 int 值来指示成功或失败(例如，当 fopen 调用失败时)。

考虑以下函数，它甚至不需要返回值，因为没有错误模式:

typedef unsigned long long ullong;
void fcount(FILE *f, ullong *char_count, ullong *word_count, ullong *line_count)
{
    rewind(f);
    *char_count = 0;
    *word_count = 0;
    *line_count = 0;
    for (;;) {
        int c = fgetc(f);
        switch (c) {
            case EOF:  return;
            case '\n': (*line_count)++;
            case '\t':
            case ' ':  (*word_count)++;
            default:   (*char_count)++;
        }
    }
}

现在考虑该函数的包装器，它确实需要返回值

int count(char *destination, char *filename) {
    char fn[strlen(filename) + 8];
    sprintf(fn, "/files/%s", filename);

    FILE *f = fopen(fn, "r");
    if (f == NULL) {
        /* Note: This exit code is defined within <stdlib.h> */
        return EXIT_FAILURE;
    }

    ullong char_count, word_count, line_count;
    count(f, &char_count, &word_count, &line_count);
    fclose(f);

    sprintf(destination, "%s contains %llu words, %llu characters and %llu lines.\n", filename, word_count, char_count, line_count);
    return EXIT_SUCCESS;
}

现在您可以判断您的函数是失败还是成功，就像许多标准库函数一样!耶!

char buf[128];
count(buf, file_name);
printf("%s", buf);

您还可以访问您想要写入的字符串...这一切都在一起吗？