socket_io_bouncer.c
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:6k
源码类别:

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: socket_io_bouncer.c,v $
  4.  * PRODUCTION Revision 1000.0  2003/10/29 17:02:45  gouriano
  5.  * PRODUCTION PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.7
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: socket_io_bouncer.c,v 1000.0 2003/10/29 17:02:45 gouriano Exp $
  10.  * ===========================================================================
  11.  *
  12.  *                            PUBLIC DOMAIN NOTICE
  13.  *               National Center for Biotechnology Information
  14.  *
  15.  *  This software/database is a "United States Government Work" under the
  16.  *  terms of the United States Copyright Act.  It was written as part of
  17.  *  the author's official duties as a United States Government employee and
  18.  *  thus cannot be copyrighted.  This software/database is freely available
  19.  *  to the public for use. The National Library of Medicine and the U.S.
  20.  *  Government have not placed any restriction on its use or reproduction.
  21.  *
  22.  *  Although all reasonable efforts have been taken to ensure the accuracy
  23.  *  and reliability of the software and data, the NLM and the U.S.
  24.  *  Government do not and cannot warrant the performance or results that
  25.  *  may be obtained by using this software or data. The NLM and the U.S.
  26.  *  Government disclaim all warranties, express or implied, including
  27.  *  warranties of performance, merchantability or fitness for any particular
  28.  *  purpose.
  29.  *
  30.  *  Please cite the author in any work or product based on this material.
  31.  *
  32.  * ===========================================================================
  33.  *
  34.  * Author:  Denis Vakatov
  35.  *
  36.  * File Description:
  37.  *   A simple test program to listen on a socket and just bounce all
  38.  *   incoming data (e.g., can be used to test "ncbi_socket_connector.[ch]")
  39.  *
  40.  */
  41. #include "../ncbi_ansi_ext.h"
  42. #include <connect/ncbi_socket.h>
  43. #include <stdio.h>
  44. #include <stdlib.h>
  45. /* This header must go last */
  46. #include "test_assert.h"
  47. static FILE* s_LogFile;
  48. /* Listen on the specified "port".
  49.  * Accept connections and bounce the incoming data back to the client
  50.  * (only one client at a time).
  51.  * Perform up to "n_cycle" sessions, then exit.
  52.  */
  53. static void s_DoServer(unsigned short port, int n_cycle)
  54. {
  55.     LSOCK      lsock;
  56.     EIO_Status status;
  57.     fprintf(s_LogFile, "DoServer(port = %hu, n_cycle = %u)n", port, n_cycle);
  58.     /* Create listening socket */
  59.     status = LSOCK_Create(port, 1, &lsock);
  60.     assert(status == eIO_Success);
  61.     /* Accept connections from clients and run test sessions */
  62.     while ( n_cycle-- ) {
  63.         SOCK     sock;
  64.         size_t   n_io, n_io_done;
  65.         char     buf[1024];
  66.         STimeout timeout;
  67.         /* Accept connection */
  68.         status = LSOCK_Accept(lsock, 0, &sock);
  69.         assert(status == eIO_Success);
  70.         /* Set i/o timeouts for the accepted connection */
  71.         timeout.sec  = 100;
  72.         timeout.usec = 888;
  73.         status = SOCK_SetTimeout(sock, eIO_Read,  &timeout);
  74.         assert(status == eIO_Success);
  75.         status = SOCK_SetTimeout(sock, eIO_Write, &timeout);
  76.         assert(status == eIO_Success);
  77.         /* Bounce all incoming data back to the client */
  78.         while ((status = SOCK_Read(sock,buf,sizeof(buf),&n_io,eIO_ReadPlain))
  79.                == eIO_Success) {
  80.             status = SOCK_Write(sock, buf, n_io, &n_io_done, eIO_WritePersist);
  81.             if (status != eIO_Success) {
  82.                 fprintf(s_LogFile,
  83.                         "Failed to write -- DoServer(n_cycle = %d): %sn",
  84.                         n_cycle, IO_StatusStr(status));
  85.                 break;
  86.             }
  87.         }
  88.         assert(status == eIO_Closed);
  89.         /* Close connection */
  90.         status = SOCK_Close(sock);
  91.         assert(status == eIO_Success  ||  status == eIO_Closed);
  92.     }
  93.     /* Close listening socket */
  94.     status = LSOCK_Close(lsock);
  95.     assert(status == eIO_Success);
  96. }
  97. int main(int argc, const char* argv[])
  98. {
  99.     /* variables */
  100. #define MIN_PORT 5001
  101.     unsigned short port = 0;
  102.     int n_cycle = 100;
  103.     const char* env;
  104.     /* logging */
  105.     s_LogFile = fopen("socket_io_bouncer.log", "ab");
  106.     assert(s_LogFile);
  107.     /* cmd.-line args */
  108.     switch ( argc ) {
  109.     case 3: {
  110.         if (sscanf(argv[2], "%d", &n_cycle) != 1  ||  n_cycle <= 0)
  111.             break;
  112.     }
  113.     case 2: {
  114.         int i_port = -1;
  115.         if (sscanf(argv[1], "%d", &i_port) != 1  ||
  116.             i_port < MIN_PORT  ||  65535 < i_port)
  117.             break;
  118.         port = (unsigned short) i_port;
  119.     }
  120.     } /* switch */
  121.     if (port == 0) {
  122.         fprintf(stderr,
  123.                 "Usage: %s <port> [n_cycle]n where <port> not less than %dn",
  124.                 argv[0], (int) MIN_PORT);
  125.         fprintf(s_LogFile,
  126.                 "Usage: %s <port> [n_cycle]n where <port> not less than %dn",
  127.                 argv[0], (int) MIN_PORT);
  128.         return 1;
  129.     }
  130.     if ((env = getenv("CONN_DEBUG_PRINTOUT")) != 0 &&
  131.         (strcasecmp(env, "true") == 0 || strcasecmp(env, "1") == 0 ||
  132.          strcasecmp(env, "data") == 0 || strcasecmp(env, "all") == 0)) {
  133.         SOCK_SetDataLoggingAPI(eOn);
  134.     }
  135.     /* run */
  136.     s_DoServer(port, n_cycle);
  137.     /* cleanup */
  138.     verify(SOCK_ShutdownAPI() == eIO_Success);
  139.     fclose(s_LogFile);
  140.     return 0;
  141. }
  142. /*
  143.  * --------------------------------------------------------------------------
  144.  * $Log: socket_io_bouncer.c,v $
  145.  * Revision 1000.0  2003/10/29 17:02:45  gouriano
  146.  * PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.7
  147.  *
  148.  * Revision 6.7  2002/12/04 19:50:31  lavr
  149.  * #include "../ncbi_ansi_ext.h" instead of <string.h> to define strcasecmp()
  150.  *
  151.  * Revision 6.6  2002/12/04 17:00:18  lavr
  152.  * Open log file in append mode; toggle logging from the environment
  153.  *
  154.  * Revision 6.5  2002/08/12 15:10:43  lavr
  155.  * Use persistent SOCK_Write()
  156.  *
  157.  * Revision 6.4  2002/08/07 16:38:08  lavr
  158.  * EIO_ReadMethod enums changed accordingly; log moved to end
  159.  *
  160.  * Revision 6.3  2002/03/22 19:46:11  lavr
  161.  * Test_assert.h made last among the include files
  162.  *
  163.  * Revision 6.2  2002/01/16 21:23:14  vakatov
  164.  * Utilize header "test_assert.h" to switch on ASSERTs in the Release mode too
  165.  *
  166.  * Revision 6.1  2000/04/07 20:04:37  vakatov
  167.  * Initial revision
  168.  *
  169.  * ==========================================================================
  170.  */