src/vfer_test_client.c

Go to the documentation of this file.

/*
 * Copyright 2005, 2006, Internet2
 * Legal conditions are in file LICENSE
 * (MD5 = c434f2e53b8089d8b4d0172c7ce07360).
 */
/**
 * @file   vfer_test_client.c
 * @author Ivan Beschastnikh
 * @brief  Test: A connection client that connects to the server and send a small buffer.
 *
 * This test tests c_control.c and api.c by setting up a socket for
 * connection, connecting to a server and sending the server a
 * constant size buffer(arg to program) a variable number of times
 * (arg to program), sleeping in between each for some time (also arg
 * to program).
 *
 * -     08/05/05        ivan            Added extra cmd line options to control interval, size of frames, number of frames
 * -     07/30/05        ivan            Added to the CVS tree (old test_client.c retired)
 */

#include <stdio.h>
#include <netdb.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <sys/select.h>
#include "vfer_tests.h"
#include "vfer_api.h"

vfer_fd connect_skt;
int port;
int num_frames;
int frame_size;
int ms_interval;

/*
 * main()
 */
int main(int argc, char** argv) {

        struct sockaddr_in client_sa;
        struct sockaddr_in server_sa;
        char hostname_buf[256];
        char hostname[INET_ADDRSTRLEN];
        struct hostent* h;
        char *buffer;
        struct timeval interval;
        int i;
        int ret;

        if (argc != 6) {
                printf("usage: %s [server_ip] [server_port] [num_frames] [frame_size] [interval]\n", argv[0]);
                printf("\tserver_ip  : server ip address (do not use hostname or the address 127.0.0.1)\n");
                printf("\tserver_port: server port to connect to\n");
                printf("\tnum_frames : total number of frames to send (< 0 => no bound)\n");
                printf("\tframe_size : size of each frame to send\n");
                printf("\tinternval  : interval in milliseconds between each frames\n");
                return -1;
        }

        TEST_INIT("TEST_CLIENT", "test_client.c", stdout, stdout);
        vfer_debug(stdout, stdout, "scap");
        
        port            = atoi(argv[2]);
        num_frames      = atoi(argv[3]);
        if ((frame_size = atoi(argv[4])) < 0) {
                TEST_PRINT("main","ERROR: negative frame size");
                return -1;
        }
        if ((ms_interval        = atoi(argv[5])) < 0) {
                TEST_PRINT("main","ERROR: invalid interval");
                return -1;
        }

        if ((buffer = calloc(1, frame_size)) == NULL) {
                TEST_PRINT("main","ERROR: calloc failed");
                return -1;
        }

        /* find out local addr */
        if (gethostname(hostname_buf, 256) == -1) {
                perror("gethostname");
                return -1;
        }

        if ((h = gethostbyname((const char*)hostname_buf)) == NULL) {
                perror("gethostbyname");
                return -1;
        }

        if (h->h_addr_list != 0) {
                inet_ntop(h->h_addrtype, (h->h_addr_list)[0], hostname, sizeof(hostname));
        }
        
        TEST_PRINT("main","my ip: %s", hostname);

        /* setup sockaddr structure for client */
        bzero((char*)(&client_sa), sizeof(client_sa));
        client_sa.sin_port = htons(0);
        client_sa.sin_family = AF_INET;
        if (inet_pton(AF_INET, hostname, &client_sa.sin_addr) != 1) {
                TEST_PRINT("main","inet_pton failed");
                return -1;
        }

        /* setup sockaddr structure for server */
        bzero((char*)(&server_sa), sizeof(server_sa));
        server_sa.sin_family = AF_INET;
        server_sa.sin_port = htons(port);

        if (inet_pton(AF_INET, argv[1], &server_sa.sin_addr) != 1) {
                TEST_PRINT("main","inet_pton failed");
                return -1;
        }
        
        if ((connect_skt = vfer_socket (SOCK_DGRAM)) < 0) {        
                TEST_PRINT("main","vfer_socket failed");
                return -1;
        }

        if ((ret = vfer_bind  (connect_skt, (struct sockaddr*) &client_sa, sizeof(client_sa))) < 0) {
                TEST_PRINT("main","vfer_bind failed [%s]", vfer_errortext(ret));
                return -1;
        }

        if ((ret = vfer_connect (connect_skt, (SA*) (&server_sa), sizeof(server_sa))) < 0) {
                TEST_PRINT("main","vfer_connect failed [%s]", vfer_errortext(ret));
                return -1;
        }

        TEST_PRINT("main","connected!");
        if (num_frames < 0) {
                while (1) {
                        if ((ret = vfer_send (connect_skt, buffer, frame_size)) <= 0) {
                                TEST_PRINT("main","vfer_send returned[%d] [%s]", ret, vfer_errortext(ret));
                                vfer_close(connect_skt);
                                return -1;
                        }
                        interval.tv_sec = 0;
                        interval.tv_usec = ms_interval * 1000;
                        select(0,0,0,0,&interval);
                }
        } else {
                for (i = 0; i < num_frames; i++) {
                        TEST_PRINT("main","sending frame[%d]", i);
                        if ((ret = vfer_send (connect_skt, buffer, frame_size)) <= 0) {
                                TEST_PRINT("main","vfer_send returned[%d] [%s] on frame[%d]", ret, vfer_errortext(ret), i);
                                vfer_close(connect_skt);
                                return -1;
                        }
                        if (i != num_frames - 1) {
                                interval.tv_sec = 0;
                                interval.tv_usec = ms_interval * 1000;
                                select(0,0,0,0,&interval);
                        }
                }
        }
        sleep(10);              /* allow packets to be processed */
        vfer_close(connect_skt);
        return 0;
} /* main() */          

---