gecko-dev/nsprpub/tools/tail.c

136 lines
3.4 KiB
C

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "prio.h"
#include "prprf.h"
#include "prinit.h"
#include "prthread.h"
#include "prinrval.h"
#include "plerror.h"
#include "plgetopt.h"
#include <stdlib.h>
#define BUFFER_SIZE 500
static PRFileDesc *out = NULL, *err = NULL;
static void Help(void) {
PR_fprintf(err, "Usage: tail [-n <n>] [-f] [-h] <filename>\n");
PR_fprintf(err, "\t-t <n> Dally time in milliseconds\n");
PR_fprintf(err, "\t-n <n> Number of bytes before <eof>\n");
PR_fprintf(err, "\t-f Follow the <eof>\n");
PR_fprintf(err, "\t-h This message and nothing else\n");
} /* Help */
PRIntn main(PRIntn argc, char** argv) {
PRIntn rv = 0;
PLOptStatus os;
PRStatus status;
PRFileDesc* file;
PRFileInfo fileInfo;
PRIntervalTime dally;
char buffer[BUFFER_SIZE];
PRBool follow = PR_FALSE;
const char* filename = NULL;
PRUint32 position = 0, seek = 0, time = 0;
PLOptState* opt = PL_CreateOptState(argc, argv, "hfn:");
out = PR_GetSpecialFD(PR_StandardOutput);
err = PR_GetSpecialFD(PR_StandardError);
while (PL_OPT_EOL != (os = PL_GetNextOpt(opt))) {
if (PL_OPT_BAD == os) {
continue;
}
switch (opt->option) {
case 0: /* it's the filename */
filename = opt->value;
break;
case 'n': /* bytes before end of file */
seek = atoi(opt->value);
break;
case 't': /* dally time */
time = atoi(opt->value);
break;
case 'f': /* follow the end of file */
follow = PR_TRUE;
break;
case 'h': /* user wants some guidance */
Help(); /* so give him an earful */
return 2; /* but not a lot else */
break;
default:
break;
}
}
PL_DestroyOptState(opt);
if (0 == time) {
time = 1000;
}
dally = PR_MillisecondsToInterval(time);
if (NULL == filename) {
(void)PR_fprintf(out, "Input file not specified\n");
rv = 1;
goto done;
}
file = PR_Open(filename, PR_RDONLY, 0);
if (NULL == file) {
PL_FPrintError(err, "File cannot be opened for reading");
return 1;
}
status = PR_GetOpenFileInfo(file, &fileInfo);
if (PR_FAILURE == status) {
PL_FPrintError(err, "Cannot acquire status of file");
rv = 1;
goto done;
}
if (seek > 0) {
if (seek > fileInfo.size) {
seek = 0;
}
position = PR_Seek(file, (fileInfo.size - seek), PR_SEEK_SET);
if (-1 == (PRInt32)position) {
PL_FPrintError(err, "Cannot seek to starting position");
}
}
do {
while (position < fileInfo.size) {
PRInt32 read, bytes = fileInfo.size - position;
if (bytes > sizeof(buffer)) {
bytes = sizeof(buffer);
}
read = PR_Read(file, buffer, bytes);
if (read != bytes) {
PL_FPrintError(err, "Cannot read to eof");
}
position += read;
PR_Write(out, buffer, read);
}
if (follow) {
PR_Sleep(dally);
status = PR_GetOpenFileInfo(file, &fileInfo);
if (PR_FAILURE == status) {
PL_FPrintError(err, "Cannot acquire status of file");
rv = 1;
goto done;
}
}
} while (follow);
done:
PR_Close(file);
return rv;
} /* main */
/* tail.c */