// tsplice.c - test splice syscall 
#include <stdio.h>
#include <string.h>
#include <stdlib.h>

#include <poll.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>



int 
sa_pton(int af, const char *cp, struct sockaddr *sa)
{
  sa->sa_family = af;
  return inet_pton(af, cp, &((struct sockaddr_in *) sa)->sin_addr);

}

void 
sa_set_port(struct sockaddr *sa, int port)
{
  ((struct sockaddr_in *) sa)->sin_port = htons(port);
}


int 
sa_ptonp(int af, const char *cp, struct sockaddr *sa)
{
  char *ports;
  int r;
  char *cp2 = alloca(strlen(cp) + 1);

  strcpy(cp2, cp);
  ports = strrchr(cp2, ':');
  if (ports)
    *ports = 0;

  sa->sa_family = af;
  r =  inet_pton(af, cp2, &((struct sockaddr_in *) sa)->sin_addr);
  if (r > 0)
    sa_set_port(sa, atoi(ports+1));

  return r;

}


void error(const char *m)
{
  perror(m);
  exit(1);
}

char data[] = "012345678";

#define DATASIZE (sizeof(data))

char  rbuf[256];

main(int argc, char *argv[])
{
  struct sockaddr sa;
  struct sockaddr lsa;
  int s1, s2, s3;
  int pipes[2];
  int r1, r2, r3;
  loff_t o1 = 0, o2 = 0;
  int alen = sizeof(lsa);

  sa_ptonp(AF_INET, argv[1], &sa);

  s1 = socket(AF_INET, SOCK_STREAM, 0);
  if (s1 < 0)
    error("socket");

  r1 = bind(s1, &sa, sizeof(sa));
  if (r1 < 0)
    error("bind");

  r1 = listen(s1, 1);
  if (r1 < 0)
    error("listen");


  s2 = socket(AF_INET, SOCK_STREAM, 0);
  if (s2 < 0)
    error("second socket");

  r1 = connect(s2, &sa, sizeof(sa));
  if (r1 < 0)
    error("connect");

  s3 = accept(s1, &lsa, &alen);
  if (s3 < 0)
    error("connect");

  pipe(pipes);
  
  r1 = write(s2, data, DATASIZE);
  r2 = splice(s3, &o1, pipes[1], &o2, DATASIZE, 0);
  if (r2 < 0)
    {
      error("splice syscall");
    }

  printf("r1=%d, r2=%d\n", r1, r2);


  r3 = read(pipes[0], rbuf, DATASIZE);


  printf("r1=%d, r2=%d, r3=%d, data=%s\n", r1, r2, r3, rbuf);
  

}