sys/t/mdup-test.c: Cope with Cygwin setting top inode-number bits.

[mLib] / sys / t / mdup-test.c

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
#include "mdup.h"

#define MAXFD 256

/* For some reason, Cygwin has started leaving cruft in the top bits of inode
 * numbers, which breaks this test.  Mask them off.
 */
#if defined(__CYGWIN__) && defined(__amd64__)
#  define HACK_INODE(ino) ((ino)&0xffffffffffff)
#else
#  define HACK_INODE(ino) (ino)
#endif

static void fail(const char *what) { perror(what); exit(1); }

int main(int argc, char *argv[])
{
  int i, n, j;
  int fd, fd2;
  struct stat st;
  int ino[MAXFD];
  int flag[MAXFD];
  mdup_fd fds[MAXFD];
  int win = 1;
  int verbose = 0;

  for (i = 1, j = 0; i < argc; i++) {
    if (strcmp(argv[i], "-v") == 0) { verbose++; continue; }
    if (j >= MAXFD) { fprintf(stderr, "too many\n"); exit(1); }
    if (sscanf(argv[i], "%d:%d", &fds[j].cur, &fds[j].want) < 2 ||
	fds[j].cur >= MAXFD)
      { fprintf(stderr, "bad syntax\n"); exit(1); }
    j++;
  }
  n = j;
  for (i = 0; i < MAXFD; i++) flag[i] = -1;
  for (i = 0; i < n; i++) {
    fd = fds[i].cur;
    if (flag[fd] >= 0) {
      ino[i] = ino[flag[fd]];
      if (verbose)
	printf("exist fd %d[%d] = ino %lu\n", fd, i, (unsigned long)ino[i]);
    } else {
      flag[fd] = i;
      if ((fd2 = open(",delete-me",
		      O_WRONLY | O_CREAT | O_EXCL,
		      0700)) < 0)
	fail("creat");
      unlink(",delete-me");
      if (fd2 != fd) { if (dup2(fd2, fd) < 0) fail("dup2"); close(fd2); }
      if (fstat(fd, &st)) fail("fstat");
      ino[i] = HACK_INODE(st.st_ino);
      if (verbose)
	printf("open fd %d[%d] = ino %lu\n", fd, i, (unsigned long)ino[i]);
    }
  }

  if (mdup(fds, n)) fail("mdup");

  for (i = 0; i < n; i++) {
    fd = fds[i].cur;
    if (fds[i].want != -1 && fds[i].want != fd) {
      printf("fd %d[%d] != %d\n", fd, i, fds[i].want);
      win = 0;
    } else if (fstat(fd, &st)) {
      printf("fstat %d[%d] failed: %s\n", fd, i, strerror(errno));
      win = 0;
    } else if (HACK_INODE(st.st_ino) != ino[i]) {
      if (!verbose) printf("ino %d[%d] wrong\n", fd, i);
      else printf("ino %d[%d] = %lu != %lu\n", fd, i,
		  (unsigned long)HACK_INODE(st.st_ino),
		  (unsigned long)ino[i]);
      win = 0;
    }
  }

  return (!win);
}