-2
#include <winsock2.h>
#include <stdio.h>

const int PORT = 6667;
const char *SERVER = "irc.freenode.org";
const char *CHAN = "#channela";
const char *NICK = "loveMilk";

const int MAX_BUFF_SIZE = 512;

int sock_conn(SOCKET *socketn, const char *HOST, int portn);
int sock_send(SOCKET *socketn, char* msg, ...);

int main(int argc, char *argv[])
{
    WSADATA wsadata;
    char buff[MAX_BUFF_SIZE];
    char oBuff[MAX_BUFF_SIZE];
    int buffRec;

    if (WSAStartup(MAKEWORD(2,2), &wsadata) != 0)
        return 0;

    SOCKET sock;

    if(sock_conn(&sock, SERVER, PORT) != 0)
    {
        WSACleanup();
        return 0;
    }

    printf("connected.\n");

    sock_send(&sock, "USER %s \"\" \"127.0.0.1\" :%s\r\n", NICK, NICK);
    sock_send(&sock, "NICK %s\r\n", NICK);
    Sleep(100);

    sock_send(&sock, "JOIN %s\r\n", CHAN);
    printf("Joined channel.\n");

    while(1)
    {
        memset(buff, 0, MAX_BUFF_SIZE);
        memset(oBuff, 0, MAX_BUFF_SIZE);

        buffRec = recv(sock, buff, MAX_BUFF_SIZE, 0);
        if((buffRec == 0) || (buffRec == SOCKET_ERROR)) break;

        /* New line: Terminate buffer as a string */
        buff[buffRec] = '\0';

        if(buff[0] != ':')
        {
            strcpy(oBuff, "PONG :");
            printf("PONG");
            sock_send(&sock, oBuff);
        }
        else
        {
            if(strstr(buff, "PRIVMSG"))
            {
                int i, num = 0;
                for(i = 0; i < strlen(buff); ++i) if(buff[i] = ' ') ++num;
                char** parts = malloc(sizeof(char*) * num);
                char *p;
                p = strtok(buff, " ");
                int j = 0;
                while(p != NULL)
                {
                    parts[j] = p;
                    j++;
                    p = strtok(NULL, " ");
                }
                printf("%s", parts[3]);

                free(parts);
            }
        }
    }
    closesocket(sock);
    return 1;
}

int sock_conn(SOCKET *socketn, const char *HOST, int portn)
{
    WSADATA wsadata;
    SOCKADDR_IN sockA;
    LPHOSTENT hostE;

    if(WSAStartup(MAKEWORD(2,2), &wsadata) == -1) return -1;
    if(!(hostE = gethostbyname(HOST)))
    {
        WSACleanup();
        return -1;
    }
    if ((*socketn = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP)) == INVALID_SOCKET)
    {
        WSACleanup();
        return -1;
    }

    sockA.sin_family = AF_INET;
    sockA.sin_port = htons(portn);
    sockA.sin_addr = *((LPIN_ADDR)*hostE->h_addr_list);

    if(connect(*socketn, (LPSOCKADDR)&sockA, sizeof(struct sockaddr)) == SOCKET_ERROR)
    {
        WSACleanup();
        return -1;
    }
}

int sock_send(SOCKET *socketn, char* msg, ...)
{
    char buff[MAX_BUFF_SIZE];
    va_list va;
    va_start(va, msg);
    vsprintf(buff, msg, va);
    va_end(va);
    send(*socketn, buff, strlen(buff), 0);
    return 1;
}

パーツは常にnullです、なぜですか?誰かが私にこの行でバフを終了するように言いました:buff[buffRec] = '\0';しかし、それでも私は何も得られませんでした、どうすればバフを文字列に変更できますか?これが問題だと信じています...ボットは接続されており、すべて通信できますが、パーツに手を伸ばそうとすると、NULLになります。

4

1 に答える 1

1
char buff[MAX_BUFF_SIZE];

/* ... */

buffRec = recv(sock, buff, MAX_BUFF_SIZE, 0);

buff[buffRec] = '\0';

buffRec値がの場合MAX_BUFF_SIZE、バッファ オーバーフローが発生しています。

于 2012-10-17T14:59:30.103 に答える