int chirp_stream_printf(struct chirp_stream *s, time_t stoptime, const char *fmt, ...) { va_list args; int result; if(s->mode != CHIRP_STREAM_WRITE) { errno = EINVAL; return -1; } while(1) { int avail = s->buffer_length - s->buffer_valid; va_start(args, fmt); result = vsnprintf(&s->buffer[s->buffer_valid], avail, fmt, args); va_end(args); if(result < 0) { return result; } else if(result <= avail) { s->buffer_valid += result; break; } else { chirp_stream_flush(s, stoptime); if(result > s->buffer_length) { s->buffer = realloc(s->buffer, result + 1); s->buffer_length = result + 1; } } } return result; }
int chirp_stream_close(struct chirp_stream *s, time_t stoptime) { chirp_stream_flush(s, stoptime); chirp_client_disconnect(s->client); free(s->buffer); free(s); return 0; }
int chirp_stream_write(struct chirp_stream *s, const void *data, int length, time_t stoptime) { if(s->mode != CHIRP_STREAM_WRITE) { errno = EINVAL; return -1; } int avail = s->buffer_length - s->buffer_valid; if(length > avail) { chirp_stream_flush(s, stoptime); } if(length >= s->buffer_length) { return chirp_client_putstream_write(s->client, data, length, stoptime); } memcpy(&s->buffer[s->buffer_valid], data, length); s->buffer_valid += length; return length; }
int main(int argc, char *argv[]) { int did_explicit_auth = 0; time_t stoptime; char c; int i, srcindex, nstreams; FILE *localfile; struct chirp_stream *stream[argc - 2]; const char *localmode; int remotemode; debug_config(argv[0]); while((c = getopt(argc, argv, "a:b:d:t:vh")) != (char) -1) { switch (c) { case 'a': auth_register_byname(optarg); did_explicit_auth = 1; break; case 'b': buffer_size = atoi(optarg); break; case 'd': debug_flags_set(optarg); break; case 't': timeout = string_time_parse(optarg); break; case 'v': show_version(argv[0]); exit(0); break; case 'h': show_help(argv[0]); exit(0); break; } } if(!did_explicit_auth) auth_register_all(); if((argc - optind) < 4) { show_help(argv[0]); return 1; } if(!strcmp(argv[optind], "split")) { stream_mode = MODE_SPLIT; localmode = "r"; remotemode = CHIRP_STREAM_WRITE; } else if(!strcmp(argv[optind], "copy")) { stream_mode = MODE_COPY; localmode = "r"; remotemode = CHIRP_STREAM_WRITE; } else if(!strcmp(argv[optind], "join")) { stream_mode = MODE_JOIN; localmode = "w"; remotemode = CHIRP_STREAM_READ; } else { fprintf(stderr, "unknown operation: %s\n", argv[0]); show_help(argv[0]); return 1; } char *buffer = malloc(buffer_size); srcindex = optind + 1; nstreams = (argc - optind - 2) / 2; stoptime = time(0) + timeout; localfile = fopen(argv[srcindex], localmode); if(!localfile) { fprintf(stderr, "couldn't open %s: %s\n", argv[srcindex], strerror(errno)); return 1; } char **hostname = malloc(sizeof(*hostname) * nstreams); char **filename = malloc(sizeof(*filename) * nstreams); for(i = 0; i < nstreams; i++) { hostname[i] = argv[srcindex + (2 * i) + 1]; filename[i] = argv[srcindex + (2 * i) + 2]; stream[i] = chirp_stream_open(hostname[i], filename[i], remotemode, stoptime); if(!stream[i]) { fprintf(stderr, "couldn't open %s:%s: %s\n", hostname[i], filename[i], strerror(errno)); return 1; } } if(stream_mode == MODE_SPLIT) { i = 0; while(fgets(buffer, buffer_size, localfile)) { int length = strlen(buffer); int actual = chirp_stream_write(stream[i], buffer, length, stoptime); if(actual != length) { fprintf(stderr, "couldn't write to %s:%s: %s\n", hostname[i], filename[i], strerror(errno)); return 1; } i = (i + 1) % nstreams; } } else if(stream_mode == MODE_COPY) { while(fgets(buffer, buffer_size, localfile)) { int length = strlen(buffer); for(i = 0; i < nstreams; i++) { int actual = chirp_stream_write(stream[i], buffer, length, stoptime); if(actual != length) { fprintf(stderr, "couldn't write to %s:%s: %s\n", hostname[i], filename[i], strerror(errno)); return 1; } } } } else { int streams_left = nstreams; while(streams_left > 0) { for(i = 0; i < nstreams; i++) { if(!stream[i]) continue; int length = chirp_stream_readline(stream[i], buffer, buffer_size, stoptime); if(length > 0) { length = strlen(buffer); fprintf(localfile, "%s\n", buffer); } else { streams_left--; } } } } for(i = 0; i < nstreams; i++) { chirp_stream_flush(stream[i], stoptime); chirp_stream_close(stream[i], stoptime); } return 0; }
int main(int argc, char *argv[]) { int did_explicit_auth = 0; time_t stoptime; signed char c; int i, srcindex, nstreams; FILE *localfile; struct chirp_stream *stream[argc - 2]; const char *localmode; int remotemode; char *tickets = NULL; debug_config(argv[0]); static struct option long_options[] = { {"auth", required_argument, 0, 'a'}, {"block-size", required_argument, 0, 'b'}, {"debug", required_argument, 0, 'd'}, {"tickets", required_argument, 0, 'i'}, {"timeout", required_argument, 0, 't'}, {"version", no_argument, 0, 'v'}, {"help", no_argument, 0, 'h'}, {0, 0, 0, 0} }; while((c = getopt_long(argc, argv, "a:b:d:i:t:vh", long_options, NULL)) > -1) { switch (c) { case 'a': auth_register_byname(optarg); did_explicit_auth = 1; break; case 'b': buffer_size = atoi(optarg); break; case 'd': debug_flags_set(optarg); break; case 'i': tickets = strdup(optarg); break; case 't': timeout = string_time_parse(optarg); break; case 'v': cctools_version_print(stdout, argv[0]); exit(0); break; case 'h': show_help(argv[0]); exit(0); break; } } cctools_version_debug(D_DEBUG, argv[0]); if(!did_explicit_auth) auth_register_all(); if(tickets) { auth_ticket_load(tickets); free(tickets); } else if(getenv(CHIRP_CLIENT_TICKETS)) { auth_ticket_load(getenv(CHIRP_CLIENT_TICKETS)); } else { auth_ticket_load(NULL); } if((argc - optind) < 4) { show_help(argv[0]); return 1; } if(!strcmp(argv[optind], "split")) { stream_mode = MODE_SPLIT; localmode = "r"; remotemode = CHIRP_STREAM_WRITE; } else if(!strcmp(argv[optind], "copy")) { stream_mode = MODE_COPY; localmode = "r"; remotemode = CHIRP_STREAM_WRITE; } else if(!strcmp(argv[optind], "join")) { stream_mode = MODE_JOIN; localmode = "w"; remotemode = CHIRP_STREAM_READ; } else { fprintf(stderr, "unknown operation: %s\n", argv[0]); show_help(argv[0]); return 1; } char *buffer = malloc(buffer_size); srcindex = optind + 1; nstreams = (argc - optind - 2) / 2; stoptime = time(0) + timeout; localfile = fopen(argv[srcindex], localmode); if(!localfile) { fprintf(stderr, "couldn't open %s: %s\n", argv[srcindex], strerror(errno)); return 1; } char **hostname = malloc(sizeof(*hostname) * nstreams); char **filename = malloc(sizeof(*filename) * nstreams); for(i = 0; i < nstreams; i++) { hostname[i] = argv[srcindex + (2 * i) + 1]; filename[i] = argv[srcindex + (2 * i) + 2]; stream[i] = chirp_stream_open(hostname[i], filename[i], remotemode, stoptime); if(!stream[i]) { fprintf(stderr, "couldn't open %s:%s: %s\n", hostname[i], filename[i], strerror(errno)); return 1; } } if(stream_mode == MODE_SPLIT) { i = 0; while(fgets(buffer, buffer_size, localfile)) { int length = strlen(buffer); int actual = chirp_stream_write(stream[i], buffer, length, stoptime); if(actual != length) { fprintf(stderr, "couldn't write to %s:%s: %s\n", hostname[i], filename[i], strerror(errno)); return 1; } i = (i + 1) % nstreams; } } else if(stream_mode == MODE_COPY) { while(fgets(buffer, buffer_size, localfile)) { int length = strlen(buffer); for(i = 0; i < nstreams; i++) { int actual = chirp_stream_write(stream[i], buffer, length, stoptime); if(actual != length) { fprintf(stderr, "couldn't write to %s:%s: %s\n", hostname[i], filename[i], strerror(errno)); return 1; } } } } else { int streams_left = nstreams; while(streams_left > 0) { for(i = 0; i < nstreams; i++) { if(!stream[i]) continue; int length = chirp_stream_readline(stream[i], buffer, buffer_size, stoptime); if(length > 0) { length = strlen(buffer); fprintf(localfile, "%s\n", buffer); } else { streams_left--; } } } } for(i = 0; i < nstreams; i++) { chirp_stream_flush(stream[i], stoptime); chirp_stream_close(stream[i], stoptime); } return 0; }