C ++ Winsock IRC客戶端問題

Question

我正在嘗試使用C ++中的Winsock制作IRC聊天機器人。 我是編程新手，還是使用套接字編程的新手。

我正在嘗試連接到我的Twitch頻道。 我可以成功建立連接，並傳遞幾個緩沖區（即我的機器人的密碼或oauth令牌，用戶名以及我要加入的渠道）。

但是，當我調用recv() ，Twitch服務器沒有發送任何數據。

#define WIN32_LEAN_AND_MEAN
#define _WINSOCK_DEPRECATED_NO_WARNINGS
#define DEFAULT_BUFLEN 512

#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <WinSock2.h>
#include <Ws2tcpip.h>
#include <cstdlib>
#include <iostream>

#pragma comment(lib, "Ws2_32.lib")
#pragma comment (lib, "Mswsock.lib")
#pragma comment (lib, "AdvApi32.lib")


using namespace std;

int main()
{
    string Buffer;
    char  buffers[1024 * 8] = { "0" };
    string oauth = "oauthtoken";
    string nick = "text_adventure_bot";
    char recvbuf[DEFAULT_BUFLEN];
    int recvbuflen = DEFAULT_BUFLEN;
    int iResult;
    string hostname = "irc.chat.twitch.tv";
    struct addrinfo *result = NULL,
        *ptr = NULL,
        hints;


    WSABUF DataBuf;
    WSADATA  wsadata;
    WORD wVersionRequested;

    WORD DllVersion = MAKEWORD(2, 1);
    iResult = WSAStartup(MAKEWORD(2, 2), &wsadata);
    if(iResult != 0)
    {
        MessageBoxA(NULL, "Winsock startup failed", "Error", MB_OK | MB_ICONERROR);
        exit(1);
    }

    SOCKADDR_IN addr; //the ip
    int sizeofaddr = sizeof(addr);
    addr.sin_addr.s_addr = inet_addr("52.25.27.117");
    addr.sin_port = htons(6667);
    addr.sin_family = AF_INET;

    SOCKET sock = socket(AF_INET, SOCK_STREAM, NULL);

    if (connect(sock, (SOCKADDR*)&addr, sizeofaddr) != 0)
    {
        cout << "Connection error" << endl;
    }

    cout << "connected" << endl;

    Buffer = "PASS " + oauth;

    send(sock, Buffer.c_str(), (int)strlen(Buffer.c_str()), 0); 
    recv(sock, buffers, 1024 * 8, 0);
    cout << Buffer.c_str() << endl << buffers << endl << endl;

    Buffer + "NICK " + nick;

    send(sock, Buffer.c_str(), strlen(Buffer.c_str()), 0);
    recv(sock, buffers, 1024 * 8, 0);
    cout << Buffer.c_str() << endl << buffers << endl << endl;

    while (true) {
        recv(sock, buffers, 1024 * 8, 0);
        cout << buffers << endl << endl;
        if (buffers[0] == 'PING') {
            Buffer = "PONG :" + hostname + "\r\n";
            send(sock, Buffer.c_str(), strlen(Buffer.c_str()), 0);
            cout << Buffer.c_str() << endl << buffers << endl << endl;
        }
    }
    return 0;
}

當我運行它時，我所看到的只是我的變量被傳遞，然后是無限數量的零。

Answer 1

您的代碼有很多問題。

1. 您沒有在檢查socket()的返回值是否有錯誤（我假設您是在事先調用WSAStartup() ，對嗎？）。

2. 您不會在PASS和NICK命令的末尾發送任何換行符。 IRC是基於行的協議。 這就是為什么您沒有從服務器獲取任何數據的原因-它正在等待您首先完成命令。

3. IRC中的各種保留字符都必須轉義。

4. 您將發送兩次PASS命令，因為在設置NICK命令時使用+運算符而不是=運算符。

5. 您沒有發送任何USER和JOIN命令。

6. 您不應該使用strlen()計算std::string的長度。 為此，它具有自己的length()和size()方法。

7. 您將忽略send()和recv()的返回值。 TCP是字節流，但是您沒有考慮send()和recv()返回的字節數少於請求的字節數。 您需要循環調用它們，直到發送/接收了所有期望的字節為止。

嘗試類似這樣的方法：

#include <windows.h>
#include <winsock.h>

#include <iostream>
#include <string>
#include <algorithm>

void replaceStr(std::string &str, const std::string &oldStr, const std::string &newStr)
{
    std::string::size_type index = 0;
    do
    {
        index = str.find(oldStr, index);
        if (index == std::string::npos)
            return;

        str.replace(index, oldStr.length(), newStr);
        index += newStr.length();
    }
    while (true);
}

std::string quote(const std::string &s)
{
    std::string result = s;
    replaceStr(result, "\x10", "\x10""\x10");
    replaceStr(result, "\0", "\x10""0");
    replaceStr(result, "\n", "\x10""n");
    replaceStr(result, "\r", "\x10""r");
    return result;
}

