C++ winsock irc客户端问题

问题描述:

我正在尝试使用C++中的Winsock制作一个IRC聊天机器人。我是一名编程新手,对于使用套接字进行编程来说是新手。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; 
} 

当我运行这个,我看到的是我的变量被传递,然后无限量的零。

您的代码有许多问题。

  1. 你是不是检查socket()错误的返回值(我假设你调用WSAStartup()事先吧?)。

  2. 您的PASSNICK命令结束时没有发送任何换行符。 IRC是一个基于行的协议。这就是为什么你没有从服务器获取任何数据 - 它正在等待你先完成你的命令。

  3. IRC中的各种保留字符必须转义。

  4. 你两次发送PASS命令,当你正在设置NICK命令时使用+运营商,而不是运营商=的。

  5. 您未发送任何USERJOIN命令。

  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 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; 
} 
+0

是的,我是用我调用WSAStartup试图复制,我认为最relavent我编辑的评论,显示缺少的部分代码片段。感谢您的深入解答:-) – user2350585

+0

也有任何插座教程,我应该看看?我在这里和那里观看了几个YouTube Vids ... – user2350585

首先,你忽略了返回值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”字符。