diff options
Diffstat (limited to 'client/util.cpp')
| -rw-r--r-- | client/util.cpp | 231 |
1 files changed, 231 insertions, 0 deletions
diff --git a/client/util.cpp b/client/util.cpp new file mode 100644 index 0000000..3448cfa --- /dev/null +++ b/client/util.cpp @@ -0,0 +1,231 @@ +/* -*- c-file-style: "java"; indent-tabs-mode: nil; fill-column: 78; -*- + * + * distcc -- A simple distributed compiler system + * + * Copyright (C) 2002, 2003 by Martin Pool <mbp@samba.org> + * + * This program is free software; you can redistribute it and/or + * modify it under the terms of the GNU General Public License as + * published by the Free Software Foundation; either version 2 of the + * License, or (at your option) any later version. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * along with this program; if not, write to the Free Software + * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 + * USA + */ + +#include "config.h" + +#include <stdio.h> +#include <stdlib.h> +#include <unistd.h> +#include <string.h> +#include <errno.h> +#include <fcntl.h> +#include <signal.h> +#include <time.h> + +#include <sys/types.h> +#include <pwd.h> + +#include <sys/stat.h> +#include <sys/file.h> + +#include "exitcode.h" +#include "logging.h" +#include "util.h" + +using namespace std; + +/** + * Set the `FD_CLOEXEC' flag of DESC if VALUE is nonzero, + * or clear the flag if VALUE is 0. + * + * From the GNU C Library examples. + * + * @returns 0 on success, or -1 on error with `errno' set. + **/ +int set_cloexec_flag (int desc, int value) +{ + int oldflags = fcntl (desc, F_GETFD, 0); + /* If reading the flags failed, return error indication now. */ + if (oldflags < 0) + return oldflags; + /* Set just the flag we want to set. */ + if (value != 0) + oldflags |= FD_CLOEXEC; + else + oldflags &= ~FD_CLOEXEC; + /* Store modified flag word in the descriptor. */ + return fcntl (desc, F_SETFD, oldflags); +} + +/** + * Ignore or unignore SIGPIPE. + * + * The server and child ignore it, because distcc code wants to see + * EPIPE errors if something goes wrong. However, for invoked + * children it is set back to the default value, because they may not + * handle the error properly. + **/ +int dcc_ignore_sigpipe(int val) +{ + if (signal(SIGPIPE, val ? SIG_IGN : SIG_DFL) == SIG_ERR) { + log_warning() << "signal(SIGPIPE, " << ( val ? "ignore" : "default" ) << ") failed: " + << strerror(errno) << endl; + return EXIT_DISTCC_FAILED; + } + return 0; +} + +/** + * Return a pointer to the basename of the file (everything after the + * last slash.) If there is no slash, return the whole filename, + * which is presumably in the current directory. + **/ +string find_basename(const string &sfile) +{ + size_t index = sfile.find_last_of( '/' ); + if ( index == string::npos ) + return sfile; + return sfile.substr( index + 1); +} + + +/** + * Get an exclusive, non-blocking lock on a file using whatever method + * is available on this system. + * + * @retval 0 if we got the lock + * @retval -1 with errno set if the file is already locked. + **/ +static int sys_lock(int fd, bool block) +{ +#if defined(F_SETLK) + struct flock lockparam; + + lockparam.l_type = F_WRLCK; + lockparam.l_whence = SEEK_SET; + lockparam.l_start = 0; + lockparam.l_len = 0; /* whole file */ + + return fcntl(fd, block ? F_SETLKW : F_SETLK, &lockparam); +#elif defined(HAVE_FLOCK) + return flock(fd, LOCK_EX | (block ? 0 : LOCK_NB)); +#elif defined(HAVE_LOCKF) + return lockf(fd, block ? F_LOCK : F_TLOCK, 0); +#else +# error "No supported lock method. Please port this code." +#endif +} + + +bool dcc_unlock(int lock_fd) +{ + /* All our current locks can just be closed */ + if (close(lock_fd)) { + log_perror("close failed:"); + return false; + } + return true; +} + + +/** + * Open a lockfile, creating if it does not exist. + **/ +static bool dcc_open_lockfile(const string &fname, int &plockfd) +{ + /* Create if it doesn't exist. We don't actually do anything with + * the file except lock it. + * + * The file is created with the loosest permissions allowed by the user's + * umask, to give the best chance of avoiding problems if they should + * happen to use a shared lock dir. */ + plockfd = open(fname.c_str(), O_WRONLY|O_CREAT, 0666); + if (plockfd == -1 && errno != EEXIST) { + log_error() << "failed to creat " << fname << ": " << strerror(errno) << endl; + return false; + } + + return true; +} + +bool dcc_lock_host(int &lock_fd) +{ + string fname = "/tmp/.icecream-"; + struct passwd *pwd = getpwuid( getuid() ); + if ( pwd ) + fname += pwd->pw_name; + else { + char buffer[10]; + sprintf( buffer, "%ld", ( long )getuid() ); + fname += buffer; + } + + if ( mkdir( fname.c_str(), 0700 ) && errno != EEXIST ) { + log_perror( "mkdir" ); + return false; + } + + fname += "/local_lock"; + + lock_fd = 0; + if (!dcc_open_lockfile(fname, lock_fd) ) + return false; + + if (sys_lock(lock_fd, true) == 0) { + return true; + } else { + switch (errno) { +#if defined(EWOULDBLOCK) && EWOULDBLOCK != EAGAIN + case EWOULDBLOCK: +#endif + case EAGAIN: + case EACCES: /* HP-UX and Cygwin give this for exclusion */ + trace() << fname << " is busy" << endl; + break; + default: + log_error() << "lock " << fname << " failed: " << strerror(errno) << endl; + break; + } + + ::close(lock_fd); + return false; + } +} + +bool colorify_wanted() +{ + const char* term_env = getenv("TERM"); + + return isatty(2) && !getenv("EMACS") && term_env && strcasecmp(term_env, "DUMB"); +} + +void colorify_output(const string& _s_ccout) +{ + string s_ccout(_s_ccout); + string::size_type end; + + while ( (end = s_ccout.find('\n')) != string::npos ) { + + string cline = s_ccout.substr(string::size_type(0), end ); + s_ccout = s_ccout.substr(end+1); + + if (cline.find(": error:") != string::npos) + fprintf(stderr, "\x1b[1;31m%s\x1b[0m\n", cline.c_str()); + else if (cline.find(": warning:") != string::npos) + fprintf(stderr, "\x1b[36m%s\x1b[0m\n", cline.c_str()); + else + fprintf(stderr, "%s\n", cline.c_str()); + } + fprintf(stderr, "%s", s_ccout.c_str()); +} + + |