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

extern const char *__progname;

#include "test.h"

static char *to_do;

static void print_test_list(void)
{
 int i;

 for (i=0;i<n_tests;i++)
  { printf("%d\t%s\n",i+1,test_names[i]);
  }
}

static unsigned long long int raw_get(int base)
{
 errno = 0;
 return( (0x100000000ULL * (0xffffffff & (unsigned int)pidconn(PIDCONN_DEBUG,base,0))) |
	 (1ULL * (0xffffffff & (unsigned int)pidconn(PIDCONN_DEBUG,base+1,0))) );
 if (errno) fprintf(stderr,"%s: can't get stats, base %d: %s\n",__progname,base,strerror(errno));
}

static unsigned long long int get_stat(int base)
{
 unsigned long long int v1;
 unsigned long long int v2;

 v2 = raw_get(base);
 do
  { v1 = v2;
    v2 = raw_get(base);
  } while (v1 != v2);
 return(v1);
}

static void checkstats(void)
{
 unsigned long long int stat_get;
 unsigned long long int stat_put;

 stat_get = get_stat(0);
 stat_put = get_stat(2);
 if (stat_get != stat_put)
  { printf("pidconn leak: gets %llu, puts %llu\n",stat_get,stat_put);
    exit(1);
  }
 stat_get = get_stat(4);
 stat_put = get_stat(6);
 if (stat_get != stat_put)
  { printf("bufpair leak: gets %llu, puts %llu\n",stat_get,stat_put);
    exit(1);
  }
}

static void sepline(int i)
{
 if (i < 0)
  { printf("================\n");
  }
 else
  { printf("================ %d %s\n",i+1,test_names[i]);
  }
}

int main(int, char **);
int main(int ac, char **av)
{
 int i;
 int j;
 char *ep;
 char *ep2;
 long int li;
 long int li2;

 to_do = malloc(n_tests);
 bzero(to_do,n_tests);
 if (ac < 2)
  { printf("arg is a list of test number or names to run, or\n");
    printf("`list' to list test numbers and names, or\n");
    printf("`all' to run all tests\n");
    printf("N+ means test N and all later tests\n");
    printf("N-M means tests [N..M] (provided N < M)\n");
    exit(1);
  }
 else if ((ac == 2) && !strcmp(av[1],"list"))
  { print_test_list();
    exit(0);
  }
 else if ((ac == 2) && !strcmp(av[1],"all"))
  { memset(to_do,1,n_tests);
  }
 else
  { for <"arg"> (i=1;av[i];i++)
     { for (j=n_tests-1;j>=0;j--)
	{ if (! strcmp(av[i],test_names[j]))
	   { to_do[j] = 1;
	     continue <"arg">;
	   }
	}
       li = strtol(av[i],&ep,0);
       li2 = li;
       if (ep == av[i])
	{ fprintf(stderr,"%s: bad arg %s (not a name and no number found)\n",__progname,av[i]);
	  exit(1);
	}
       if ((ep[0] == '+') && !ep[1])
	{ li2 = n_tests;
	}
       else if (ep[0] == '-')
	{ li2 = strtol(ep+1,&ep2,0);
	  if (ep2 == ep+1)
	   { fprintf(stderr,"%s: bad range %s (no number after -)\n",__progname,av[i]);
	     exit(1);
	   }
	  if (*ep2)
	   { fprintf(stderr,"%s: bad range %s (junk after second number)\n",__progname,av[i]);
	     exit(1);
	   }
	  if (li2 < li)
	   { fprintf(stderr,"%s: bad range %s (endpoints out of order)\n",__progname,av[i]);
	     exit(1);
	   }
	}
       else if (*ep)
	{ fprintf(stderr,"%s: bad arg %s (not a name and junk after number)\n",__progname,av[i]);
	  exit(1);
	}
       if ((li < 1) || (li > n_tests))
	{ fprintf(stderr,"%s: bad arg %s (%ld out of range 1..%d)\n",__progname,av[i],li,n_tests);
	  exit(1);
	}
       if ((li2 < 1) || (li2 > n_tests))
	{ fprintf(stderr,"%s: bad arg %s (%ld out of range 1..%d)\n",__progname,av[i],li2,n_tests);
	  exit(1);
	}
       for (;li<=li2;li++) to_do[li-1] = 1;
     }
  }
 test_init();
 pidconn(PIDCONN_DEBUG,8,0);
 for (i=0;i<n_tests;i++)
  { if (! to_do[i]) continue;
    sleep(1);
    sepline(i);
    (*tests[i]->run)();
    if (test_failed())
     { sleep(1);
       printf("Exiting on test failure\n");
       break;
     }
    sleep(1);
    checkstats();
  }
 sepline(-1);
 return(0);
}