std::string unquote(const std::string &s)
{
    std::string result = s;
    std::string::size_type len = result.length();
    std::string::size_type index = 0;
    while (index < len)
    {
        index = result.find("\x10", index);
        if (index = std::string::npos)
            break;

        result.erase(index, 1);
        --len;

        if (index >= len)
            break;

        switch (result[index])
        {
            case '0':
                result[index] = '\0';
                break;
            case 'n':
                result[index] := '\n';
                break;
            case 'r':
                result[index] = '\r';
                break;
        }

        ++index;
    }

    return result;
}

std::string fetch(std::string &s, const std::string &delim)
{
    std::string result;
    std::string::size_type pos = s.find(delim);
    if (pos == std::string::npos)
    {
        result = s;
        s = "";
    }
    else
    {
        result = s.substr(0, pos);
        s.erase(0, pos+delim.length());
    }
    return result;
}

bool sendStr(SOCKET sock, const std::string &s)
{
    const char *ptr = s.c_str();
    int len = s.length();

    while (len > 0)
    {
        int ret = send(sock, ptr, len, 0);
        if (ret == SOCKET_ERROR)
        {
            std::cout << "send() error: " << WSAGetLastError() << std::endl;
            return false;
        }
        ptr += ret;
        len -= ret;
    }

    return true;
}

bool sendCmd(SOCKET sock, const std::string &cmd)
{
    std::cout << "Sending: " << cmd << std::endl;
    return sendStr(sock, quote(cmd)) && sendStr(sock, "\r\n");
}

int main()
{
    int exitCode = -1;

    WSADATA wsa;
    int ret = WSAStartup(MAKEWORD(2, 0), &wsa);
    if (ret != 0)
    {
        std::cout << "Winsock init error: " << ret << std::endl;
        goto done;
    }

    SOCKET sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
    if (sock == INVALID_SOCKET)
    {
        std::cout << "socket() error: " << WSAGetLastError() << std::endl;
        goto done;
    }

    SOCKADDR_IN addr = {0};
    addr.sin_family = AF_INET;
    addr.sin_addr.s_addr = inet_addr("52.25.27.117"); //the ip
    addr.sin_port = htons(6667);

    if (connect(sock, (SOCKADDR*)&addr, sizeof(addr)) != 0)
    {
        std::cout << "connect() error: " << WSAGetLastError() << std::endl;
        goto cleanup:
    }

    std::cout << "connected" << std::endl;

    std::string oauth = ...;
    std::string nick = ...;
    std::string user = ...;
    std::string channel = ...;

    sendCmd("PASS " + oauth);
    sendCmd("NICK " + nick);
    sendCmd("USER " + user);
    sendCmd("JOIN " + channel);

    char buf[1024];
    std::string LineBuffer;
    std::string::size_type StartIdx = 0;

    do
    {
        int ret = recv(sock, buf, sizeof(buf), 0);
        if (ret == SOCKET_ERROR)
        {
            std::cout << "recv() error: " << WSAGetLastError() << std::endl;
            goto cleanup;
        }

        if (ret == 0)
        {
            std::cout << "Server disconnected" << std::endl;
            break;
        }

        LineBuffer.append(buf, ret);

        do
        {
            std::string::size_type pos = LineBuffer.find('\n', StartIdx);
            if (pos == std::string::npos)
                break;

            std::string::size_type len = pos;
            if ((pos > 0) && (LineBuffer[pos-1] == '\r'))
                --len;

            std::string msg = unquote(LineBuffer.substr(0, len));
            LineBuffer.erase(0, pos+1);
            StartIdx = 0;

            std::string senderNick;
            std::string senderHost;

            if (!msg.empty() && (msg[0] == ':'))
            {
                std::string tmp = fetch(msg, " ");
                tmp.erase(0, 1); // remove ':'
                senderNick = fetch(tmp, "!");
                senderHost = tmp;
            }

            std::cout << "Received: " << msg << std::endl;

            if (msg == "PING")
                sendCmd("PONG :" + hostname);
        }
        while(true);
    }
    while (true);

    exitCode = 0;

cleanup:
    closesocket(sock);

done:
    return exitCode;
}

Answer 2

首先，您忽略recv的返回值，因此您不知道收到了多少字節。

其次，您實際上沒有在任何地方實現IRC消息協議（請參閱RFC1459的2.3節）。 因此，您沒有理由假定緩沖區的第一個字節將包含IRC協議消息的第一個字節。 僅IRC消息協議的實際實現可以產生一個緩沖區，其第一個字節是IRC消息的第一個字節。

同樣，您不能執行以下操作：

    cout << buffers << endl << endl;

流的operator<<(const char *)需要一個指向C樣式字符串的指針。 在解析從TCP連接接收到的數據並從中產生C樣式的字符串之前，不得將其視為C樣式的字符串。

也：

    if (buffers[0] == 'PING') {

您真的是說PING是一個多字節字符常量嗎？ 據我所知，沒有名為PING的多字節字符。 而且，無論如何，IRC服務器都會發送原義的四個字符串“ PING”，而不是單個“ PING”字符。