Programming

int sample(int s, char *msg)
{

       int len=100;
       int nbytes=0;
       
       memset(msg, 0x0, 100);
       nbytes= recv(s, msg, len, MSG_PEEK);
       if(nbytes<0 && errno== EWOULDBLOCK)
          return EWOULDBLOCK * -1;
       else
          return nbytes;
}

00098DT001234

        while (1)
        {
                stat =0;
                execReturn=0;
                s2 = accept(s, (struct sockaddr *)&remote, &t);

                if(s2== -1)
                {
                         perror("accept");
                         outfile<<strerror(errno)<<endl;
                         exit(1);
                }

                pidc = fork();
                if(pidc == 0)
                {
                        close(s);
                        int retVal;
                        string xs;
                        retVal = RecieveAllData(s2,outfile,xs);
                        if(retVal !=SUCCESS)
                        {
                                close(s2);
                        }

                        ipMsg parsedValue;
                        TokenizeString(xs,parsedValue);
                        outfile<<"User----"<<parsedValue.user<<endl;
                        outfile<<"Pwd ----"<<parsedValue.pwd<<endl;
                        outfile<<"origin ----"<<parsedValue.origin<<endl;
                        outfile<<"cmd ----"<<parsedValue.command<<endl;


                        pidf = fork();
                        if (pidf == 0)
                        {
                                close(s2);
                                close(s);
                                sleep(5);
                                execReturn =execl("/root/test.sh","test.sh",NULL);
                                outfile<<"Errror"<<endl;
                                _exit(-1);
                        }
                        else if (pidf > 0)
                        {
                                int id = waitpid(pidf,&stat,0);
                                int sat = WEXITSTATUS(stat);
                                outfile<<"status is "<<sat<<"pid is "<<pidf<<endl;
                                char stg[32];
                                sprintf(stg,"%d",sat);
                                send(s2,stg,strlen(stg), 0);
                                close(s2);
                        }
                }
                sleep(5); /* wait 30 seconds */
        }

int main(void)
{

        /* Our process ID and Session ID */
        pid_t pid, sid;

        /* Fork off the parent process */
        pid = fork();
        if (pid < 0)
        {
                exit(EXIT_FAILURE);
        }

        /* If we got a good PID, then
           we can exit the parent process. */
        if (pid > 0) {
                exit(EXIT_SUCCESS);
        }

        /* Change the file mode mask */
        umask(0);

        /* Open any logs here */
        ofstream outfile("/root/log.txt");


        /* Create a new SID for the child process */
        sid = setsid();
        if (sid < 0)
        {
                /* Log the failure */
                exit(EXIT_FAILURE);
        }

        /* Change the current working directory */
        if ((chdir("/")) < 0)
        {
                /* Log the failure */
                exit(EXIT_FAILURE);
        }

        /* Close out the standard file descriptors */
        close(STDIN_FILENO);
        close(STDOUT_FILENO);
        close(STDERR_FILENO);

        /* Daemon-specific initialization goes here */

        int s, s2,  len;
        socklen_t t;
        struct sockaddr_un local, remote;

        if ((s = socket(AF_UNIX, SOCK_STREAM, 0)) == -1)
        {
                perror("socket");
                exit(1);
        }

        local.sun_family = AF_UNIX;
        strcpy(local.sun_path, SOCK_PATH);

        unlink(local.sun_path);
        len = strlen(local.sun_path) + sizeof(local.sun_family);

        if (bind(s, (struct sockaddr *)&local, len) == -1)
        {
                perror("bind");
                exit(1);
        }


        if (listen(s, 5) == -1)
        {
                perror("listen");
                exit(1);
        }


        pid_t  pidf,pidc;
        int stat=0;
        int execReturn =0;
        int n;
        char str[100];
        vector<int> pidV;

        while (1)
        {
                stat =0;
                execReturn=0;
                s2 = accept(s, (struct sockaddr *)&remote, &t);

                if(s2== -1)
                {
                         perror("accept");
                         outfile<<strerror(errno)<<endl;
                         exit(1);
                }

                pidc = fork();
                if(pidc!=0)
                        pidV.push_back(pidc);

                if(pidc == 0)
                {
                        close(s);
                        n = recv(s2, str, 100, 0);
                        str[n]='\0';
                        outfile<<"Message is "<<str;
                        pidf = fork();
                        if (pidf == 0)
                        {
                                close(s2);
                                execReturn =execl("/root/cpp/test.sh","test.sh",NULL);
                                outfile<<"Errror"<<endl;
                                _exit(-1);
                        }
                        else if (pidf > 0)
                        {
                                int id = waitpid(pidf,&stat,0);
                                int sat = WEXITSTATUS(stat);
                                outfile<<"status is "<<sat<<"pid is "<<pidf<<endl;
                                char stg[32];
                                sprintf(stg,"%d",sat);
                                send(s2,stg,strlen(stg), 0);
                                close(s2);
                                _exit(0);
                        }
                }
                else if(pidc > 0)
                {
                        close(s2);
                        //REAP ALL CHILDREN
                        /*
                        vector<int>::iterator beg = pidV.begin();
                        while(beg!=pidV.end())
                        {
                                int id = waitpid(*beg,&stat,WNOHANG);
                                if(id!=0)
                                {
                                        beg = pidV.erase(beg);
                                }
                                else
                                {
                                        ++beg;
                                }

                        }
                       */

                sleep(5); /* wait 30 seconds */
        }
        exit(EXIT_SUCCESS);
}