File: Pipe.cpp

package info (click to toggle)
libassa 3.5.0-1
  • links: PTS, VCS
  • area: main
  • in suites: lenny
  • size: 3,084 kB
  • ctags: 2,323
  • sloc: cpp: 15,641; sh: 8,704; makefile: 372; perl: 51
file content (154 lines) | stat: -rw-r--r-- 3,417 bytes parent folder | download | duplicates (9)
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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
// -*- c++ -*-
//------------------------------------------------------------------------------
// $Id: Pipe.cpp,v 1.5 2006/07/20 02:30:54 vlg Exp $
//------------------------------------------------------------------------------
//                            Pipe.cpp
//------------------------------------------------------------------------------
//  Copyright (c) 1997-2002,2005 by Vladislav Grinchenko
//
//  This library is free software; you can redistribute it and/or
//  modify it under the terms of the GNU Library General Public
//  License as published by the Free Software Foundation; either
//  version 2 of the License, or (at your option) any later version.
//------------------------------------------------------------------------------
//
// Pipe class is based on ./unixsysdevel-734/homework3/src/bin/pipe/pipe.c
// code I wrote for UNIX system development class of JohnsHopking 
// part time engineering school in November 1997. 
// The class implementation is a rip off of Stevens' popen ()/ pclose()
// function, p. 438.
//
//------------------------------------------------------------------------------

#include <errno.h>
#include <sys/types.h>			// kill (2)
#include <signal.h>				// kill (2)
#include <unistd.h>				// pipe (2)

#include "assa/Pipe.h"
#include "assa/Fork.h"

using namespace ASSA;

Pipe::
Pipe ()
	: m_fp (NULL), 
	  m_child_pid (0)
 {
	trace_with_mask("Pipe::Pipe", PIPE);
	/* no-op */
}

Pipe::
~Pipe ()
{
	trace_with_mask("Pipe::~Pipe", PIPE);
	close ();
}

FILE*
Pipe::
open (const string& cmd_, const string& type_)
{
	trace_with_mask("Pipe::open", PIPE);

#if !defined(WIN32)       // not yet implemented

	if (type_ != "r" && type_ != "w") {
		EL((ASSAERR,"Wrong type \"%s\"\n", type_.c_str ()));
		errno = EINVAL;
		return NULL;
	}
	
	int fd [2];	
	if (pipe (fd) < 0) {
		EL((ASSAERR,"failed: pipe(2)\n"));
		return NULL;
	}
	Fork f (Fork::KILL_ON_EXIT, Fork::IGNORE_STATUS);

	if (f.isChild ()) {
		if (type_ == "r") {
			::close (fd [0]);
			if (fd [1] != STDOUT_FILENO) {
				dup2 (fd [1], STDOUT_FILENO);
				::close (fd [1]);
			}
		}
		else {					// 'w'
			::close (fd [1]);
			if (fd [0] != STDIN_FILENO) {
				dup2 (fd [0], STDIN_FILENO);
				::close (fd [0]);
			}
		}

		DL((PIPE,"Executing cmd: \"%s\"\n", cmd_.c_str ()));
		execl ("/bin/sh", "sh", "-c", cmd_.c_str (), (char* ) 0);
		EL((ASSAERR,"failed: execl(2)\n"));
		_exit (127);
	}
	/* parent */
	if (type_ == "r") {
		::close (fd [1]);
		if ((m_fp = fdopen (fd [0], type_.c_str ())) == NULL) {
			EL((ASSAERR,"failed: fdopen ()\n"));
			return NULL;
		}
	}
	else {					// 'w'
		::close (fd [0]);
		if ((m_fp = fdopen (fd [1], type_.c_str ())) == NULL) {
			EL((ASSAERR,"failed: fdopen ()\n"));
			return NULL;
		}
	}
	m_child_pid = f.getChildPID ();
	DL((PIPE,"m_child_pid = %d\n",m_child_pid));
	return m_fp;
	
#else
     DL((ASSAERR|PIPE,"Not implemented for win32!\n"));
     return NULL;
#endif
}

int 
Pipe::
kill ()
{
	trace_with_mask("Pipe::kill", PIPE);

#if !defined(WIN32)
	if (m_child_pid == 0) return -1;

	int ret = ::kill (m_child_pid, SIGTERM);
	close ();
	return ret;
#else
     DL((ASSAERR|PIPE,"Not implemented for win32!\n"));
     return -1;
#endif
}

int 
Pipe::
close ()
{
	trace_with_mask("Pipe::close", PIPE);

	int ret = 0;
	if (m_child_pid == 0) {
		ret = EOF;
	}

	if (m_fp) {
		ret = fclose (m_fp);
	}
	m_fp = NULL;
	m_child_pid = 0;
	return ret == EOF ? -1 : 0;
}