#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(void)
{
 printf("================\n");
}

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

 to_do = malloc(sizeof(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");
    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);
       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)
	{ 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 (out of range 1..%d)\n",__progname,av[i],n_tests);
	  exit(1);
	}
       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();
    (*tests[i]->run)();
    sleep (1);
    checkstats();
    if (test_failed()) break;
  }
 sepline();
 return(0);
}