/* -*- mode: C; c-basic-offset: 4 -*- */
/* ex: set shiftwidth=4 tabstop=4 expandtab: */
/*
 * Copyright (c) 2011-2013, Georgia Tech Research Corporation
 * All rights reserved.
 *
 * Author(s): Neil T. Dantam <ntd@gatech.edu>
 * Georgia Tech Humanoid Robotics Lab
 * Under Direction of Prof. Mike Stilman <mstilman@cc.gatech.edu>
 *
 *
 * This file is provided under the following "BSD-style" License:
 *
 *
 *   Redistribution and use in source and binary forms, with or
 *   without modification, are permitted provided that the following
 *   conditions are met:
 *
 *   * Redistributions of source code must retain the above copyright
 *     notice, this list of conditions and the following disclaimer.
 *
 *   * Redistributions in binary form must reproduce the above
 *     copyright notice, this list of conditions and the following
 *     disclaimer in the documentation and/or other materials provided
 *     with the distribution.
 *
 *   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
 *   CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES,
 *   INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF
 *   MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 *   DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
 *   CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 *   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 *   LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
 *   USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 *   AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 *   LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 *   ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 *   POSSIBILITY OF SUCH DAMAGE.
 *
 */

#ifdef HAVE_CONFIG
#include "config.h"
#endif //HAVE_CONFIG

#include <ipcbench.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/un.h>

#define PORT 8070
#define UNIX_PATH_MAX    108

static int *sock;
static struct sockaddr_un *addr;

#define NAME "/tmp/ipcbench-%lu.dsock"


static void s_init(void) {
    size_t i;
    for( i = 0; i < ipcbench_cnt; i ++ ) {
        char buf[64];
        snprintf(buf, sizeof(buf), NAME, i);
        unlink( buf );
    }
    sock = (int*)calloc(ipcbench_cnt, sizeof(int));
    addr = (struct sockaddr_un*)calloc(ipcbench_cnt, sizeof(struct sockaddr_un));
}

static void s_init_send(void) {
    size_t i;
    for( i = 0; i < ipcbench_cnt; i ++ ) {
        sock[i] = socket( PF_UNIX, SOCK_DGRAM, 0 );
        if( sock[i] < 0 ) {
            perror( "Could not create socket");
            abort();
        }

        char buf[64];
        snprintf(buf, sizeof(buf), NAME, i);
        addr[i].sun_family = AF_UNIX;
        snprintf(addr[i].sun_path, UNIX_PATH_MAX, buf );
    }
}

static void s_init_recv(void) {
    s_init_send();

    size_t i;
    for( i = 0; i < ipcbench_cnt; i ++ ) {
        if (bind(sock[i], (struct sockaddr *) &addr[i], sizeof(struct sockaddr_un)) < 0) {
            perror("Failed to bind the socket");
            abort();
        }
        ipcbench_pfd[i].fd = sock[i];
    }
}

static void s_send( const struct timespec *ts ) {
    size_t i = pubnext();
    ssize_t r = sendto( sock[i], ts, sizeof(*ts), 0,
                        (struct sockaddr *) &addr[i], sizeof(struct sockaddr_un) );
    if( sizeof(*ts) != r ) {
        fprintf(stderr, "error on socket %lu\n", i);
        perror( "could not send data " );
        abort();
    }
}

static void s_recv( struct timespec *ts ) {
    size_t i = pollin();
    ssize_t r = recvfrom( sock[i], ts, sizeof(*ts), 0,
                          NULL, 0 );
    if( sizeof(*ts) != r ) {
        perror( "could not receive data on pipe" );
        abort();
    }
}

struct ipcbench_vtab ipc_bench_vtab_local_dgram = {
    .init = s_init,
    .init_send = s_init_send,
    .init_recv = s_init_recv,
    .send = s_send,
    .recv = s_recv
};