root/lang/c/friendsd/testfriendsd.cc @ 32953

#include "../friends++/friends++.h"
#include "tcpcalls.h"

#define DAEMON_PASSWORD "KOISHIKAWA"
#define PASSCMD "/home/g540002/friends/bin/authenticate -env /home/g540002/friends/bin/secret/Password"

void usage();
int main( int argc, char **argv );
int secret( void );
int sendCommand( ushort cmd, char *message );
int receiveMessage( void );
void dump( int len, char *str );

static int sock;

void usage() {
  fputs( "TESTFRIENDSD:  an utility for friendsd.\n", stdout );
  fputs( "         written by OKU Kazuho(g540002)\n", stdout );
  fputs( "         usage - testfriendsd [command]\n", stdout );
  fputs( "\n", stdout );
}

int main( int argc, char **argv ) {
  ushort cmd;
  char sendMessage[ 256 ];
  
  if( argc != 2 ) {
    usage();
    exit( 1 );
  }
  cmd = ( ushort )atoi( argv[ 1 ] );
  if( cmd >= 0x1000 ) {
    if( secret() ) {
      fprintf( stderr, "Wrong password.\n" );
      exit( 1 );
    } else
      strcpy( sendMessage, DAEMON_PASSWORD );
  } else
    strcpy( sendMessage, "\0" );
  
  if( 0 >= ( sock = openService() ) ) {
    fprintf( stderr, "Couldn't connect to daemon via TCP.\n" );
    exit( 1 );
  }
  if( sendCommand( cmd, sendMessage ) ) {
    fprintf( stderr, "Couldn't send message via TCP.\n" );
    closeService( sock );
    exit( 1 );
  }
  if( receiveMessage() ) {
    fprintf( stderr, "Couldn't receive message via TCP.\n" );
    closeService( sock );
    exit( 1 );
  }
  closeService( sock );
  return( 0 );
}

int secret() {
  FILE *fp;
  int c, x;
  
  if( NULL == ( fp = popen( PASSCMD, "r" ) ) ) {
    fprintf( stderr, "Couldn't create pipe.\n" );
    exit( 1 );
  }
  for( x = 0; ; x++ ) {
    if( EOF == ( c = fgetc( fp ) ) )
      break;
    else
      putchar( c );
  }
  pclose( fp );
  return( x );
}

int sendCommand( ushort cmd, char *message ) {
  if( 0 >= write_short( sock, cmd ) )
    goto ON_ERROR;
  if( message[ 0 ] )
    if( 0 >= write_blk( sock, ACK, strlen( message ) + 1, message ) )
      goto ON_ERROR;
  return( 0 );
 ON_ERROR:
  return( 1 );
}

int receiveMessage() {
  ushort opcode;
  int max, x;
  char str[ 256 ];
  
  if( 0 >= read_short( sock, &opcode ) )
    goto ON_ERROR;
  max = ( int )opcode;
  for( x = 0; x < max; x++ ) {
    if( 0 >= read_blk( sock, &opcode, str ) )
      goto ON_ERROR;
    dump( 84, str );
  }
  return( 0 );
 ON_ERROR:
  return( 1 );
}




void dump( int len, char *str ) {
  int x, y;
  
  for( x = 0; ; x += 16 ) {
    printf( "%08X  ", x );
    for( y = 0; y < 16; y++ )
      if( x + y < len )
        printf( "%02X ", ( unsigned char )str[ x + y ] );
      else
        printf( "   " );
    printf( "  " );
    for( y = 0; y < 16; y++ )
      if( x + y < len ) {
        if( ' ' <= str[ x + y ] && str[ x + y ] < 0x7F )
          printf( "%c", str[ x + y ] );
        else
        printf( "." );
      } else
        printf( " " );
    printf( "\n" );
    if( x + y >= len )
      return;
  }
}