test_fw.c
上传用户:yhdzpy8989
上传日期:2007-06-13
资源大小:13604k
文件大小:5k
- /*
- * ===========================================================================
- * PRODUCTION $Log: test_fw.c,v $
- * PRODUCTION Revision 1000.0 2003/10/29 17:03:19 gouriano
- * PRODUCTION PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.8
- * PRODUCTION
- * ===========================================================================
- */
- /* $Id: test_fw.c,v 1000.0 2003/10/29 17:03:19 gouriano Exp $
- * ===========================================================================
- *
- * PUBLIC DOMAIN NOTICE
- * National Center for Biotechnology Information
- *
- * This software/database is a "United States Government Work" under the
- * terms of the United States Copyright Act. It was written as part of
- * the author's official duties as a United States Government employee and
- * thus cannot be copyrighted. This software/database is freely available
- * to the public for use. The National Library of Medicine and the U.S.
- * Government have not placed any restriction on its use or reproduction.
- *
- * Although all reasonable efforts have been taken to ensure the accuracy
- * and reliability of the software and data, the NLM and the U.S.
- * Government do not and cannot warrant the performance or results that
- * may be obtained by using this software or data. The NLM and the U.S.
- * Government disclaim all warranties, express or implied, including
- * warranties of performance, merchantability or fitness for any particular
- * purpose.
- *
- * Please cite the author in any work or product based on this material.
- *
- * ===========================================================================
- *
- * Author: Denis Vakatov
- *
- * File Description:
- * Read data from a file, and send these data line-by-line to a socket.
- * Then, read data from the socket.
- * Trace all I/O going through the socket in both directions.
- *
- */
- #include <connect/ncbi_socket.h>
- #include <connect/ncbi_util.h>
- #include <stdlib.h>
- #include <string.h>
- /* This header must go last */
- #include "test_assert.h"
- static void s_SendData(FILE* fp, SOCK sock)
- {
- char buf[2048];
- size_t n_written;
- while ( fgets(buf, sizeof(buf)-3, fp) ) {
- size_t len = strlen(buf);
- assert(buf[len - 1] == 'n');
- assert(buf[len] == ' ');
- assert(!feof(fp) && !ferror(fp));
- len--;
- if (buf[len-1] == '\') {
- len--;
- } else {
- buf[len++] = 'r';
- buf[len++] = 'n';
- }
- assert(SOCK_Write(sock, buf, len, &n_written, eIO_WritePersist) ==
- eIO_Success);
- }
- assert(feof(fp));
- }
- static void s_ReadData(SOCK sock)
- {
- char buf[512];
- size_t n_read;
- while (SOCK_Read(sock, buf, sizeof(buf), &n_read, eIO_ReadPlain) ==
- eIO_Success) {
- continue;
- }
- assert(SOCK_Status(sock, eIO_Read) == eIO_Closed);
- }
- /* Main function
- */
- extern int main(int argc, char** argv)
- {
- EIO_Status status;
- SOCK sock;
- int port;
- FILE* fp;
- /* Cmd.-line args */
- if (argc != 4 ||
- (fp = fopen(argv[1], "r")) == 0 ||
- (port = atoi(argv[3])) <= 0) {
- perror("nUSAGE: test_fw <inp_file> <http_host> <http_port>n");
- return 1;
- }
- /* Error logging */
- {{
- FILE* log_fp = fopen("test_fw.log", "a");
- if ( !fp ) {
- perror("Failed to open "test_fw.log" for writing");
- return 2;
- }
- CORE_SetLOGFILE(log_fp, 1/*true*/);
- }}
- SOCK_SetDataLoggingAPI(eOn);
- /* Connect to Web server */
- status = SOCK_Create(argv[2], (unsigned short) port, 0, &sock);
- assert(status == eIO_Success);
- /* Send data from inp.file to the socket */
- s_SendData(fp, sock);
- fclose(fp);
- /* Read up response from the socket */
- s_ReadData(sock);
- /* Close and exit */
- status = SOCK_Close(sock);
- assert(status == eIO_Success || status == eIO_Closed);
- assert(SOCK_ShutdownAPI() == eIO_Success);
- CORE_SetLOG(0);
- return 0;
- }
- /*
- * ---------------------------------------------------------------------------
- * $Log: test_fw.c,v $
- * Revision 1000.0 2003/10/29 17:03:19 gouriano
- * PRODUCTION: IMPORTED [ORIGINAL] Dev-tree R6.8
- *
- * Revision 6.8 2002/12/04 16:59:51 lavr
- * Open log file in append mode
- *
- * Revision 6.7 2002/10/17 20:49:25 lavr
- * SOCK_Write(): forgotten write mode argument added
- *
- * Revision 6.6 2002/08/07 16:38:08 lavr
- * EIO_ReadMethod enums changed accordingly; log moved to end
- *
- * Revision 6.5 2002/03/22 19:46:22 lavr
- * Test_assert.h made last among the include files
- *
- * Revision 6.4 2002/01/16 21:23:14 vakatov
- * Utilize header "test_assert.h" to switch on ASSERTs in the Release mode too
- *
- * Revision 6.3 2001/09/10 21:28:11 lavr
- * Typo fix
- *
- * Revision 6.2 2001/03/02 20:03:48 lavr
- * Typos fixed
- *
- * Revision 6.1 2000/11/17 22:36:22 vakatov
- * Initial revision
- *
- * ===========================================================================
- */