Annotation of embedaddon/rsync/pipe.c, revision 1.1

1.1     ! misho       1: /*
        !             2:  * Routines used to setup various kinds of inter-process pipes.
        !             3:  *
        !             4:  * Copyright (C) 1996-2000 Andrew Tridgell
        !             5:  * Copyright (C) 1996 Paul Mackerras
        !             6:  * Copyright (C) 2001, 2002 Martin Pool <mbp@samba.org>
        !             7:  * Copyright (C) 2004-2009 Wayne Davison
        !             8:  *
        !             9:  * This program is free software; you can redistribute it and/or modify
        !            10:  * it under the terms of the GNU General Public License as published by
        !            11:  * the Free Software Foundation; either version 3 of the License, or
        !            12:  * (at your option) any later version.
        !            13:  *
        !            14:  * This program is distributed in the hope that it will be useful,
        !            15:  * but WITHOUT ANY WARRANTY; without even the implied warranty of
        !            16:  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
        !            17:  * GNU General Public License for more details.
        !            18:  *
        !            19:  * You should have received a copy of the GNU General Public License along
        !            20:  * with this program; if not, visit the http://fsf.org website.
        !            21:  */
        !            22: 
        !            23: #include "rsync.h"
        !            24: 
        !            25: extern int am_sender;
        !            26: extern int am_server;
        !            27: extern int blocking_io;
        !            28: extern int filesfrom_fd;
        !            29: extern mode_t orig_umask;
        !            30: extern char *logfile_name;
        !            31: extern struct chmod_mode_struct *chmod_modes;
        !            32: 
        !            33: /**
        !            34:  * Create a child connected to us via its stdin/stdout.
        !            35:  *
        !            36:  * This is derived from CVS code
        !            37:  *
        !            38:  * Note that in the child STDIN is set to blocking and STDOUT
        !            39:  * is set to non-blocking. This is necessary as rsh relies on stdin being blocking
        !            40:  *  and ssh relies on stdout being non-blocking
        !            41:  *
        !            42:  * If blocking_io is set then use blocking io on both fds. That can be
        !            43:  * used to cope with badly broken rsh implementations like the one on
        !            44:  * Solaris.
        !            45:  **/
        !            46: pid_t piped_child(char **command, int *f_in, int *f_out)
        !            47: {
        !            48:        pid_t pid;
        !            49:        int to_child_pipe[2];
        !            50:        int from_child_pipe[2];
        !            51: 
        !            52:        if (verbose >= 2)
        !            53:                print_child_argv("opening connection using:", command);
        !            54: 
        !            55:        if (fd_pair(to_child_pipe) < 0 || fd_pair(from_child_pipe) < 0) {
        !            56:                rsyserr(FERROR, errno, "pipe");
        !            57:                exit_cleanup(RERR_IPC);
        !            58:        }
        !            59: 
        !            60:        pid = do_fork();
        !            61:        if (pid == -1) {
        !            62:                rsyserr(FERROR, errno, "fork");
        !            63:                exit_cleanup(RERR_IPC);
        !            64:        }
        !            65: 
        !            66:        if (pid == 0) {
        !            67:                if (dup2(to_child_pipe[0], STDIN_FILENO) < 0 ||
        !            68:                    close(to_child_pipe[1]) < 0 ||
        !            69:                    close(from_child_pipe[0]) < 0 ||
        !            70:                    dup2(from_child_pipe[1], STDOUT_FILENO) < 0) {
        !            71:                        rsyserr(FERROR, errno, "Failed to dup/close");
        !            72:                        exit_cleanup(RERR_IPC);
        !            73:                }
        !            74:                if (to_child_pipe[0] != STDIN_FILENO)
        !            75:                        close(to_child_pipe[0]);
        !            76:                if (from_child_pipe[1] != STDOUT_FILENO)
        !            77:                        close(from_child_pipe[1]);
        !            78:                umask(orig_umask);
        !            79:                set_blocking(STDIN_FILENO);
        !            80:                if (blocking_io > 0)
        !            81:                        set_blocking(STDOUT_FILENO);
        !            82:                execvp(command[0], command);
        !            83:                rsyserr(FERROR, errno, "Failed to exec %s", command[0]);
        !            84:                exit_cleanup(RERR_IPC);
        !            85:        }
        !            86: 
        !            87:        if (close(from_child_pipe[1]) < 0 || close(to_child_pipe[0]) < 0) {
        !            88:                rsyserr(FERROR, errno, "Failed to close");
        !            89:                exit_cleanup(RERR_IPC);
        !            90:        }
        !            91: 
        !            92:        *f_in = from_child_pipe[0];
        !            93:        *f_out = to_child_pipe[1];
        !            94: 
        !            95:        return pid;
        !            96: }
        !            97: 
        !            98: /* This function forks a child which calls child_main().  First,
        !            99:  * however, it has to establish communication paths to and from the
        !           100:  * newborn child.  It creates two socket pairs -- one for writing to
        !           101:  * the child (from the parent) and one for reading from the child
        !           102:  * (writing to the parent).  Since that's four socket ends, each
        !           103:  * process has to close the two ends it doesn't need.  The remaining
        !           104:  * two socket ends are retained for reading and writing.  In the
        !           105:  * child, the STDIN and STDOUT file descriptors refer to these
        !           106:  * sockets.  In the parent, the function arguments f_in and f_out are
        !           107:  * set to refer to these sockets. */
        !           108: pid_t local_child(int argc, char **argv, int *f_in, int *f_out,
        !           109:                  int (*child_main)(int, char*[]))
        !           110: {
        !           111:        pid_t pid;
        !           112:        int to_child_pipe[2];
        !           113:        int from_child_pipe[2];
        !           114: 
        !           115:        /* The parent process is always the sender for a local rsync. */
        !           116:        assert(am_sender);
        !           117: 
        !           118:        if (fd_pair(to_child_pipe) < 0 ||
        !           119:            fd_pair(from_child_pipe) < 0) {
        !           120:                rsyserr(FERROR, errno, "pipe");
        !           121:                exit_cleanup(RERR_IPC);
        !           122:        }
        !           123: 
        !           124:        pid = do_fork();
        !           125:        if (pid == -1) {
        !           126:                rsyserr(FERROR, errno, "fork");
        !           127:                exit_cleanup(RERR_IPC);
        !           128:        }
        !           129: 
        !           130:        if (pid == 0) {
        !           131:                am_sender = 0;
        !           132:                am_server = 1;
        !           133:                filesfrom_fd = -1;
        !           134:                chmod_modes = NULL; /* Let the sending side handle this. */
        !           135: 
        !           136:                /* Let the client side handle this. */
        !           137:                if (logfile_name) {
        !           138:                        logfile_name = NULL;
        !           139:                        logfile_close();
        !           140:                }
        !           141: 
        !           142:                if (dup2(to_child_pipe[0], STDIN_FILENO) < 0 ||
        !           143:                    close(to_child_pipe[1]) < 0 ||
        !           144:                    close(from_child_pipe[0]) < 0 ||
        !           145:                    dup2(from_child_pipe[1], STDOUT_FILENO) < 0) {
        !           146:                        rsyserr(FERROR, errno, "Failed to dup/close");
        !           147:                        exit_cleanup(RERR_IPC);
        !           148:                }
        !           149:                if (to_child_pipe[0] != STDIN_FILENO)
        !           150:                        close(to_child_pipe[0]);
        !           151:                if (from_child_pipe[1] != STDOUT_FILENO)
        !           152:                        close(from_child_pipe[1]);
        !           153: #ifdef ICONV_CONST
        !           154:                setup_iconv();
        !           155: #endif
        !           156:                child_main(argc, argv);
        !           157:        }
        !           158: 
        !           159:        if (close(from_child_pipe[1]) < 0 ||
        !           160:            close(to_child_pipe[0]) < 0) {
        !           161:                rsyserr(FERROR, errno, "Failed to close");
        !           162:                exit_cleanup(RERR_IPC);
        !           163:        }
        !           164: 
        !           165:        *f_in = from_child_pipe[0];
        !           166:        *f_out = to_child_pipe[1];
        !           167: 
        !           168:        return pid;
        !           169: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>