int main(int argc, char **argv)
{
int c, error = 0;
struct hist_data h;
int retlog;
char tmp[MAXBUFSIZ];
char *fullurl;
extern char *optarg;
extern int optind;
sigset_t set;
main_tid = pthread_self();
/* Allocate heap for download request
* struct request stores all the information that might be
* of interest
*/
req = (struct request *)calloc(1, sizeof(struct request));
/* This is required because if download is aborted before
* this is initialized, problems will occur
*/
wthread = NULL;
/* except from signal handler thread,
* no other thread is receiving signals!
* No signals are caught here since we need to save download job only
* after wthread structures are initialized.
* TODO:: We need to create helper thread as soon as head request is over
* since the user might wish to stop downloading as soon as he sees the
* content length.
*/
sigfillset(&set);
pthread_sigmask(SIG_BLOCK, &set, NULL);
/* Safe to exit before download starts.
* This is set back to DEFERRED before creating helper thread because
* we want this to exit only at certain safe points specified using
* pthread_testcancel().
* Helper thread is started just before creating worker threads and after
* wthread structures are initialized. Only after this point do we need to
* save download job if an interrupt comes - not before. Helper thread exists
* only to catch SIGINT and to save download job. It doesn't make sense to
* start it before wthread structures are initialized!
*/
pthread_setcanceltype(PTHREAD_CANCEL_DEFERRED, NULL);
/* Read the RC file if exist */
readrc();
/* If set in rc file */
if (preferredthread != -1)
nthreads = preferredthread;
while (!error && (c = getopt(argc,argv,"p:l:n:hfv")) != -1) {
switch(c) {
case 'p':
req->port = atoi(optarg);
break;
case 'l':
strncpy(req->lfile, optarg, MAXBUFSIZ -1);
break;
case 'n':
fsuggested = 1;
if ((nthreads = atoi(optarg)) > MAXTHREADS) {
Log("Error: Maximum # of threads allowed is %d\n", MAXTHREADS);
exit(1);
}
break;
case 'h':
printf("%s\n", PROGVERSION);
usage();
exit(0);
break;
case 'v':
printf("%s\nby Murat BALABAN <*****@*****.**>\n", PROGVERSION);
exit(0);
break;
default:
error = 1;
usage();
exit(1);
break;
}
}
if (error) {
usage();
exit(1);
}
if (fsuggested == 1 && nthreads == 0) {
fprintf(stderr, "ERROR: -f and -n should be used together!, exiting...\n");
usage();
exit(1);
}
if (argc == 2) /* If only url is supplied... */
fullurl = strdup(argv[1]);
else
if (optind < argc)
if (argc > 2)
fullurl = strdup(argv[optind]);
else {
usage();
exit(1);
}
else
if (optind == argc) {
usage();
exit(1);
}
parse_url(fullurl, req);
if(strlen(req->lfile) == 0)
strncpy(req->lfile, req->file, MAXBUFSIZ - 2);
getcwd(tmp, MAXBUFSIZ -2);
strncpy(tmp+strlen(tmp), "/", MAXBUFSIZ - strlen(tmp) - 2);
strncpy(tmp+strlen(tmp), req->lfile, MAXBUFSIZ - strlen(tmp) - 3);
strncpy(req->lfile, tmp, MAXBUFSIZ - 2);
/* If a log file for a previous try has been found, read it and
* resume the download job (resume_get), otherwise, start with
* a clean job (get)
*
* Logfile is of the pattern: aget-$file_name.log
*/
switch(req->proto) {
case PROTO_HTTP:
if ((retlog = read_log(&h)) != -1)
resumeDownload(&h, PROTO_HTTP);
else
startHTTP(req);
break;
case PROTO_FTP:
if ((retlog = read_log(&h)) != -1)
resumeDownload(&h, PROTO_FTP);
else
startFTP(req);
break;
}
return 0;
}
int
main(int argc, char *argv[])
{
register int i;
int c;
char *p;
struct rlimit rlim;
/*
** since priviliged actions will be done later on, at this stage
** the root-priviliges are dropped by switching effective user-id
** to real user-id (security reasons)
*/
if (! droprootprivs() )
{
fprintf(stderr, "not possible to drop root privs\n");
exit(42);
}
/*
** preserve command arguments to allow restart of other version
*/
argvp = argv;
/*
** read defaults-files /etc/atoprc en $HOME/.atoprc (if any)
*/
readrc("/etc/atoprc");
if ( (p = getenv("HOME")) )
{
char path[1024];
snprintf(path, sizeof path, "%s/.atoprc", p);
readrc(path);
}
/*
** check if we are supposed to behave as 'atopsar'
** i.e. system statistics only
*/
if ( (p = strrchr(argv[0], '/')))
p++;
else
p = argv[0];
if ( memcmp(p, "atopsar", 7) == 0)
return atopsar(argc, argv);
/*
** interpret command-line arguments & flags
*/
if (argc > 1)
{
/*
** gather all flags for visualization-functions
**
** generic flags will be handled here;
** unrecognized flags are passed to the print-routines
*/
i = 0;
while (i < MAXFL-1 && (c=getopt(argc, argv, allflags)) != EOF)
{
switch (c)
{
case '?': /* usage wanted ? */
prusage(argv[0]);
break;
case 'V': /* version wanted ? */
printf("%s\n", getstrvers());
exit(0);
case 'w': /* writing of raw data ? */
if (optind >= argc)
prusage(argv[0]);
strncpy(rawname, argv[optind++], RAWNAMESZ-1);
vis.show_samp = rawwrite;
break;
case 'r': /* reading of raw data ? */
if (optind < argc && *(argv[optind]) != '-')
strncpy(rawname, argv[optind++],
RAWNAMESZ-1);
rawreadflag++;
break;
case 'S': /* midnight limit ? */
midnightflag++;
break;
case 'a': /* all processes per sample ? */
deviatonly=0;
break;
case 'b': /* begin time ? */
if ( !hhmm2secs(optarg, &begintime) )
prusage(argv[0]);
break;
case 'e': /* end time ? */
if ( !hhmm2secs(optarg, &endtime) )
prusage(argv[0]);
break;
case 'j': /* show disk partitions statistics */
diskpartview = 1;
break;
case 'P': /* parseable output? */
if ( !parsedef(optarg) )
prusage(argv[0]);
vis.show_samp = parseout;
break;
case 'L': /* line length */
if ( !numeric(optarg) )
prusage(argv[0]);
linelen = atoi(optarg);
break;
default: /* gather other flags */
flaglist[i++] = c;
}
}
/*
** get optional interval-value and optional number of samples
*/
if (optind < argc && optind < MAXFL)
{
if (!numeric(argv[optind]))
prusage(argv[0]);
interval = atoi(argv[optind]);
optind++;
if (optind < argc)
{
if (!numeric(argv[optind]) )
prusage(argv[0]);
if ( (nsamples = atoi(argv[optind])) < 1)
prusage(argv[0]);
}
}
}
/*
** determine the name of this node (without domain-name)
** and the kernel-version
*/
(void) uname(&utsname);
if ( (p = strchr(utsname.nodename, '.')) )
*p = '\0';
utsnodenamelen = strlen(utsname.nodename);
sscanf(utsname.release, "%d.%d.%d", &osrel, &osvers, &ossub);
/*
** determine the clock rate and memory page size for this machine
*/
hertz = sysconf(_SC_CLK_TCK);
pagesize = sysconf(_SC_PAGESIZE);
/*
** check if raw data from a file must be viewed
*/
if (rawreadflag)
{
rawread();
cleanstop(0);
}
/*
** determine start-time for gathering current statistics
*/
curtime = getboot() / hertz;
/*
** catch signals for proper close-down
*/
signal(SIGHUP, cleanstop);
signal(SIGTERM, cleanstop);
/*
** regain the root-priviliges that we dropped at the beginning
** to do some priviliged work
*/
regainrootprivs();
/*
** lock ATOP in memory to get reliable samples (also when
** memory is low);
** ignored if not running under superuser priviliges!
*/
rlim.rlim_cur = RLIM_INFINITY;
rlim.rlim_max = RLIM_INFINITY;
(void) setrlimit(RLIMIT_MEMLOCK, &rlim);
(void) mlockall(MCL_CURRENT|MCL_FUTURE);
/*
** increment CPU scheduling-priority to get reliable samples (also
** during heavy CPU load);
** ignored if not running under superuser priviliges!
*/
if ( nice(-20) == -1)
;
/*
** switch-on the process-accounting mechanism to register the
** (remaining) resource-usage by processes which have finished
*/
acctreason = acctswon();
/*
** determine properties (like speed) of all interfaces
*/
initifprop();
/*
** open socket to the IP layer to issue getsockopt() calls later on
*/
netatop_ipopen();
/*
** since priviliged activities are finished now, there is no
** need to keep running under root-priviliges, so switch
** effective user-id to real user-id
*/
if (! droprootprivs() )
cleanstop(42);
/*
** start the engine now .....
*/
engine();
cleanstop(0);
return 0; /* never reached */
}
int
main(int argc, char *argv[])
{
register int i;
int c;
char *p;
char path[MAXPATHLEN];
pmOptions opts = {
.short_options = allflags,
.flags = PM_OPTFLAG_BOUNDARIES,
};
/*
** preserve command arguments to allow restart of other version
*/
argvp = argv;
/*
** read defaults-files /etc/atoprc en $HOME/.atoprc (if any)
*/
readrc("/etc/atoprc", 1);
if ( (p = getenv("HOME")) )
{
snprintf(path, sizeof(path), "%s/.atoprc", p);
path[sizeof(path)-1] = '\0';
readrc(path, 0);
}
/*
** check if we are supposed to behave as 'atopsar'
** i.e. system statistics only
*/
__pmSetProgname(argv[0]);
if (strcmp(pmProgname, "pcp-atopsar") == 0)
return atopsar(argc, argv);
__pmStartOptions(&opts);
if (opts.narchives > 0)
rawreadflag++;
/*
** interpret command-line arguments & flags
*/
if (argc > 1)
{
/*
** gather all flags for visualization-functions
**
** generic flags will be handled here;
** unrecognized flags are passed to the print-routines
*/
i = 0;
while (i < MAXFL-1 && (c = pmgetopt_r(argc, argv, &opts)) != EOF)
{
switch (c)
{
case '?': /* usage wanted ? */
prusage(pmProgname);
break;
case 'V': /* version wanted ? */
printf("%s\n", getstrvers());
exit(0);
case 'w': /* writing of raw data ? */
rawname = opts.optarg;
rawwriteflag++;
break;
case 'r': /* reading of raw data ? */
rawarchive(&opts, opts.optarg);
rawreadflag++;
break;
case 'S': /* midnight limit ? */
midnightflag++;
break;
case 'a': /* all processes per sample ? */
deviatonly = 0;
break;
case 'R': /* all processes per sample ? */
calcpss = 1;
break;
case 'b': /* begin time ? */
opts.start_optarg = abstime(opts.optarg);
break;
case 'e': /* end time ? */
opts.finish_optarg = abstime(opts.optarg);
break;
case 'P': /* parseable output? */
if ( !parsedef(opts.optarg) )
prusage(pmProgname);
vis.show_samp = parseout;
break;
case 'L': /* line length */
if ( !numeric(opts.optarg) )
prusage(pmProgname);
linelen = atoi(opts.optarg);
break;
default: /* gather other flags */
flaglist[i++] = c;
}
}
/*
** get optional interval-value and optional number of samples
*/
if (opts.optind < argc && opts.optind < MAXFL)
{
char *endnum, *arg;
arg = argv[opts.optind++];
if (pmParseInterval(arg, &opts.interval, &endnum) < 0)
{
pmprintf(
"%s: %s option not in pmParseInterval(3) format:\n%s\n",
pmProgname, arg, endnum);
free(endnum);
opts.errors++;
}
else
interval = opts.interval;
if (opts.optind < argc)
{
arg = argv[opts.optind];
if (!numeric(arg))
prusage(pmProgname);
if ((opts.samples = atoi(arg)) < 1)
prusage(pmProgname);
nsamples = opts.samples;
}
}
}
__pmEndOptions(&opts);
if (opts.errors)
prusage(pmProgname);
/*
** find local host details (no privileged access required)
*/
setup_globals(&opts);
/*
** check if we are in data recording mode
*/
if (rawwriteflag)
{
rawwrite(&opts, rawname, &interval, nsamples, midnightflag);
cleanstop(0);
}
/*
** catch signals for proper close-down
*/
signal(SIGHUP, cleanstop);
signal(SIGTERM, cleanstop);
/*
** switch-on the process-accounting mechanism to register the
** (remaining) resource-usage by processes which have finished
*/
acctreason = acctswon();
/*
** determine properties (like speed) of all interfaces
*/
initifprop();
/*
** open socket to the IP layer to issue getsockopt() calls later on
*/
netatop_ipopen();
/*
** start the engine now .....
*/
engine();
cleanstop(0);
return 0; /* never reached */
}
