Annotation of embedaddon/rsync/support/savetransfer.c, revision 1.1.1.1
1.1 misho 1: /* This program can record the stream of data flowing to or from a program.
2: * This allows it to be used to check that rsync's data that is flowing
3: * through a remote shell is not being corrupted (for example).
4: *
5: * Usage: savetransfer [-i|-o] OUTPUT_FILE PROGRAM [ARGS...]
6: * -i Save the input going to PROGRAM to the OUTPUT_FILE
7: * -o Save the output coming from PROGRAM to the OUTPUT_FILE
8: *
9: * If you want to capture the flow of data for an rsync command, use one of
10: * the following commands (the resulting files should be identical):
11: *
12: * rsync -av --rsh="savetransfer -i /tmp/to.server ssh"
13: * --rsync-path="savetransfer -i /tmp/from.client rsync" SOURCE DEST
14: *
15: * rsync -av --rsh="savetransfer -o /tmp/from.server ssh"
16: * --rsync-path="savetransfer -o /tmp/to.client rsync" SOURCE DEST
17: *
18: * Note that this program aborts after 30 seconds of inactivity, so you'll need
19: * to change it if that is not enough dead time for your transfer. Also, some
20: * of the above commands will not notice that the transfer is done (if we're
21: * saving the input to a PROGRAM and the PROGRAM goes away: we won't notice
22: * that it's gone unless more data comes in) -- when this happens it will delay
23: * at the end of the transfer until the timeout period expires.
24: */
25:
26: #include "../rsync.h"
27:
28: #define TIMEOUT_SECONDS 30
29:
30: #ifdef HAVE_SIGACTION
31: static struct sigaction sigact;
32: #endif
33:
34: void run_program(char **command);
35:
36: char buf[4096];
37: int save_data_from_program = 0;
38:
39: int
40: main(int argc, char *argv[])
41: {
42: int fd_file, len;
43: struct timeval tv;
44: fd_set fds;
45:
46: argv++;
47: if (--argc && argv[0][0] == '-') {
48: if (argv[0][1] == 'o')
49: save_data_from_program = 1;
50: else if (argv[0][1] == 'i')
51: save_data_from_program = 0;
52: else {
53: fprintf(stderr, "Unknown option: %s\n", argv[0]);
54: exit(1);
55: }
56: argv++;
57: argc--;
58: }
59: if (argc < 2) {
60: fprintf(stderr, "Usage: savetransfer [-i|-o] OUTPUT_FILE PROGRAM [ARGS...]\n");
61: fprintf(stderr, "-i Save the input going to PROGRAM to the OUTPUT_FILE\n");
62: fprintf(stderr, "-o Save the output coming from PROGRAM to the OUTPUT_FILE\n");
63: exit(1);
64: }
65: if ((fd_file = open(*argv, O_WRONLY|O_TRUNC|O_CREAT|O_BINARY, 0644)) < 0) {
66: fprintf(stderr, "Unable to write to `%s': %s\n", *argv, strerror(errno));
67: exit(1);
68: }
69: set_blocking(fd_file);
70:
71: SIGACTION(SIGPIPE, SIG_IGN);
72:
73: run_program(argv + 1);
74:
75: #if defined HAVE_SETMODE && O_BINARY
76: setmode(STDIN_FILENO, O_BINARY);
77: setmode(STDOUT_FILENO, O_BINARY);
78: #endif
79: set_nonblocking(STDIN_FILENO);
80: set_blocking(STDOUT_FILENO);
81:
82: while (1) {
83: FD_ZERO(&fds);
84: FD_SET(STDIN_FILENO, &fds);
85: tv.tv_sec = TIMEOUT_SECONDS;
86: tv.tv_usec = 0;
87: if (!select(STDIN_FILENO+1, &fds, NULL, NULL, &tv))
88: break;
89: if (!FD_ISSET(STDIN_FILENO, &fds))
90: break;
91: if ((len = read(STDIN_FILENO, buf, sizeof buf)) <= 0)
92: break;
93: if (write(STDOUT_FILENO, buf, len) != len) {
94: fprintf(stderr, "Failed to write data to stdout: %s\n", strerror(errno));
95: exit(1);
96: }
97: if (write(fd_file, buf, len) != len) {
98: fprintf(stderr, "Failed to write data to fd_file: %s\n", strerror(errno));
99: exit(1);
100: }
101: }
102: return 0;
103: }
104:
105: void
106: run_program(char **command)
107: {
108: int pipe_fds[2], ret;
109: pid_t pid;
110:
111: if (pipe(pipe_fds) < 0) {
112: fprintf(stderr, "pipe failed: %s\n", strerror(errno));
113: exit(1);
114: }
115:
116: if ((pid = fork()) < 0) {
117: fprintf(stderr, "fork failed: %s\n", strerror(errno));
118: exit(1);
119: }
120:
121: if (pid == 0) {
122: if (save_data_from_program)
123: ret = dup2(pipe_fds[1], STDOUT_FILENO);
124: else
125: ret = dup2(pipe_fds[0], STDIN_FILENO);
126: if (ret < 0) {
127: fprintf(stderr, "Failed to dup (in child): %s\n", strerror(errno));
128: exit(1);
129: }
130: close(pipe_fds[0]);
131: close(pipe_fds[1]);
132: set_blocking(STDIN_FILENO);
133: set_blocking(STDOUT_FILENO);
134: execvp(command[0], command);
135: fprintf(stderr, "Failed to exec %s: %s\n", command[0], strerror(errno));
136: exit(1);
137: }
138:
139: if (save_data_from_program)
140: ret = dup2(pipe_fds[0], STDIN_FILENO);
141: else
142: ret = dup2(pipe_fds[1], STDOUT_FILENO);
143: if (ret < 0) {
144: fprintf(stderr, "Failed to dup (in parent): %s\n", strerror(errno));
145: exit(1);
146: }
147: close(pipe_fds[0]);
148: close(pipe_fds[1]);
149: }
150:
151: void
152: set_nonblocking(int fd)
153: {
154: int val;
155:
156: if ((val = fcntl(fd, F_GETFL, 0)) == -1)
157: return;
158: if (!(val & NONBLOCK_FLAG)) {
159: val |= NONBLOCK_FLAG;
160: fcntl(fd, F_SETFL, val);
161: }
162: }
163:
164: void
165: set_blocking(int fd)
166: {
167: int val;
168:
169: if ((val = fcntl(fd, F_GETFL, 0)) < 0)
170: return;
171: if (val & NONBLOCK_FLAG) {
172: val &= ~NONBLOCK_FLAG;
173: fcntl(fd, F_SETFL, val);
174: }
175: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>
![[BACK]](/icons/cvsweb/back.gif){kind=icon}
Return to savetransfer.c CVS log ![[TXT]](/icons/cvsweb/text.gif){kind=icon}
![[DIR]](/icons/cvsweb/dir.gif){kind=icon}
Up to [ELWIX - Embedded LightWeight unIX -] / embedaddon / rsync / support