pipes.c

#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>

#include "decls.h"

typedef struct state { int tx[2], rx[2]; } state;

state *new_state() { return calloc(sizeof(state), 1); }

void free_state(state *state) { free(state); }

int pre_fork_setup(state *state) {
  if (pipe(state->tx)) {
    // uh-oh
  }
  if (pipe(state->rx)) {
    // uh-oh
  }
  return 0;
}

int cleanup(state *state __attribute__((unused))) { return 0; }

int child_post_fork_setup(state *state) {
  close(state->tx[1]);
  close(state->rx[0]);

  return 0;
}

int child_warmup(int warmup_iters __attribute__((unused)),
                 state *state __attribute__((unused))) {
  return 0;
}

int child_loop(int iters __attribute__((unused)), state *state) {
  int tx_fd = state->rx[1];
  int rx_fd = state->tx[0];

  for (;;) {
    char msg;
    read(rx_fd, &msg, 1);
    write(tx_fd, "1", 1);
  }
  return 0;
}

int child_cleanup(state *state) {
  close(state->tx[0]);
  close(state->rx[1]);

  return 0;
}

int parent_post_fork_setup(state *state) {
  close(state->tx[0]);
  close(state->rx[1]);

  return 0;
}

int parent_warmup(int warmup_iters, state *state) {
  int i;
  int tx_fd = state->tx[1];
  int rx_fd = state->rx[0];

  for (i = 0; i < warmup_iters; ++i) {
    char resp;
    if (write(tx_fd, "0", 1) == -1) {
      perror("could not write");
      break;
    };

    if (read(rx_fd, &resp, 1) == -1) {
      perror("could not read");
      break;
    }
  }
  return 0;
}

int parent_loop(int iters, state *state) {
  int i;
  int tx_fd = state->tx[1];
  int rx_fd = state->rx[0];

  for (i = 0; i < iters; ++i) {
    char resp;
    if (write(tx_fd, "0", 1) == -1) {
      perror("could not write");
      break;
    };

    if (read(rx_fd, &resp, 1) == -1) {
      perror("could not read");
      break;
    }
  }
  return 0;
}

int parent_cleanup(state *state) {
  close(state->tx[1]);
  close(state->rx[0]);

  return 0;
}