source: branches/fscctrl_safety_limits/dim/src/examples/rshServer.cxx@ 19365

#include <dis.hxx>
#include <sys/stat.h>
#include <sys/types.h>
#ifndef WIN32
#include <sys/wait.h>
#include <unistd.h>
#else
#include <process.h>
#endif
#include <ctype.h>

#ifndef WIN32
char outfile[] = "/tmp/dim_rsh_server.dat";
#else
char outfile[] = "c:\\dim_rch_server.dat";
#endif

int my_system(char *);

class Cmd: public DimCommand
{
        char *result;
        DimService *resultSrvc;

        void commandHandler();
public:
        Cmd(char *cmdName, char *resultName): DimCommand(cmdName, "C") 
        {
                result = new char[10];
                strcpy(result,"empty");
                resultSrvc = new DimService(resultName, "C", result, strlen(result)+1);
        };
};

void Cmd::commandHandler()
{
        char *str, *client;
        char commandString[256];
        struct stat buf;
        FILE *fd;
        int ret, size, index, sz;

        str = getString();

        client = DimServer::getClientName();

        cout << "Received " << str << " from " << client << endl;
        unlink(outfile);
        strcpy(commandString,str);
#ifndef WIN32
        strcat(commandString," >& ");
        strcat(commandString,outfile);
        strcat(commandString,"< /dev/null");
#else
        strcat(commandString," > ");
        strcat(commandString,outfile);
        strcat(commandString," 2>&1 ");
#endif
        my_system(commandString);

        delete result;
        ret = stat(outfile, &buf);
        if(ret == -1)
    {
                result = new char[20];
                strcpy(result,"File does not exist");
    }
        else
    {
                size = buf.st_size;
                result = new char[size +1];
                fd = fopen(outfile, "r");
                index = 0;
                while(!feof(fd) && size)
                {
                        sz = fread(&result[index], 1, 512, fd);
                        size -= sz;
                        index += sz;
                        if(!sz)
                        break;
                }
                fclose(fd);
                result[index] = '\0';
        }
        resultSrvc->updateService(result, strlen(result)+1);
}

#ifndef WIN32

extern char **environ;

int my_system (char *command)
{
    int pid, status, ret, n = 0;

    if (command == 0)
        return 1;
    pid = fork();
    if (pid == -1)
        return -1;
    if (pid == 0) {
        char *argv[4];
        argv[0] = "sh";
        argv[1] = "-c";
        argv[2] = command;
        argv[3] = 0;
        execve("/bin/sh", argv, environ);
        exit(127);
    }
    do {
      ret = waitpid(pid, &status,WNOHANG);
      if(ret == -1)
        break;
      usleep(100000);
      n++;
    } while(n < 1200);
    return status;
}

#else
int my_system (char *command)
{
        return system(command);
}

#endif

#ifdef WIN32
int init_sock()
{
        WORD wVersionRequested;
        WSADATA wsaData;
        int err;
        static int sock_init_done = 0;

        if(sock_init_done) return(1);
        wVersionRequested = MAKEWORD( 2, 0 );
        err = WSAStartup( wVersionRequested, &wsaData );

        if ( err != 0 ) {
        return(0);
        }

        if ( LOBYTE( wsaData.wVersion ) != 2 ||
        HIBYTE( wsaData.wVersion ) != 0 ) {

            WSACleanup( );
        return(0); 
        }
        sock_init_done = 1;
        return(1);
}
#endif

void main(int argc, char **argv)
{
        char host[64], cmd_name[128], result_name[128];
        Cmd *cmd;

#ifdef WIN32
        init_sock();
#endif
        gethostname(host, 64);

//Setup command reception and result publishing
        sprintf(cmd_name,"%s/ExecuteCmd",host);
        sprintf(result_name,"%s/CmdResult", host);
        cmd = new Cmd(cmd_name, result_name);

        sprintf(cmd_name,"%s_server",host);
        DimServer::start(cmd_name);
        while(1)
    {
                pause();
    }
}