/* * main() */ int main(int ac, char *av[]) { struct ngm_connect ngc; const char *path = NULL; const char *hook = DEFAULT_HOOKNAME; int csock, dsock; int asciiFlag = 0; int loopFlag = 0; int noInput = 0; int execFlag = 0; int ch; if ((msgs = sl_init()) == NULL) err(EX_OSERR, NULL); /* Parse flags */ while ((ch = getopt(ac, av, "aedlm:nsS")) != -1) { switch (ch) { case 'a': asciiFlag = 1; break; case 'd': NgSetDebug(NgSetDebug(-1) + 1); break; case 'e': execFlag = 1; break; case 'l': loopFlag = 1; break; case 'n': noInput = 1; break; case 'm': if (sl_add(msgs, optarg) == -1) err(EX_OSERR, NULL); break; case 's': outfd = STDIN_FILENO; break; case 'S': infd = STDOUT_FILENO; break; case '?': default: Usage(); } } ac -= optind; av += optind; if (execFlag) { if (asciiFlag || loopFlag) { fprintf(stderr, "conflicting options\n"); Usage(); } if (ac < 3) Usage(); path = av[0]; hook = av[1]; av += 2; ac -= 2; } else { /* Get params */ switch (ac) { case 2: hook = av[1]; /* FALLTHROUGH */ case 1: path = av[0]; break; default: Usage(); } } /* Get sockets */ if (NgMkSockNode(NULL, &csock, &dsock) < 0) errx(EX_OSERR, "can't get sockets"); /* Connect socket node to specified node */ snprintf(ngc.path, sizeof(ngc.path), "%s", path); snprintf(ngc.ourhook, sizeof(ngc.ourhook), NG_SOCK_HOOK_NAME); snprintf(ngc.peerhook, sizeof(ngc.peerhook), "%s", hook); if (NgSendMsg(csock, ".", NGM_GENERIC_COOKIE, NGM_CONNECT, &ngc, sizeof(ngc)) < 0) errx(EX_OSERR, "can't connect to node"); if (execFlag) { /* move dsock to fd 0 and 1 */ (void)close(0); (void)close(1); if (!noInput) (void)dup2(dsock, 0); (void)dup2(dsock, 1); send_msgs(csock, path); /* try executing the program */ (void)execv(av[0], av); err(EX_OSERR, "%s", av[0]); } else send_msgs(csock, path); /* Close standard input if not reading from it */ if (noInput) fclose(stdin); /* Relay data */ while (1) { fd_set rfds; /* Setup bits */ FD_ZERO(&rfds); if (!noInput) FD_SET(infd, &rfds); FD_SET(dsock, &rfds); /* Wait for something to happen */ if (select(FD_SETSIZE, &rfds, NULL, NULL, NULL) < 0) err(EX_OSERR, "select"); /* Check data from socket */ if (FD_ISSET(dsock, &rfds)) { char buf[BUF_SIZE]; int rl, wl; /* Read packet from socket */ if ((rl = NgRecvData(dsock, buf, sizeof(buf), NULL)) < 0) err(EX_OSERR, "read(hook)"); if (rl == 0) errx(EX_OSERR, "read EOF from hook?!"); /* Write packet to stdout */ if (asciiFlag) WriteAscii((u_char *) buf, rl); else if ((wl = write(outfd, buf, rl)) != rl) { if (wl < 0) { err(EX_OSERR, "write(stdout)"); } else { errx(EX_OSERR, "stdout: read %d, wrote %d", rl, wl); } } /* Loopback */ if (loopFlag) { if (NgSendData(dsock, NG_SOCK_HOOK_NAME, buf, rl) < 0) err(EX_OSERR, "write(hook)"); } } /* Check data from stdin */ if (FD_ISSET(infd, &rfds)) { char buf[BUF_SIZE]; int rl; /* Read packet from stdin */ if ((rl = read(infd, buf, sizeof(buf))) < 0) err(EX_OSERR, "read(stdin)"); if (rl == 0) errx(EX_OSERR, "EOF(stdin)"); /* Write packet to socket */ if (NgSendData(dsock, NG_SOCK_HOOK_NAME, buf, rl) < 0) err(EX_OSERR, "write(hook)"); } } }
Var* ff_write(vfuncptr func, Var* arg) { Var* ob = NULL; char* filename = NULL; char* title = NULL; char* type = NULL; char* separator = NULL; /* for csv */ int header = 0; /* for csv */ int force = 0; /* Force file overwrite */ int hdf_old = 0; // write hdf file backward like davinci used to unsigned short iom_type_idx, iom_type_found; Alist alist[9]; alist[0] = make_alist("object", ID_UNK, NULL, &ob); alist[1] = make_alist("filename", ID_STRING, NULL, &filename); alist[2] = make_alist("type", ID_ENUM, NULL, &type); alist[3] = make_alist("title", ID_STRING, NULL, &title); alist[4] = make_alist("force", DV_INT32, NULL, &force); alist[5] = make_alist("separator", ID_STRING, NULL, &separator); alist[6] = make_alist("header", DV_INT32, NULL, &header); alist[7] = make_alist("hdf_old", DV_INT32, NULL, &hdf_old); alist[8].name = NULL; if (parse_args(func, arg, alist) == 0) return (NULL); /** ** Make sure user specified an object **/ if (ob == NULL) { parse_error("%s: No object specified.", func->name); return (NULL); } /** ** get filename. Verify type **/ if (filename == NULL) { parse_error("No filename specified."); return (NULL); } filename = dv_locate_file(filename); if (type == NULL) { parse_error("No type specified."); return (NULL); } /* ** Get title string */ if (title == NULL) { title = (char*)"DV data product"; } /* Check type against list of types supported by iomedley. */ iom_type_idx = iom_type_found = 0; while (iom_filetypes[iom_type_idx]) { if (!strcasecmp(type, iom_filetypes[iom_type_idx])) { iom_type_found = 1; break; } iom_type_idx++; } if (iom_type_found) dv_WriteIOM(ob, filename, type, force); else if (!strcasecmp(type, "raw")) dv_WriteRaw(ob, filename, force); else if (!strcasecmp(type, "vicar")) dv_WriteVicar(ob, filename, force); else if (!strcasecmp(type, "grd")) dv_WriteGRD(ob, filename, force, title, (char*)"davinci"); /* else if (!strcasecmp(type, "pnm")) dv_WritePNM(ob, filename, force); */ else if (!strcasecmp(type, "pgm")) dv_WritePGM(ob, filename, force); else if (!strcasecmp(type, "ppm")) dv_WritePPM(ob, filename, force); else if (!strcasecmp(type, "ascii")) WriteAscii(ob, filename, force); else if (!strcasecmp(type, "csv")) dv_WriteCSV(ob, filename, separator, header, force); else if (!strcasecmp(type, "ers")) dv_WriteERS(ob, filename, force); else if (!strcasecmp(type, "imath")) dv_WriteIMath(ob, filename, force); else if (!strcasecmp(type, "isis")) dv_WriteISIS(ob, filename, force, title); else if (!strcasecmp(type, "envi")) dv_WriteENVI(ob, filename, force); else if (!strcasecmp(type, "specpr")) { if (!force && file_exists(filename)) { parse_error("File %s already exists.\n", filename); return NULL; } WriteSpecpr(ob, filename, title); } /* ** Below here are optional packages */ #ifdef HAVE_LIBHDF5 else if (!strcasecmp(type, "hdf")) { struct stat statbuf; if (!force && !stat(filename, &statbuf)) { parse_error("File %s already exists.\n", filename); return NULL; } /* force ? */ WriteHDF5(-1, filename, ob, hdf_old); } #endif #ifdef BUILD_MODULE_SUPPORT else if (iomod_handler_for_type(type)) write_to_io_module(ob, filename, type, force); #endif #if 0 #ifdef HAVE_LIBMAGICK // else if (dvio_ValidGfx(type, GFX_type)) dv_WriteGFX_Image(ob, filename, force, GFX_type); else if (1) paramdvWriteImage(ob, filename, type, force); #endif #endif else { sprintf(error_buf, "Unrecognized type: %s", type); parse_error(NULL); return (NULL); } free(filename); return (NULL); }