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

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: test_fw.c,v $
  4.  * PRODUCTION Revision 1000.0  2003/10/29 17:03:19  gouriano
  5.  * PRODUCTION PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.8
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: test_fw.c,v 1000.0 2003/10/29 17:03:19 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.  *   Read data from a file, and send these data line-by-line to a socket.
  38.  *   Then, read data from the socket.
  39.  *   Trace all I/O going through the socket in both directions.
  40.  *
  41.  */
  42. #include <connect/ncbi_socket.h>
  43. #include <connect/ncbi_util.h>
  44. #include <stdlib.h>
  45. #include <string.h>
  46. /* This header must go last */
  47. #include "test_assert.h"
  48. static void s_SendData(FILE* fp, SOCK sock)
  49. {
  50.     char   buf[2048];
  51.     size_t n_written;
  52.     while ( fgets(buf, sizeof(buf)-3, fp) ) {
  53.         size_t len = strlen(buf);
  54.         assert(buf[len - 1] == 'n');
  55.         assert(buf[len]     == '');
  56.         assert(!feof(fp)  &&  !ferror(fp));
  57.         len--;
  58.         if (buf[len-1] == '\') {
  59.             len--;
  60.         } else {
  61.             buf[len++] = 'r';
  62.             buf[len++] = 'n';
  63.         }
  64.         assert(SOCK_Write(sock, buf, len, &n_written, eIO_WritePersist) ==
  65.                eIO_Success);
  66.     }
  67.     assert(feof(fp));
  68. }
  69. static void s_ReadData(SOCK sock)
  70. {
  71.     char   buf[512];
  72.     size_t n_read;
  73.     while (SOCK_Read(sock, buf, sizeof(buf), &n_read, eIO_ReadPlain) ==
  74.            eIO_Success) {
  75.         continue;
  76.     }
  77.     assert(SOCK_Status(sock, eIO_Read) == eIO_Closed);
  78. }
  79. /* Main function
  80.  */
  81. extern int main(int argc, char** argv)
  82. {
  83.     EIO_Status status;
  84.     SOCK  sock;
  85.     int   port;
  86.     FILE* fp;
  87.     /* Cmd.-line args */
  88.     if (argc != 4  ||
  89.         (fp = fopen(argv[1], "r")) == 0  ||
  90.         (port = atoi(argv[3])) <= 0) {
  91.         perror("nUSAGE:  test_fw <inp_file> <http_host> <http_port>n");
  92.         return 1;
  93.     }
  94.     /* Error logging */
  95.     {{
  96.         FILE* log_fp = fopen("test_fw.log", "a");
  97.         if ( !fp ) {
  98.             perror("Failed to open "test_fw.log" for writing");
  99.             return 2;
  100.         }
  101.         CORE_SetLOGFILE(log_fp, 1/*true*/);
  102.     }}
  103.     SOCK_SetDataLoggingAPI(eOn);
  104.     /* Connect to Web server */
  105.     status = SOCK_Create(argv[2], (unsigned short) port, 0, &sock);
  106.     assert(status == eIO_Success);
  107.     /* Send data from inp.file to the socket */
  108.     s_SendData(fp, sock);
  109.     fclose(fp);
  110.     /* Read up response from the socket */
  111.     s_ReadData(sock);
  112.     /* Close and exit */
  113.     status = SOCK_Close(sock);
  114.     assert(status == eIO_Success  ||  status == eIO_Closed);
  115.     assert(SOCK_ShutdownAPI() == eIO_Success);
  116.     CORE_SetLOG(0);
  117.     return 0;
  118. }
  119. /*
  120.  * ---------------------------------------------------------------------------
  121.  * $Log: test_fw.c,v $
  122.  * Revision 1000.0  2003/10/29 17:03:19  gouriano
  123.  * PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.8
  124.  *
  125.  * Revision 6.8  2002/12/04 16:59:51  lavr
  126.  * Open log file in append mode
  127.  *
  128.  * Revision 6.7  2002/10/17 20:49:25  lavr
  129.  * SOCK_Write(): forgotten write mode argument added
  130.  *
  131.  * Revision 6.6  2002/08/07 16:38:08  lavr
  132.  * EIO_ReadMethod enums changed accordingly; log moved to end
  133.  *
  134.  * Revision 6.5  2002/03/22 19:46:22  lavr
  135.  * Test_assert.h made last among the include files
  136.  *
  137.  * Revision 6.4  2002/01/16 21:23:14  vakatov
  138.  * Utilize header "test_assert.h" to switch on ASSERTs in the Release mode too
  139.  *
  140.  * Revision 6.3  2001/09/10 21:28:11  lavr
  141.  * Typo fix
  142.  *
  143.  * Revision 6.2  2001/03/02 20:03:48  lavr
  144.  * Typos fixed
  145.  *
  146.  * Revision 6.1  2000/11/17 22:36:22  vakatov
  147.  * Initial revision
  148.  *
  149.  * ===========================================================================
  150.  */