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

生物技术

开发平台:

C/C++

  1. /*
  2.  * ===========================================================================
  3.  * PRODUCTION $Log: test_ncbi_connutil_hit.c,v $
  4.  * PRODUCTION Revision 1000.0  2003/10/29 17:03:52  gouriano
  5.  * PRODUCTION PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.12
  6.  * PRODUCTION
  7.  * ===========================================================================
  8.  */
  9. /*  $Id: test_ncbi_connutil_hit.c,v 1000.0 2003/10/29 17:03:52 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.  *   Test "ncbi_connutil.c"::URL_Connect()
  38.  *
  39.  */
  40. #include <connect/ncbi_connutil.h>
  41. #include <connect/ncbi_util.h>
  42. #include <stdlib.h>
  43. /* This header must go last */
  44. #include "test_assert.h"
  45. int main(int argc, char** argv)
  46. {
  47.     /* Prepare to connect:  parse and check cmd.-line args, etc. */
  48.     const char*    host        = (argc > 1) ? argv[1] : "";
  49.     unsigned short port        = (argc > 2) ? (unsigned short) atoi(argv[2]):0;
  50.     const char*    path        = (argc > 3) ? argv[3] : "";
  51.     const char*    args        = (argc > 4) ? argv[4] : "";
  52.     const char*    inp_file    = (argc > 5) ? argv[5] : "";
  53.     const char*    user_header = (argc > 6) ? argv[6] : "";
  54.     size_t   content_length;
  55.     STimeout timeout;
  56.     SOCK sock;
  57.     EIO_Status status;
  58.     char buffer[10000];
  59.     CORE_SetLOGFILE(stderr, 0/*false*/);
  60.     fprintf(stderr, "Running...n"
  61.             "  Executable:      '%s'n"
  62.             "  URL host:        '%s'n"
  63.             "  URL port:         %hun"
  64.             "  URL path:        '%s'n"
  65.             "  URL args:        '%s'n"
  66.             "  Input data file: '%s'n"
  67.             "  User header:     '%s'n"
  68.             " Reply(if any) from the hit URL goes to the standard output.nn",
  69.             argv[0],
  70.             host, (unsigned short) port, path, args, inp_file, user_header);
  71.     if ( argc < 4 ) {
  72.         fprintf(stderr,
  73.                 "Usage:   %s host port path args inp_file [user_header]n"
  74.                 "Example: %s ............n"
  75.                 "nTwo few arguments.n",
  76.                 argv[0], argv[0]);
  77.         return 1;
  78.     }
  79.     {{
  80.         FILE *fp = fopen(inp_file, "rb");
  81.         long offset;
  82.         if ( !fp ) {
  83.             fprintf(stderr, "Non-existent file '%s'n", inp_file);
  84.             return 2;
  85.         }
  86.         if ( fseek(fp, 0, SEEK_END) != 0  ||  (offset = ftell(fp)) < 0 ) {
  87.             fprintf(stderr, "Cannot obtain size of file '%s'n", inp_file);
  88.             return 2;
  89.         }
  90.         fclose(fp);
  91.         content_length = (size_t) offset;
  92.     }}
  93.     timeout.sec  = 10;
  94.     timeout.usec = 0;
  95.     
  96.     /* Connect */
  97.     sock = URL_Connect(host, port, path, args,
  98.                        eReqMethod_Any, content_length,
  99.                        &timeout, &timeout, user_header, 1/*true*/, eDefault);
  100.     if ( !sock )
  101.         return 3;
  102.     
  103.     {{ /* Pump data from the input file to socket */
  104.         FILE* fp = fopen(inp_file, "rb");
  105.         if ( !fp ) {
  106.             fprintf(stderr, "Cannot open file '%s' for readingn", inp_file);
  107.             return 4;
  108.         }
  109.         for (;;) {
  110.             size_t n_written;
  111.             size_t n_read = fread(buffer, 1, sizeof(buffer), fp);
  112.             if ( n_read <= 0 ) {
  113.                 if ( content_length ) {
  114.                     fprintf(stderr,
  115.                             "Cannot read last %lu bytes from file '%s'n",
  116.                             (unsigned long) content_length, inp_file);
  117.                     return 5;
  118.                 }
  119.                 break;
  120.             }
  121.             assert(content_length >= n_read);
  122.             content_length -= n_read;
  123.             status = SOCK_Write(sock, buffer, n_read,
  124.                                 &n_written, eIO_WritePersist);
  125.             if ( status != eIO_Success ) {
  126.                 fprintf(stderr, "Error writing to socket(%s)n",
  127.                         IO_StatusStr(status));
  128.                 return 6;
  129.             }
  130.         }
  131.         fclose(fp);
  132.     }}
  133.     /* Read reply from socket, write it to STDOUT */
  134.     {{
  135.         size_t n_read;
  136.         for (;;) {
  137.             status = SOCK_Read(sock, buffer, sizeof(buffer), &n_read,
  138.                                eIO_ReadPlain);
  139.             if (status != eIO_Success)
  140.                 break;
  141.             fwrite(buffer, 1, n_read, stdout);
  142.         }
  143.         if ( status != eIO_Closed ) {
  144.             fprintf(stderr,
  145.                     "Error occurred after reading %ld bytes from socket(%s)n",
  146.                     (long) content_length, IO_StatusStr(status));
  147.         }
  148.         fprintf(stdout, "n");
  149.     }}
  150.     /* Success:  close the socket, cleanup, and exit */
  151.     SOCK_Close(sock);
  152.     CORE_SetLOG(0);
  153.     return 0;
  154. }
  155. /*
  156.  * ---------------------------------------------------------------------------
  157.  * $Log: test_ncbi_connutil_hit.c,v $
  158.  * Revision 1000.0  2003/10/29 17:03:52  gouriano
  159.  * PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.12
  160.  *
  161.  * Revision 6.12  2002/08/12 15:10:43  lavr
  162.  * Use persistent SOCK_Write()
  163.  *
  164.  * Revision 6.11  2002/08/07 16:38:08  lavr
  165.  * EIO_ReadMethod enums changed accordingly; log moved to end
  166.  *
  167.  * Revision 6.10  2002/04/15 19:21:44  lavr
  168.  * +#include "../test/test_assert.h"
  169.  *
  170.  * Revision 6.9  2002/03/22 19:48:56  lavr
  171.  * Removed <stdio.h>: included from ncbi_util.h or ncbi_priv.h
  172.  *
  173.  * Revision 6.8  2002/02/05 21:45:55  lavr
  174.  * Included header files rearranged
  175.  *
  176.  * Revision 6.7  2001/03/02 20:03:57  lavr
  177.  * Typos fixed
  178.  *
  179.  * Revision 6.6  2001/01/23 23:21:21  lavr
  180.  * Added new argument to URL_Connect
  181.  *
  182.  * Revision 6.5  2001/01/08 22:42:14  lavr
  183.  * eRequestMethodAny -> eRequestMethod_Any
  184.  *
  185.  * Revision 6.4  2000/12/29 18:24:40  lavr
  186.  * File size is now discovered without use of stat call.
  187.  *
  188.  * Revision 6.3  2000/09/27 13:49:29  lavr
  189.  * URL_Connect args adjusted
  190.  *
  191.  * Revision 6.2  2000/03/29 17:21:47  vakatov
  192.  * + CORE_SetLOG(0) at the program end.
  193.  *
  194.  * Revision 6.1  2000/03/24 22:53:38  vakatov
  195.  * Initial revision
  196.  *
  197.  * ===========================================================================
  198.  */