shithub: lwext4

ref: 0d2c2b5509e8489b94dfe782eb1bf455fa86a493
dir: /fs_test/lwext4_client.c/

#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <unistd.h>
#include <errno.h>
#include <stdbool.h>
#include <getopt.h>

#ifdef WIN32
#include <windows.h>
#include <winsock2.h>
#include <ws2tcpip.h>

static int inet_pton(int af, const char *src, void *dst);

#else
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <sys/types.h>
#endif

static int winsock_init(void);
static void winsock_fini(void);

/**@brief   Default server addres.*/
static char *server_addr = "127.0.0.1";

/**@brief   Default connection port.*/
static int connection_port = 1234;

/**@brief   Call op*/
static char *op_code;

static const char *usage = "                                    \n\
Welcome in lwext4_client.                                       \n\
Copyright (c) 2013 Grzegorz Kostka (kostka.grzegorz@gmail.com)  \n\
Usage:                                                          \n\
    --call (-c) - call opt                                      \n\
    --port (-p) - server port                                   \n\
    --addr (-a) - server ip address                             \n\
\n";



static int client_connect(void)
{
    int fd = 0;
    struct sockaddr_in serv_addr;

    if(winsock_init() < 0) {
        printf("winsock_init error\n");
        exit(-1);
    }

    memset(&serv_addr, '0', sizeof(serv_addr));
    fd = socket(AF_INET, SOCK_STREAM, 0);
    if(fd < 0) {
        printf("socket() error: %s\n", strerror(errno));
        exit(-1);
    }

    serv_addr.sin_family = AF_INET;
    serv_addr.sin_port = htons(connection_port);

    if(!inet_pton(AF_INET, server_addr, &serv_addr.sin_addr)){
        printf("inet_pton() error\n");
        exit(-1);
    }

    if(connect(fd, (struct sockaddr *)&serv_addr, sizeof(serv_addr))){
        printf("connect() error: %s\n", strerror(errno));
        exit(-1);
    }

    return fd;
}



static bool parse_opt(int argc, char **argv)
{
    int option_index = 0;
    int c;

    static struct option long_options[] =
    {
            {"call",    required_argument, 0, 'c'},
            {"port",    required_argument, 0, 'p'},
            {"addr",    required_argument, 0, 'a'},
            {0, 0, 0, 0}
    };

    while(-1 != (c = getopt_long (argc, argv, "c:p:a:", long_options, &option_index))) {

        switch(c){
        case 'a':
            server_addr = optarg;
            break;
        case 'p':
            connection_port = atoi(optarg);
            break;
        case 'c':
            op_code = optarg;
            break;
        default:
            printf("%s", usage);
            return false;

        }
    }
    return true;
}


int main(int argc, char *argv[])
{
    int sockfd;
    int n;
    int rc;
    char recvBuff[1024];

    if(!parse_opt(argc, argv))
        return -1;

    sockfd = client_connect();


    n = send(sockfd, op_code, strlen(op_code), 0);
    if(n < 0) {
        printf("\tWrite error: %s fd = %d\n", strerror(errno), sockfd);
        return -1;
    }

    n = recv(sockfd, (void *)&rc, sizeof(rc), 0);
    if(n < 0) {
        printf("\tWrite error: %s fd = %d\n", strerror(errno), sockfd);
        return -1;
    }

    printf("rc: %d %s\n", rc, strerror(rc));
    if(rc)
        printf("\t%s\n",op_code);

    return rc;
}

static int winsock_init(void)
{
#if WIN32
    int rc;
    static WSADATA wsaData;
    rc = WSAStartup(MAKEWORD(2,2), &wsaData);
    if (rc != 0) {
        return -1;
    }
#endif
    return 0;
}

static void winsock_fini(void)
{
#if WIN32
    WSACleanup();
#endif
}


#if WIN32
static int inet_pton(int af, const char *src, void *dst)
{
    struct sockaddr_storage ss;
    int size = sizeof(ss);
    char src_copy[INET6_ADDRSTRLEN+1];

    ZeroMemory(&ss, sizeof(ss));
    /* stupid non-const API */
    strncpy (src_copy, src, INET6_ADDRSTRLEN+1);
    src_copy[INET6_ADDRSTRLEN] = 0;

    if (WSAStringToAddress(src_copy, af, NULL, (struct sockaddr *)&ss, &size) == 0) {
        switch(af) {
        case AF_INET:
            *(struct in_addr *)dst = ((struct sockaddr_in *)&ss)->sin_addr;
            return 1;
        case AF_INET6:
            *(struct in6_addr *)dst = ((struct sockaddr_in6 *)&ss)->sin6_addr;
            return 1;
        }
    }
    return 0;
}
#endif