summaryrefslogtreecommitdiff
path: root/src/process.h
blob: 68678bda034949d2e2e9997b11a3d174e9f92959 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
/*
 * Copyright (C) 2007-2010 Xagasoft, All rights reserved.
 *
 * This file is part of the libbu++ library and is released under the
 * terms of the license contained in the file LICENSE.
 */

#ifndef BU_PROCESS_H
#define BU_PROCESS_H

#include <stdint.h>
#include <sys/types.h>

#include "bu/stream.h"
#include "bu/fstring.h"

namespace Bu
{
	/**
	 * Runs a program and attaches streams to it's stdin, stdout, and stderr.
	 * Reading from a Bu::Process will read from the program's standard output,
	 * writing to a Bu::Process will write to the program's standard input.
	 */
	class Process : public Bu::Stream
	{
	public:
		enum Flags
		{
			None	=	0x00,
			StdOut	=	0x01,
			StdErr	=	0x02,
			Both	=	0x03
		};

	public:
		Process( Flags eFlags, const char *sName, char *const argv[] );
		Process( Flags eFlags, const char *sName, const char *argv, ...);
		virtual ~Process();

		/**
		 * Waits until the process exits.  This blocks the caller until the
		 * child process terminates.
		 */
		void wait();

		virtual void close();
		virtual size_t read( void *pBuf, size_t nBytes );
		virtual size_t readErr( void *pBuf, size_t nBytes );
		virtual size_t write( const void *pBuf, size_t nBytes );
		using Stream::write;

		virtual long tell();
		virtual void seek( long offset );
		virtual void setPos( long pos );
		virtual void setPosEnd( long pos );
		virtual bool isEos();
		virtual bool isOpen();

		virtual void flush();

		virtual bool canRead();
		virtual bool canWrite();

		virtual bool isReadable();
		virtual bool isWritable();
		virtual bool isSeekable();

		virtual bool isBlocking();
		virtual void setBlocking( bool bBlocking=true );

		virtual void setSize( long iSize );

		void select( bool &bStdOut, bool &bStdErr );

		bool isRunning();
		void ignoreStdErr();

		/**
		 * Returns the pid of the child process, or zero if there is no
		 * currently running child.  Note that a read operation must be
		 * performed in order to discover that the child has ended.
		 */
		pid_t getPid();

		/**
		 * Returns true if the child exited normally (by calling exit or
		 * returning from main).
		 */
		bool childExited();

		/**
		 * Returns the 8 bit integer value returned from the child program if
		 * childExited returned true.
		 */
		int childExitStatus();

		/**
		 * Returns true if the child exited because of a signal.
		 */
		bool childSignaled();

		/**
		 * Returns the signal ID if the childSignaled return true.
		 */
		int childSignal();

		/**
		 * Returns true if the child left a core dump behind when it exited.
		 */
		bool childCoreDumped();

	private:
		int iStdIn;
		int iStdOut;
		int iStdErr;
		pid_t iPid;
		int iProcStatus;
		bool bBlocking;
		bool bStdOutEos;
		bool bStdErrEos;

		void gexec( Flags eFlags, const char *sName, char *const argv[] );
		void checkClose();
	};
}

#endif