#include <sys/types.h>
#include <time.h>
#include <sys/time.h>
#include <sys/resource.h>
#include "pathexec.h"
#include "sgetopt.h"
#include "strerr.h"
#include "scan.h"
#include "str.h"

#define FATAL "softlimit: fatal: "

void die_usage(void)
{
  strerr_die1x(100,"softlimit: usage: softlimit [-a allbytes] [-c corebytes] [-d databytes] [-f filebytes] [-l lockbytes] [-m membytes] [-o openfiles] [-p processes] [-r residentbytes] [-s stackbytes] [-t cpusecs] child");
}

void doit(int resource,const char *arg)
{
  unsigned long u;
  struct rlimit r;

  if (getrlimit(resource,&r) == -1)
	 strerr_die2sys(111,FATAL,"getrlimit failed: ");

  if (str_equal(arg,"="))
	 r.rlim_cur = r.rlim_max;
  else {
	 if (arg[scan_ulong(arg,&u)]) die_usage();
	 r.rlim_cur = u;
	 if (r.rlim_cur > r.rlim_max)
		r.rlim_cur = r.rlim_max;
  }

  if (setrlimit(resource,&r) == -1)
	 strerr_die2sys(111,FATAL,"setrlimit failed: ");
}

int main(int argc,const char *const *argv,const char *const *envp)
{
  int opt;

  while ((opt = getopt(argc,argv,"a:c:d:f:l:m:o:p:r:s:t:")) != opteof)
	 switch(opt) {
		case '?':
	die_usage();
		case 'a':
#ifdef RLIMIT_AS
		  doit(RLIMIT_AS,optarg);
#endif
#ifdef RLIMIT_VMEM
		  doit(RLIMIT_VMEM,optarg);
#endif
		  break;
		case 'c':
#ifdef RLIMIT_CORE
		  doit(RLIMIT_CORE,optarg);
#endif
		  break;
		case 'd':
#ifdef RLIMIT_DATA
		  doit(RLIMIT_DATA,optarg);
#endif
		  break;
		case 'f':
#ifdef RLIMIT_FSIZE
		  doit(RLIMIT_FSIZE,optarg);
#endif
		  break;
		case 'l':
#ifdef RLIMIT_MEMLOCK
		  doit(RLIMIT_MEMLOCK,optarg);
#endif
		  break;
		case 'm':
#ifdef RLIMIT_DATA
		  doit(RLIMIT_DATA,optarg);
#endif
#ifdef RLIMIT_STACK
		  doit(RLIMIT_STACK,optarg);
#endif
#ifdef RLIMIT_MEMLOCK
		  doit(RLIMIT_MEMLOCK,optarg);
#endif
#ifdef RLIMIT_VMEM
		  doit(RLIMIT_VMEM,optarg);
#endif
#ifdef RLIMIT_AS
		  doit(RLIMIT_AS,optarg);
#endif
	break;
		case 'o':
#ifdef RLIMIT_NOFILE
		  doit(RLIMIT_NOFILE,optarg);
#endif
#ifdef RLIMIT_OFILE
		  doit(RLIMIT_OFILE,optarg);
#endif
		  break;
		case 'p':
#ifdef RLIMIT_NPROC
		  doit(RLIMIT_NPROC,optarg);
#endif
		  break;
		case 'r':
#ifdef RLIMIT_RSS
		  doit(RLIMIT_RSS,optarg);
#endif
		  break;
		case 's':
#ifdef RLIMIT_STACK
		  doit(RLIMIT_STACK,optarg);
#endif
		  break;
		case 't':
#ifdef RLIMIT_CPU
		  doit(RLIMIT_CPU,optarg);
#endif
		  break;
	 }

  argv += optind;
  if (!*argv) die_usage();

  pathexec_run(*argv,argv,envp);
  strerr_die4sys(111,FATAL,"unable to run ",*argv,": ");
}