summaryrefslogtreecommitdiff
path: root/daemon/serve.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'daemon/serve.cpp')
-rw-r--r--daemon/serve.cpp251
1 files changed, 251 insertions, 0 deletions
diff --git a/daemon/serve.cpp b/daemon/serve.cpp
new file mode 100644
index 0000000..aaeb0a5
--- /dev/null
+++ b/daemon/serve.cpp
@@ -0,0 +1,251 @@
+/*
+ This file is part of Icecream.
+
+ Copyright (c) 2004 Stephan Kulow <coolo@suse.de>
+ 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 <setjmp.h>
+#include <unistd.h>
+#include <string.h>
+#include <fcntl.h>
+#include <errno.h>
+#include <signal.h>
+#include <cassert>
+
+#include <sys/stat.h>
+#include <sys/types.h>
+#include <sys/wait.h>
+#ifdef HAVE_SYS_SIGNAL_H
+# include <sys/signal.h>
+#endif /* HAVE_SYS_SIGNAL_H */
+#include <sys/param.h>
+#include <unistd.h>
+
+#include <job.h>
+#include <comm.h>
+
+#include "exitcode.h"
+#include "tempfile.h"
+#include "workit.h"
+#include "logging.h"
+#include "serve.h"
+
+#include <sys/time.h>
+
+#ifdef __FreeBSD__
+#include <sys/socket.h>
+#include <sys/uio.h>
+#endif
+
+#ifndef O_LARGEFILE
+#define O_LARGEFILE 0
+#endif
+
+#ifndef _PATH_TMP
+#define _PATH_TMP "/tmp"
+#endif
+
+using namespace std;
+
+int nice_level = 5;
+
+static void
+error_client( MsgChannel *client, string error )
+{
+ if ( IS_PROTOCOL_22( client ) )
+ client->send_msg( StatusTextMsg( error ) );
+}
+
+/**
+ * Read a request, run the compiler, and send a response.
+ **/
+int handle_connection( const string &basedir, CompileJob *job,
+ MsgChannel *client, int &out_fd,
+ unsigned int mem_limit, uid_t nobody_uid, gid_t nobody_gid )
+{
+ int socket[2];
+ if ( pipe( socket ) == -1)
+ return -1;
+
+ flush_debug();
+ pid_t pid = fork();
+ assert(pid >= 0);
+ if ( pid > 0) { // parent
+ close( socket[1] );
+ out_fd = socket[0];
+ fcntl(out_fd, F_SETFD, FD_CLOEXEC);
+ return pid;
+ }
+
+ reset_debug(0);
+ close( socket[0] );
+ out_fd = socket[1];
+
+ /* internal communication channel, don't inherit to gcc */
+ fcntl(out_fd, F_SETFD, FD_CLOEXEC);
+
+ nice( nice_level );
+
+ Msg *msg = 0; // The current read message
+ unsigned int job_id = 0;
+ int obj_fd = -1; // the obj_fd
+ string obj_file;
+
+ try {
+ if ( job->environmentVersion().size() ) {
+ string dirname = basedir + "/target=" + job->targetPlatform() + "/" + job->environmentVersion();
+ if ( ::access( string( dirname + "/usr/bin/gcc" ).c_str(), X_OK ) ) {
+ error_client( client, dirname + "/usr/bin/gcc is not executable" );
+ log_error() << "I don't have environment " << job->environmentVersion() << "(" << job->targetPlatform() << ") " << job->jobID() << endl;
+ throw myexception( EXIT_DISTCC_FAILED ); // the scheduler didn't listen to us!
+ }
+
+ if ( getuid() == 0 ) {
+ // without the chdir, the chroot will escape the
+ // jail right away
+ if ( chdir( dirname.c_str() ) < 0 ) {
+ error_client( client, string( "chdir to " ) + dirname + "failed" );
+ log_perror("chdir() failed" );
+ _exit(145);
+ }
+ if ( chroot( dirname.c_str() ) < 0 ) {
+ error_client( client, string( "chroot " ) + dirname + "failed" );
+ log_perror("chroot() failed" );
+ _exit(144);
+ }
+ if ( setgid( nobody_gid ) < 0 ) {
+ error_client( client, string( "setgid failed" ));
+ log_perror("setgid() failed" );
+ _exit(143);
+ }
+ if ( setuid( nobody_uid ) < 0) {
+ error_client( client, string( "setuid failed" ));
+ log_perror("setuid() failed" );
+ _exit(142);
+ }
+ }
+ else
+ if ( chdir( dirname.c_str() ) ) {
+ log_perror( "chdir" );
+ } else {
+ trace() << "chdir to " << dirname << endl;
+ }
+ }
+ else
+ chdir( "/" );
+
+ if ( ::access( _PATH_TMP + 1, W_OK ) ) {
+ error_client( client, "can't write to " _PATH_TMP );
+ log_error() << "can't write into " << _PATH_TMP << " " << strerror( errno ) << endl;
+ throw myexception( -1 );
+ }
+
+ int ret;
+ unsigned int job_stat[8];
+ CompileResultMsg rmsg;
+ job_id = job->jobID();
+
+ memset(job_stat, 0, sizeof(job_stat));
+
+ char tmp_output[PATH_MAX];
+ char prefix_output[PATH_MAX]; // I'm too lazy to calculate how many digits 2^64 is :)
+ sprintf( prefix_output, "icecc-%d", job_id );
+
+ if ( ( ret = dcc_make_tmpnam(prefix_output, ".o", tmp_output, 1 ) ) == 0 ) {
+ obj_file = tmp_output;
+ ret = work_it( *job, job_stat, client, rmsg, obj_file, mem_limit, client->fd,
+ -1 );
+ }
+
+ delete job;
+ job = 0;
+
+ if ( ret ) {
+ if ( ret == EXIT_OUT_OF_MEMORY ) { // we catch that as special case
+ rmsg.was_out_of_memory = true;
+ } else {
+ throw myexception( ret );
+ }
+ }
+
+ if ( !client->send_msg( rmsg ) ) {
+ log_info() << "write of result failed\n";
+ throw myexception( EXIT_DISTCC_FAILED );
+ }
+
+ struct stat st;
+ if (!stat(obj_file.c_str(), &st))
+ job_stat[JobStatistics::out_uncompressed] = st.st_size;
+
+ /* wake up parent and tell him that compile finished */
+ /* if the write failed, well, doesn't matter */
+ write( out_fd, job_stat, sizeof( job_stat ) );
+ close( out_fd );
+
+ if ( rmsg.status == 0 ) {
+ obj_fd = open( obj_file.c_str(), O_RDONLY|O_LARGEFILE );
+ if ( obj_fd == -1 ) {
+ log_error() << "open failed\n";
+ error_client( client, "open of object file failed" );
+ throw myexception( EXIT_DISTCC_FAILED );
+ }
+
+ unsigned char buffer[100000];
+ do {
+ ssize_t bytes = read(obj_fd, buffer, sizeof(buffer));
+ if ( bytes < 0 )
+ {
+ if ( errno == EINTR )
+ continue;
+ throw myexception( EXIT_DISTCC_FAILED );
+ }
+ if ( !bytes )
+ break;
+ FileChunkMsg fcmsg( buffer, bytes );
+ if ( !client->send_msg( fcmsg ) ) {
+ log_info() << "write of obj chunk failed " << bytes << endl;
+ throw myexception( EXIT_DISTCC_FAILED );
+ }
+ } while (1);
+ }
+
+ throw myexception( rmsg.status );
+
+ } catch ( myexception e )
+ {
+ if ( client && e.exitcode() == 0 )
+ client->send_msg( EndMsg() );
+ delete client;
+ client = 0;
+
+ delete msg;
+ delete job;
+
+ if ( obj_fd > -1)
+ close( obj_fd );
+
+ if ( !obj_file.empty() )
+ unlink( obj_file.c_str() );
+
+ _exit( e.exitcode() );
+ }
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-23 20:27:06 +0000