10

我正在用 C 编写一个 IRC 机器人,但遇到了障碍。

在我的主要功能中,我创建了我的套接字并连接,所有这些快乐的东西。然后我有一个(几乎)无限循环来读取从服务器发回的内容。然后我将读取的内容传递给辅助函数,processLine(char *line)- 问题是,以下代码读取直到我的缓冲区已满 - 我希望它只读取文本,直到出现换行符 (\n) 或回车符 (\r) (从而结束该行)

   while (buffer[0] && buffer[1]) {
        for (i=0;i<BUFSIZE;i++) buffer[i]='\0';
        if (recv(sock, buffer, BUFSIZE, 0) == SOCKET_ERROR)
            processError();

        processLine(buffer);
    }

最终发生的事情是许多线路都卡在一起了,发生这种情况时我无法正确处理线路。

如果你不熟悉 IRC 协议,一个简短的总结就是当消息发送时,它通常看起来像这样::YourNickName!YourIdent@YourHostName PRIVMSG #someChannel :The rest on from here is the message sent... 例如,登录通知是这样的::the.hostname.of.the.server ### bla some text bla### 是一个代码(? ) 用于处理 - 即 372 指示以下文本是今日消息的一部分。

当它们全部塞在一起时,我无法读取哪一行的数字是什么,因为我找不到一行的开始或结束位置!

我非常感谢您的帮助!

PS:这是在 linux 上编译/运行的,但我最终想将它移植到 windows,所以我尽可能多地制作多平台。

PSS:这是我的 processLine() 代码:

void processLine(const char *line) {
    char *buffer, *words[MAX_WORDS], *aPtr;
    char response[100];
    int count = 0, i;
    buffer = strdup(line);

    printf("BLA %s", line);

    while((aPtr = strsep(&buffer, " ")) && count < MAX_WORDS)
        words[count++] = aPtr;
        printf("DEBUG %s\n", words[1]);
    if (strcmp(words[0], "PING") == 0) {
        strcpy(response, "PONG ");
        strcat(response, words[1]);
        sendLine(NULL, response); /* This is a custom function, basically it's a send ALL function */
    } else if (strcmp(words[1], "376") == 0) { /* We got logged in, send login responses (i.e. channel joins) */
        sendLine(NULL, "JOIN #cbot");
    }
}
4

2 回答 2

11

处理这个问题的常用方法是recv进入应用程序中的持久缓冲区,然后拉出一行并处理它。稍后您可以在再次调用之前处理缓冲区中的剩余行recv。请记住,缓冲区中的最后一行可能只被部分接收;您必须通过重新输入recv来完成该行来处理这种情况。

这是一个示例(完全未经测试!还寻找 a \n, not \r\n):

#define BUFFER_SIZE 1024
char inbuf[BUFFER_SIZE];
size_t inbuf_used = 0;

/* Final \n is replaced with \0 before calling process_line */
void process_line(char *lineptr);
void input_pump(int fd) {
  size_t inbuf_remain = sizeof(inbuf) - inbuf_used;
  if (inbuf_remain == 0) {
    fprintf(stderr, "Line exceeded buffer length!\n");
    abort();
  }

  ssize_t rv = recv(fd, (void*)&inbuf[inbuf_used], inbuf_remain, MSG_DONTWAIT);
  if (rv == 0) {
    fprintf(stderr, "Connection closed.\n");
    abort();
  }
  if (rv < 0 && errno == EAGAIN) {
    /* no data for now, call back when the socket is readable */
    return;
  }
  if (rv < 0) {
    perror("Connection error");
    abort();
  }
  inbuf_used += rv;

  /* Scan for newlines in the line buffer; we're careful here to deal with embedded \0s
   * an evil server may send, as well as only processing lines that are complete.
   */
  char *line_start = inbuf;
  char *line_end;
  while ( (line_end = (char*)memchr((void*)line_start, '\n', inbuf_used - (line_start - inbuf))))
  {
    *line_end = 0;
    process_line(line_start);
    line_start = line_end + 1;
  }
  /* Shift buffer down so the unprocessed data is at the start */
  inbuf_used -= (line_start - inbuf);
  memmove(innbuf, line_start, inbuf_used);
}
于 2011-05-22T20:32:57.633 回答
7

TCP不提供任何此类排序。正如@bdonlan 已经说过的,您应该实现以下内容:

  • 不断recv地从套接字进入缓冲区
  • 在每一个上recv,检查接收到的字节是否包含\n
  • 如果\n使用缓冲区中的所有内容(并清除它)

我对此感觉不太好(我在某处读到你不应该将低级 I/O 与stdioI/O 混合使用),但你也许可以使用fdopen.

你需要做的就是

  • 用于fdopen(3)将您的套接字与FILE *
  • 用于setvbuf告诉 stdio 你希望它是行缓冲的 ( _IOLBF) 而不是默认的块缓冲。

此时,您应该已经有效地将工作从您的手中转移到stdio. 然后你可以继续fgetsFILE *.

于 2011-05-22T20:42:55.147 回答