void iopause(iopause_fd *x,unsigned int len,struct taia *deadline,struct taia *stamp)
{
struct taia t;
int millisecs;
double d;
int i;
if (taia_less(deadline,stamp))
millisecs = 0;
else {
t = *stamp;
taia_sub(&t,deadline,&t);
d = taia_approx(&t);
if (d > 1000.0) d = 1000.0;
millisecs = d * 1000.0 + 20.0;
}
for (i = 0;i < len;++i)
x[i].revents = 0;
#ifdef IOPAUSE_POLL
poll(x,len,millisecs);
/* XXX: some kernels apparently need x[0] even if len is 0 */
/* XXX: how to handle EAGAIN? are kernels really this dumb? */
/* XXX: how to handle EINVAL? when exactly can this happen? */
#else
{
struct timeval tv;
fd_set rfds;
fd_set wfds;
int nfds;
int fd;
FD_ZERO(&rfds);
FD_ZERO(&wfds);
nfds = 1;
for (i = 0;i < len;++i) {
fd = x[i].fd;
if (fd < 0) continue;
if (fd >= 8 * sizeof(fd_set)) continue; /*XXX*/
if (fd >= nfds) nfds = fd + 1;
if (x[i].events & IOPAUSE_READ) FD_SET(fd,&rfds);
if (x[i].events & IOPAUSE_WRITE) FD_SET(fd,&wfds);
}
tv.tv_sec = millisecs / 1000;
tv.tv_usec = 1000 * (millisecs % 1000);
if (select(nfds,&rfds,&wfds,(fd_set *) 0,&tv) <= 0)
return;
/* XXX: for EBADF, could seek out and destroy the bad descriptor */
for (i = 0;i < len;++i) {
fd = x[i].fd;
if (fd < 0) continue;
if (fd >= 8 * sizeof(fd_set)) continue; /*XXX*/
if (x[i].events & IOPAUSE_READ)
if (FD_ISSET(fd,&rfds)) x[i].revents |= IOPAUSE_READ;
if (x[i].events & IOPAUSE_WRITE)
if (FD_ISSET(fd,&wfds)) x[i].revents |= IOPAUSE_WRITE;
}
}
#endif
}
int sv_main(int argc, char **argv)
{
unsigned opt;
unsigned i, want_exit;
char *x;
for (i = strlen(*argv); i; --i)
if ((*argv)[i-1] == '/')
break;
*argv += i;
service = argv;
services = 1;
if ((x = getenv("SVDIR"))) varservice = x;
if ((x = getenv("SVWAIT"))) waitsec = xatoul(x);
/* TODO: V can be handled internally by getopt_ulflags */
opt = getopt32(argc, argv, "w:vV", &x);
if (opt & 1) waitsec = xatoul(x);
if (opt & 2) verbose = 1;
if (opt & 4) usage();
if (!(action = *argv++)) usage();
--argc;
service = argv; services = argc;
if (!*service) usage();
taia_now(&tnow); tstart = tnow;
if ((curdir = open_read(".")) == -1)
fatal_cannot("open current directory");
act = &control; acts = "s";
if (verbose) cbk = ✓
switch (*action) {
case 'x': case 'e':
acts = "x"; break;
case 'X': case 'E':
acts = "x"; kll = 1; cbk = ✓ break;
case 'D':
acts = "d"; kll = 1; cbk = ✓ break;
case 'T':
acts = "tc"; kll = 1; cbk = ✓ break;
case 'c':
if (!str_diff(action, "check")) {
act = 0;
acts = "c";
cbk = ✓
break;
}
case 'u': case 'd': case 'o': case 't': case 'p': case 'h':
case 'a': case 'i': case 'k': case 'q': case '1': case '2':
action[1] = 0; acts = action; break;
case 's':
if (!str_diff(action, "shutdown")) {
acts = "x";
cbk = ✓
break;
}
if (!str_diff(action, "start")) {
acts = "u";
cbk = ✓
break;
}
if (!str_diff(action, "stop")) {
acts = "d";
cbk = ✓
break;
}
act = &status; cbk = 0; break;
case 'r':
if (!str_diff(action, "restart")) {
acts = "tcu";
cbk = ✓
break;
}
usage();
case 'f':
if (!str_diff(action, "force-reload"))
{ acts = "tc"; kll = 1; cbk = ✓ break; }
if (!str_diff(action, "force-restart"))
{ acts = "tcu"; kll = 1; cbk = ✓ break; }
if (!str_diff(action, "force-shutdown"))
{ acts = "x"; kll = 1; cbk = ✓ break; }
if (!str_diff(action, "force-stop"))
{ acts = "d"; kll = 1; cbk = ✓ break; }
default:
usage();
}
servicex = service;
for (i = 0; i < services; ++i) {
if ((**service != '/') && (**service != '.')) {
if ((chdir(varservice) == -1) || (chdir(*service) == -1)) {
fail("cannot change to service directory");
*service = 0;
}
} else if (chdir(*service) == -1) {
fail("cannot change to service directory");
*service = 0;
}
if (*service) if (act && (act(acts) == -1)) *service = 0;
if (fchdir(curdir) == -1) fatal_cannot("change to original directory");
service++;
}
if (*cbk)
for (;;) {
taia_sub(&tdiff, &tnow, &tstart);
service = servicex; want_exit = 1;
for (i = 0; i < services; ++i, ++service) {
if (!*service) continue;
if ((**service != '/') && (**service != '.')) {
if ((chdir(varservice) == -1) || (chdir(*service) == -1)) {
fail("cannot change to service directory");
*service = 0;
}
} else if (chdir(*service) == -1) {
fail("cannot change to service directory");
*service = 0;
}
if (*service) { if (cbk(acts) != 0) *service = 0; else want_exit = 0; }
if (*service && taia_approx(&tdiff) > waitsec) {
kll ? printf(KILL) : printf(TIMEOUT);
if (svstatus_get() > 0) { svstatus_print(*service); ++rc; }
puts(""); /* will also flush the output */
if (kll) control("k");
*service = 0;
}
if (fchdir(curdir) == -1)
fatal_cannot("change to original directory");
}
if (want_exit) break;
usleep(420000);
taia_now(&tnow);
}
return rc > 99 ? 99 : rc;
}
int main(int argc, char **argv) {
unsigned int i, done;
char *x;
progname =*argv;
for (i =str_len(*argv); i; --i) if ((*argv)[i -1] == '/') break;
*argv +=i;
optprogname =progname =*argv;
service =argv;
services =1;
lsb =(str_diff(progname, "sv"));
if ((x =env_get("SVDIR"))) varservice =x;
if ((x =env_get("SVWAIT"))) scan_ulong(x, &wait);
while ((i =getopt(argc, (const char* const*)argv, "w:vV")) != opteof) {
switch(i) {
case 'w': scan_ulong(optarg, &wait);
case 'v': verbose =1; break;
case 'V': strerr_warn1(VERSION, 0);
case '?': usage();
}
}
argv +=optind; argc -=optind;
if (!(action =*argv++)) usage(); --argc;
if (!lsb) { service =argv; services =argc; }
if (!*service) usage();
taia_now(&tnow); tstart =tnow;
if ((curdir =open_read(".")) == -1)
fatal("unable to open current directory");
act =&control; acts ="s";
if (verbose) cbk =✓
switch (*action) {
case 'x': case 'e':
acts ="x"; break;
case 'X': case 'E':
acts ="x"; kll =1; cbk =✓ break;
case 'D':
acts ="d"; kll =1; cbk =✓ break;
case 'T':
acts ="tc"; kll =1; cbk =✓ break;
case 'c':
if (!str_diff(action, "check")) { act =0; acts ="C"; cbk =✓ break; }
case 'u': case 'd': case 'o': case 't': case 'p': case 'h':
case 'a': case 'i': case 'k': case 'q': case '1': case '2':
action[1] =0; acts =action; break;
case 's':
if (!str_diff(action, "shutdown")) { acts ="x"; cbk =✓ break; }
if (!str_diff(action, "start")) { acts ="u"; cbk =✓ break; }
if (!str_diff(action, "stop")) { acts ="d"; cbk =✓ break; }
if (lsb && str_diff(action, "status")) usage();
act =&status; cbk =0; break;
case 'r':
if (!str_diff(action, "restart")) { acts ="tcu"; cbk =✓ break; }
usage();
case 'f':
if (!str_diff(action, "force-reload"))
{ acts ="tc"; kll =1; cbk =✓ break; }
if (!str_diff(action, "force-restart"))
{ acts ="tcu"; kll =1; cbk =✓ break; }
if (!str_diff(action, "force-shutdown"))
{ acts ="x"; kll =1; cbk =✓ break; }
if (!str_diff(action, "force-stop"))
{ acts ="d"; kll =1; cbk =✓ break; }
default:
usage();
}
servicex =service;
for (i =0; i < services; ++i) {
if ((**service != '/') && (**service != '.') && **service &&
((*service)[str_len(*service) -1] != '/')) {
if ((chdir(varservice) == -1) || (chdir(*service) == -1)) {
fail("unable to change to service directory");
*service =0;
}
}
else
if (chdir(*service) == -1) {
fail("unable to change to service directory");
*service =0;
}
if (*service) if (act && (act(acts) == -1)) *service =0;
if (fchdir(curdir) == -1) fatal("unable to change to original directory");
service++;
}
if (*cbk)
for (;;) {
taia_sub(&tdiff, &tnow, &tstart);
service =servicex; done =1;
for (i =0; i < services; ++i, ++service) {
if (!*service) continue;
if ((**service != '/') && (**service != '.')) {
if ((chdir(varservice) == -1) || (chdir(*service) == -1)) {
fail("unable to change to service directory");
*service =0;
}
}
else
if (chdir(*service) == -1) {
fail("unable to change to service directory");
*service =0;
}
if (*service) { if (cbk(acts) != 0) *service =0; else done =0; }
if (*service && taia_approx(&tdiff) > wait) {
kll ? outs(KILL) : outs(TIMEOUT);
if (svstatus_get() > 0) { svstatus_print(*service); ++rc; }
flush("\n");
if (kll) control("k");
*service =0;
}
if (fchdir(curdir) == -1)
fatal("unable to change to original directory");
}
if (done) break;
usleep(420000);
taia_now(&tnow);
}
return(rc > 99 ? 99 : rc);
}
int main(int argc, char **argv)
{
kumy_file_t *kumy;
miniseed_file_t *mseed[KUMY_FILE_CHANNELS];
int32_t frame[KUMY_FILE_CHANNELS];
uint64_t frames, l, frames_per_file, frames_total, frame_count = 0;
int i;
char oname[1024], folder[1024];
uint32_t sample_rate, seconds_per_file;
int percent = 0, old_percent = -1;
int compression = 1, show_progress = 1;
char *infile = 0;
struct taia start_time; /* 1871 */
struct taia stop_time; /* 1951 */
struct taia sync_time; /* 2031 */
struct taia skew_time; /* 2111 */
struct caltime ct;
struct taia tt, dt;
program = argv[0];
parse_options(&argc, &argv, OPTIONS(
FLAG('n', "no-compression", compression, 0),
FLAG('q', "quiet", show_progress, 0),
FLAG_CALLBACK('h', "help", usage)
));
if (argc < 2) {
usage(0, 0, 0);
}
infile = argv[1];
if (!(kumy = kumy_file_open(infile))) {
fprintf(stderr, "Invalid file: %s.\n", infile);
return -1;
}
parse_text_date(&start_time, kumy->text_header[0].content + 1871);
parse_text_date(&stop_time, kumy->text_header[0].content + 1951);
parse_text_date(&sync_time, kumy->text_header[0].content + 2031);
parse_text_date(&skew_time, kumy->text_header[0].content + 2111);
sample_rate = 1000000 / kumy->binary_header[0].sample_interval;
if (sample_rate <= 250) {
seconds_per_file = 86400;
} else if (sample_rate <= 500) {
seconds_per_file = 43200;
} else if (sample_rate <= 1000) {
seconds_per_file = 21600;
} else if (sample_rate <= 2000) {
seconds_per_file = 10800;
} else {
seconds_per_file = 5400;
}
frames_per_file = sample_rate * seconds_per_file;
frames_total = kumy->binary_header[0].num_samples;
caltime_utc(&ct, &start_time.sec, 0, 0);
ct.hour = 0;
ct.minute = 0;
ct.second = 0;
caltime_tai(&ct, &tt.sec);
tt.nano = tt.atto = 0;
while (!taia_less(&start_time, &tt)) {
tt.sec.x += seconds_per_file;
}
taia_sub(&dt, &tt, &start_time);
frames = taia_approx(&dt) / seconds_per_file * frames_per_file;
l = last('.', infile);
if (l == -1 || l >= sizeof(folder)) return -1;
/* Create folder. */
copy(folder, l, infile);
folder[l] = 0;
mkdir_p(folder);
for (i = 0; i < KUMY_FILE_CHANNELS; ++i) {
caltime_utc(&ct, &start_time.sec, 0, 0);
snprintf(oname, sizeof(oname), "%s/%lld.%02lld.%02lld.%02lld.%02lld.%02lld.%s.seed",
folder, (long long)ct.date.year, (long long)ct.date.month, (long long)ct.date.day, (long long)ct.hour, (long long)ct.minute, (long long)ct.second, channel_names[i]);
if (!(mseed[i] = miniseed_file_create((char*)oname))) {
fprintf(stderr, "Invalid file: %s.\n", oname);
return -1;
}
miniseed_file_set_sample_rate(mseed[i], sample_rate);
miniseed_file_set_start_time(mseed[i], &start_time);
miniseed_file_set_info(mseed[i], "OBS", "DE", channel_names[i], "K");
miniseed_file_set_compression(mseed[i], compression);
}
while (kumy_file_read_int_frame(kumy, frame) >= 0) {
if (frames == 0) {
/* Create new files.*/
for (i = 0; i < KUMY_FILE_CHANNELS; ++i) {
miniseed_file_close(mseed[i]);
caltime_utc(&ct, &tt.sec, 0, 0);
snprintf(oname, sizeof(oname), "%s/%lld.%02lld.%02lld.%02lld.%02lld.%02lld.%s.seed",
folder, (long long)ct.date.year, (long long)ct.date.month, (long long)ct.date.day, (long long)ct.hour, (long long)ct.minute, (long long)ct.second, channel_names[i]);
if (!(mseed[i] = miniseed_file_create((char*)oname))) {
fprintf(stderr, "Invalid file: %s.\n", oname);
return -1;
}
miniseed_file_set_sample_rate(mseed[i], sample_rate);
miniseed_file_set_start_time(mseed[i], &tt);
miniseed_file_set_info(mseed[i], "OBS", "DE", channel_names[i], "K");
miniseed_file_set_compression(mseed[i], compression);
}
frames = frames_per_file;
tt.sec.x += seconds_per_file;
}
for (i = 0; i < KUMY_FILE_CHANNELS; ++i) {
miniseed_file_write_int_frame(mseed[i], frame + i);
}
if (show_progress && frame_count % 10000 == 0) {
percent = 100 * frame_count / frames_total;
if (percent != old_percent) {
progress(percent, 0);
old_percent = percent;
}
}
--frames;
++frame_count;
}
if (show_progress) progress(100, 1);
kumy_file_close(kumy);
for (i = 0; i < KUMY_FILE_CHANNELS; ++i) {
miniseed_file_close(mseed[i]);
}
return 0;
}