root/lang/c/friendsd/tcpcalls.c @ 32953

#include "tcpcalls.h"

int create_service( ushort port ) {
  struct sockaddr_in server_sockaddr;
  int server_socket;
  
  printf( "A" ); fflush( stdout );
  bzero( &server_sockaddr, sizeof( server_sockaddr ) );
  printf( "A" ); fflush( stdout );
  server_sockaddr.sin_port = htons( port );
  printf( "A" ); fflush( stdout );
  server_sockaddr.sin_family = AF_INET;
  printf( "A" ); fflush( stdout );
  server_sockaddr.sin_addr.s_addr = htonl( INADDR_ANY );
  printf( "A" ); fflush( stdout );
  server_socket = socket( AF_INET, SOCK_STREAM, 0 );
  printf( "A" ); fflush( stdout );
  if( server_socket < 0 ) {
    perror( "socket" );
    return( -1 );
  }
  printf( "A" ); fflush( stdout );
  if( bind( server_socket, &server_sockaddr, sizeof( server_sockaddr ) ) < 0 ) {
    perror( "bind" );
    close( server_socket );
    return( -1 );
  }
  printf( "A" ); fflush( stdout );
  listen( server_socket, 5 );
  printf( "A" ); fflush( stdout );
  return( server_socket );
}

static sigjmp_buf envOnCT; static int savemaskOnCT;

void onConnectTimeout() {
}

int connect_to_server( char *name, ushort port, int timeout_val ) {
  struct sockaddr_in server_sockaddr;
  struct hostent *server_hostent;
  int toserver_socket, i;
  char str[ 255 ];
  ushort opcode;
  unsigned long hostaddr;
  char server_name[ 255 ];
  
  if( ( name == NULL ) || ( *name == 0 ) ) {
    strcpy( server_name, "localhost" );
  } else
    strcpy( server_name, name );
  
  bzero( &server_sockaddr, sizeof( server_sockaddr ) );
  
  hostaddr = inet_addr( server_name );
  if( ( long )hostaddr != ( long )-1 ) {
    bcopy( &hostaddr, &server_sockaddr.sin_addr, sizeof( hostaddr ) );
  } else {
    server_hostent = gethostbyname( server_name );
    if( server_hostent == NULL )
      return( -1 );
    bcopy( server_hostent->h_addr, &server_sockaddr.sin_addr, server_hostent->h_length );
  }
  server_sockaddr.sin_family = AF_INET;
  server_sockaddr.sin_port = htons( port );
  if( ( toserver_socket = socket( AF_INET, SOCK_STREAM, 0 ) ) < 0 )
    return( -1 );
  
  signal( SIGALRM, onConnectTimeout );
  alarm( timeout_val );
  if( connect( toserver_socket, &server_sockaddr, sizeof( server_sockaddr ) ) < 0 )
    goto ON_FAIL;
  alarm( 0 );
  signal( SIGALRM, SIG_IGN );
  return( toserver_socket );
 ON_FAIL:
  alarm( 0 );
  signal( SIGALRM, SIG_IGN );
  close( toserver_socket );
  return( -1 );
}

int read_short( int sock, ushort *num ) {
  char tmp[ 2 ];
  int result, len;
  
  *num = ( short )0;
  len = 0;
  while( len < 2 ) {
    result = read( sock, tmp, 2 - len );
    if( result == -1 ) {
      if( errno == EAGAIN || errno == EWOULDBLOCK )
        continue;
    } else if( result <= 0 )
      return( result );
    len += result;
  }
  
  *num = ntohs( *( ushort * )tmp );
  return( 2 );
}

int write_short( int sock, ushort num ) {
  char tmp[ 2 ];
  int result;
  
  *( short * )tmp = htons( num );
  result  = write( sock, tmp, 2 );
  if( result <= 0 )
    return( result );
  if( result == 1 ) {
    result = write( sock, tmp +1, 1 );
    if( result <= 0 )
      return( result );
  }
  return( result );
}

int read_blk( int sock, ushort *opcode, char *str ) {
  char *ptr = str;
  int result;
  ushort block_len, b;
  
  if( ( result = read_short( sock, opcode ) ) <= 0 )
    return( result );
  if( ( result = read_short( sock, &block_len ) ) <= 0 )
    return( result );
  b = block_len;
  
  while( block_len > 0 ) {
    result = read( sock, ptr, block_len );
    if( result == -1 ) {
      if( errno == EAGAIN || errno == EWOULDBLOCK )
        continue;
    } else if( result <= 0 )
      return( result );
    block_len -= result;
    ptr += result;
  }
  return( b );
}

int write_blk( int sock, ushort opcode, int block_len, char *str ) {
  char *ptr = str;
  int result;
  
  if( ( result = write_short( sock, opcode ) ) <= 0 )
    return( result );
  if( ( result = write_short( sock, block_len ) ) <= 0 )
    return( result );
  
  while( block_len > 0 ) {
    result = write( sock, ptr, block_len );
    if( result <= 0 )
      return( result );
    block_len -= result;
    ptr += result;
  }
  return( 1 );
}