Разъем C переполняет память - PullRequest
0 голосов
/ 31 января 2012

Я пишу пользовательский VNC сервер-клиент. Однако после 159 команд каждый раз происходит сбой сервера. Я не могу понять, почему происходит сбой, но, похоже, что-то переполнено. Можно ли заполнить сокеты или стандартный ввод / вывод? Или это скорее что-то с моими элементами управления X?

Исходный код:

#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <sys/wait.h>
#include <sys/select.h>
#include <X11/Xlib.h>
#include <X11/Xutil.h>
#include <X11/extensions/XTest.h>



void dostuff(int);


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

int main(int argc, char *argv[])
        {
        int sockfd, newsockfd, portno, pid;
        socklen_t clilen;
        struct sockaddr_in serv_addr, cli_addr;

        if (argc < 2)
                {
                fprintf(stderr,"ERROR, no port provided\n");
                exit(1);
                }
     sockfd = socket(AF_INET, SOCK_STREAM, 0);
     if (sockfd < 0)
        error("ERROR opening 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(sockfd, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0)
              error("ERROR on binding");
     listen(sockfd,5);
     clilen = sizeof(cli_addr);


        while (1)
                {
                newsockfd = accept(sockfd,
                (struct sockaddr *) &cli_addr, &clilen);
                if (newsockfd < 0)
                error("ERROR on accept");
                pid = fork();
                if (pid < 0)
                        error("ERROR on fork");
                if (pid == 0)
                        {
                        close(sockfd);
                        dostuff(newsockfd);
                        exit(0);
                        }
                else close(newsockfd);
                signal(SIGCHLD,SIG_IGN);
                }

        /* end of while */
        close(sockfd);
        return 0; /* we never get here */
}

/******** DOSTUFF() *********************
 There is a separate instance of this function
 for each connection.  It handles all communication
 once a connnection has been established.
 *****************************************/
void dostuff (int sock)
{
while(1)
        {
        fflush(stdout);

        int n;
        char buffer[64];

        bzero(buffer,64);
        printf("START:\n");
        n = read(sock,buffer,63);
        printf("buffer[1]: %d \n", buffer[1]);
        if (n < 0) error("ERROR reading from socket");


        Display *thedisplay;
        Window thewindow;
        int screen;
        screen=buffer[0]-48; //first element is always the screen number
        printf("screen = %d \n", screen);

        thedisplay=XOpenDisplay(NULL);
        thewindow=XRootWindow(thedisplay,screen);
        int screenwidth = DisplayWidth(thedisplay, screen);
        int screenheight = DisplayHeight(thedisplay, screen);
        printf("width: %d, height %d \n", screenwidth, screenheight);

        printf("buff[1] = %d \n", buffer[1]);

        //switch on second char
        switch(buffer[1])
        {
        case 109: // second element == 'm' we treat it like a mousemove

                printf("mousemovement\n");
                int xcoord = 100*(buffer[3]-48)+10*(buffer[4]-48)+1*(buffer[5]-48);
                int ycoord = 100*(buffer[7]-48)+10*(buffer[8]-48)+1*(buffer[9]-48);
                if (buffer[2]==49) xcoord = xcoord*(-1);
                if (buffer[6]==49) ycoord = ycoord*(-1);
                //printf("dx: %d, dy: %d \n", xcoord, ycoord);
                Window windowreturned;
                int xroot, yroot, xwin, ywin, mask_return;
                XQueryPointer(thedisplay, thewindow, &windowreturned, &windowreturned, &xroot, &yroot, &xwin, &ywin, &mask_return);
                //printf("xroot X: %d  Y: %d  \n", xroot, yroot);
                //printf("xcoord+xroot %d %d \n", xcoord+xroot, ycoord+yroot);
                //printf("screenwidth %d %d \n", screenwidth, screenheight);
                if (xcoord+xroot > 0 && xcoord+xroot < screenwidth && ycoord+yroot>0 && ycoord+yroot<screenheight)
                    {
                    //printf("good to go\n");
                    XWarpPointer(thedisplay,None,None,0,0,0,0,xcoord,ycoord);
                    XSync(thedisplay, False);
                    }
                break;
        case 107: //second element is a 'k' so we sendkey
                SendKey (thedisplay, buffer[2]);
                break;
        case 98: // b us for mousebutton. 1 is leftclick, 2 is middle click, 3 is right click, 4 us up-scroll, 5 is downscroll
                XTestGrabControl(thedisplay, True);
//              XTestFakeButtonEvent(thedisplay, 1, True, 0);
//              XTestFakeButtonEvent(thedisplay, 1, False, 0);
                XTestFakeButtonEvent(thedisplay, buffer[2]-48, True, 0);
                XTestFakeButtonEvent(thedisplay, buffer[2]-48, False, 0);
                XSync (thedisplay, False);
                XTestGrabControl (thedisplay, False);
                break;
        case 99: //second element is 'c', so we center on the screen

                XWarpPointer(thedisplay,None,thewindow,0,0,0,0,screenwidth*.5,screenheight*.5);
                XSync(thedisplay, False);
                break;

        default:
                close(sock);
                error("ERROR incorrect formattttttt\n");
                break;
        }

    printf("got this far\n");
    n = write(sock,"spanks\n",6);
    if (n < 0) error("ERROR writing to socket\n");

 }
}

Ответы [ 3 ]

1 голос
/ 31 января 2012

Простой вызов read не гарантирует, что вы получите все 63 байта или что вы получите 63 байта, на которые вы надеялись .. Я бы посоветовал вам каким-то образом определить, сколько данных вам нужно получить (отправьте длину данныхсначала), а затем поместите функцию recv в цикл, пока у вас не будут все данные. Также должна быть проверена функция send (с клиента).

0 голосов
/ 01 февраля 2012

Решение: ошибка заключалась в том, что XOpenDisplay находился внутри бесконечного цикла, не будучи закрытым. Я просто переместил команду XOpenDisplay перед бесконечным циклом в dostuff ().

На самом деле это не ошибка сокета.

0 голосов
/ 31 января 2012

Как насчет попытки закрыть сокет в конце dostuff ()?Сервер может получить слишком много открытых соединений.

...