Process now supports setuid for the child process. Really cool.

2 files changed, 67 insertions, 0 deletions

diff --git a/src/process.cpp b/src/process.cpp
index 8ea6ce3..b3f2801 100644
--- a/src/process.cpp
+++ b/src/process.cpp
@@ -61,6 +61,51 @@ Bu::Process::Process( Flags eFlags, const char *sName, const char *argv, ...) :
         delete[] list;
 }
 
+Bu::Process::Process( Flags eFlags, const Bu::Process::Options &opt, const char *sName, char *const argv[] ) :
+        iStdIn( -1 ),
+        iStdOut( -1 ),
+        iStdErr( -1 ),
+        iPid( 0 ),
+        iProcStatus( 0 ),
+        bBlocking( true ),
+        bStdOutEos( true ),
+        bStdErrEos( true ),
+        opt( opt )
+{
+        gexec( eFlags, sName, argv );
+}
+
+Bu::Process::Process( Flags eFlags, const Bu::Process::Options &opt, const char *sName, const char *argv, ...) :
+        iStdIn( -1 ),
+        iStdOut( -1 ),
+        iStdErr( -1 ),
+        iPid( 0 ),
+        iProcStatus( 0 ),
+        bBlocking( true ),
+        bStdOutEos( true ),
+        bStdErrEos( true ),
+        opt( opt )
+{
+        int iCnt = 0;
+        va_list ap;
+        va_start( ap, argv );
+        for(; va_arg( ap, const char *); iCnt++ ) { }
+        va_end( ap );
+
+        char const **list = new char const *[iCnt+2];
+        va_start( ap, argv );
+        list[0] = argv;
+        for( int j = 1; j <= iCnt; j++ )
+        {
+                list[j] = va_arg( ap, const char *);
+        }
+        list[iCnt+1] = NULL;
+        va_end( ap );
+
+        gexec( eFlags, sName, (char *const *)list );
+        delete[] list;
+}
+
 Bu::Process::~Process()
 {
         close();
@@ -109,6 +154,10 @@ void Bu::Process::gexec( Flags eFlags, const char *sName, char *const argv[] )
                         ::close( iaStdErr[0] );
                         dup2( iaStdErr[1], 2 );
                 }
+                if( (opt.eFlags&Options::SetUid) )
+                {
+                        setuid( opt.iUid );
+                }
                 execvp( sName, argv );
                 throw Bu::ExceptionBase("Hey, execvp failed!");
         }
diff --git a/src/process.h b/src/process.h
index 4934408..f3d84b1 100644
--- a/src/process.h
+++ b/src/process.h
@@ -33,8 +33,25 @@ namespace Bu
                 };
 
         public:
+                class Options
+                {
+                public:
+                        enum OptFlags
+                        {
+                                None    = 0x00,
+                                SetUid  = 0x01,
+                        };
+
+                        Options() : eFlags( None ) {}
+
+                        OptFlags eFlags;
+                        int iUid;
+                };
+
                 Process( Flags eFlags, const char *sName, char *const argv[] );
                 Process( Flags eFlags, const char *sName, const char *argv, ...);
+                Process( Flags eFlags, const Options &opt, const char *sName, char *const argv[] );
+                Process( Flags eFlags, const Options &opt, const char *sName, const char *argv, ...);
                 virtual ~Process();
 
                 /**
@@ -125,6 +142,7 @@ namespace Bu
 
                 void gexec( Flags eFlags, const char *sName, char *const argv[] );
                 void checkClose();
+                Options opt;
         };
 }