Subprocess with pipe capturing
This commit is contained in:
parent
7a3f1955b9
commit
a0da4ce1a3
6 changed files with 225 additions and 0 deletions
1
src/Subprocess/CMakeLists.txt
Normal file
1
src/Subprocess/CMakeLists.txt
Normal file
|
@ -0,0 +1 @@
|
||||||
|
add_library(Subprocess STATIC Subprocess.c)
|
109
src/Subprocess/Subprocess.c
Normal file
109
src/Subprocess/Subprocess.c
Normal file
|
@ -0,0 +1,109 @@
|
||||||
|
#include "Subprocess.h"
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <sys/wait.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
typedef struct PipePair_s {
|
||||||
|
int read;
|
||||||
|
int write;
|
||||||
|
} PipePair;
|
||||||
|
|
||||||
|
static int _PipePair_Create(PipePair* pair)
|
||||||
|
{
|
||||||
|
int pipefds[2];
|
||||||
|
int pipe_call = pipe(pipefds);
|
||||||
|
|
||||||
|
if (pipe_call == 0) {
|
||||||
|
pair->read = pipefds[0];
|
||||||
|
pair->write = pipefds[1];
|
||||||
|
}
|
||||||
|
|
||||||
|
return pipe_call;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void _PipePair_Destroy(PipePair* pair)
|
||||||
|
{
|
||||||
|
close(pair->read);
|
||||||
|
close(pair->write);
|
||||||
|
}
|
||||||
|
|
||||||
|
int Subprocess_Create(Subprocess* subprocess, char* file, char** argv)
|
||||||
|
{
|
||||||
|
if (subprocess == NULL) {
|
||||||
|
return EDESTADDRREQ;
|
||||||
|
}
|
||||||
|
if (file == NULL) {
|
||||||
|
return EINVAL;
|
||||||
|
}
|
||||||
|
if (argv == NULL) {
|
||||||
|
return EINVAL;
|
||||||
|
}
|
||||||
|
if (subprocess->in.type != SUBPROCESSREDIRECTIONTYPE_PIPE) {
|
||||||
|
return ENOTSUP;
|
||||||
|
}
|
||||||
|
if (subprocess->out.type != SUBPROCESSREDIRECTIONTYPE_PIPE) {
|
||||||
|
return ENOTSUP;
|
||||||
|
}
|
||||||
|
if (subprocess->err.type != SUBPROCESSREDIRECTIONTYPE_PIPE) {
|
||||||
|
return ENOTSUP;
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: Implement FILE redirection
|
||||||
|
|
||||||
|
// Create pipe pairs
|
||||||
|
PipePair in_pair, out_pair, err_pair;
|
||||||
|
if (_PipePair_Create(&in_pair)) {
|
||||||
|
return errno;
|
||||||
|
}
|
||||||
|
if (_PipePair_Create(&out_pair)) {
|
||||||
|
_PipePair_Destroy(&in_pair);
|
||||||
|
return errno;
|
||||||
|
}
|
||||||
|
if (_PipePair_Create(&err_pair)) {
|
||||||
|
_PipePair_Destroy(&in_pair);
|
||||||
|
_PipePair_Destroy(&out_pair);
|
||||||
|
return errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
pid_t fork_result = fork();
|
||||||
|
if (fork_result == 0) {
|
||||||
|
// child
|
||||||
|
// TODO: Check errors
|
||||||
|
|
||||||
|
close(in_pair.write);
|
||||||
|
close(out_pair.read);
|
||||||
|
close(err_pair.read);
|
||||||
|
dup2(in_pair.read, STDIN_FILENO);
|
||||||
|
dup2(out_pair.write, STDOUT_FILENO);
|
||||||
|
dup2(err_pair.write, STDERR_FILENO);
|
||||||
|
execvp(file, argv);
|
||||||
|
|
||||||
|
} else if (fork_result != -1) {
|
||||||
|
// parent
|
||||||
|
|
||||||
|
close(in_pair.read);
|
||||||
|
close(out_pair.write);
|
||||||
|
close(err_pair.write);
|
||||||
|
subprocess->in.get.pipefd = in_pair.write;
|
||||||
|
subprocess->out.get.pipefd = out_pair.read;
|
||||||
|
subprocess->err.get.pipefd = err_pair.read;
|
||||||
|
subprocess->child_pid = fork_result;
|
||||||
|
|
||||||
|
} else {
|
||||||
|
// error
|
||||||
|
return errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return EXIT_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void Subprocess_Destroy(Subprocess* subprocess)
|
||||||
|
{
|
||||||
|
waitpid(subprocess->child_pid, NULL, 0);
|
||||||
|
|
||||||
|
if (subprocess->in .type == SUBPROCESSREDIRECTIONTYPE_PIPE) close(subprocess->in.get.pipefd);
|
||||||
|
if (subprocess->out.type == SUBPROCESSREDIRECTIONTYPE_PIPE) close(subprocess->out.get.pipefd);
|
||||||
|
if (subprocess->err.type == SUBPROCESSREDIRECTIONTYPE_PIPE) close(subprocess->err.get.pipefd);
|
||||||
|
}
|
43
src/Subprocess/Subprocess.h
Normal file
43
src/Subprocess/Subprocess.h
Normal file
|
@ -0,0 +1,43 @@
|
||||||
|
#ifndef UTILITIEC_SUBPROCESS_H
|
||||||
|
#define UTILITIEC_SUBPROCESS_H
|
||||||
|
|
||||||
|
#include "../StringView/StringView.h"
|
||||||
|
#include <sys/types.h>
|
||||||
|
|
||||||
|
// TODO: Implement FILE redirection
|
||||||
|
enum SubprocessRedirectionType {
|
||||||
|
SUBPROCESSREDIRECTIONTYPE_PIPE,
|
||||||
|
SUBPROCESSREDIRECTIONTYPE_FILE,
|
||||||
|
};
|
||||||
|
|
||||||
|
union SubprocessRedirectionContent {
|
||||||
|
int pipefd;
|
||||||
|
StringView file_name;
|
||||||
|
};
|
||||||
|
|
||||||
|
typedef struct SubprocessRedirection_s {
|
||||||
|
enum SubprocessRedirectionType type;
|
||||||
|
union SubprocessRedirectionContent get;
|
||||||
|
} SubprocessRedirection;
|
||||||
|
|
||||||
|
typedef struct Subprocess_t {
|
||||||
|
pid_t child_pid;
|
||||||
|
SubprocessRedirection in;
|
||||||
|
SubprocessRedirection out;
|
||||||
|
SubprocessRedirection err;
|
||||||
|
} Subprocess;
|
||||||
|
|
||||||
|
|
||||||
|
#if defined(__unix__) || (defined(__APPLE__) && defined(__MACH__))
|
||||||
|
|
||||||
|
|
||||||
|
int Subprocess_Create(Subprocess* subprocess, char* file, char** argv);
|
||||||
|
void Subprocess_Destroy(Subprocess* subprocess);
|
||||||
|
|
||||||
|
|
||||||
|
#else
|
||||||
|
// TODO: Implement for Windows with a windows testing machine
|
||||||
|
#error "This module is not implemented for your platform yet"
|
||||||
|
#endif // platform
|
||||||
|
|
||||||
|
#endif // header
|
|
@ -44,3 +44,11 @@ target_link_libraries(BinaryTree-test
|
||||||
pointers
|
pointers
|
||||||
allocator-interface
|
allocator-interface
|
||||||
)
|
)
|
||||||
|
|
||||||
|
add_executable(Subprocess-test Subprocess.test.c)
|
||||||
|
target_link_libraries(Subprocess-test
|
||||||
|
Subprocess
|
||||||
|
threading
|
||||||
|
pointers
|
||||||
|
allocator-interface
|
||||||
|
)
|
||||||
|
|
31
tests/Subprocess.test.c
Normal file
31
tests/Subprocess.test.c
Normal file
|
@ -0,0 +1,31 @@
|
||||||
|
#include "../src/Subprocess/Subprocess.h"
|
||||||
|
#include <assert.h>
|
||||||
|
#include <stdbool.h>
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <unistd.h>
|
||||||
|
|
||||||
|
char read_buffer[128];
|
||||||
|
|
||||||
|
void testEcho()
|
||||||
|
{
|
||||||
|
Subprocess sub;
|
||||||
|
sub.in .type = SUBPROCESSREDIRECTIONTYPE_PIPE;
|
||||||
|
sub.out.type = SUBPROCESSREDIRECTIONTYPE_PIPE;
|
||||||
|
sub.err.type = SUBPROCESSREDIRECTIONTYPE_PIPE;
|
||||||
|
|
||||||
|
char* argv[] = { "echo", "Hello World!", NULL };
|
||||||
|
|
||||||
|
assert(EXIT_SUCCESS == Subprocess_Create(&sub, "/usr/bin/echo", argv));
|
||||||
|
ssize_t read_size = read(sub.out.get.pipefd, read_buffer, 128);
|
||||||
|
assert(read_size == 13);
|
||||||
|
assert(strcmp(read_buffer, "Hello World!\n") == 0);
|
||||||
|
assert(read(sub.out.get.pipefd, read_buffer, 128) == 0);
|
||||||
|
Subprocess_Destroy(&sub);
|
||||||
|
}
|
||||||
|
|
||||||
|
int main()
|
||||||
|
{
|
||||||
|
testEcho();
|
||||||
|
return 0;
|
||||||
|
}
|
|
@ -79,6 +79,38 @@ void testGroup(void)
|
||||||
assert(RegexMatch_HaveNumberedCapture(&match, 1));
|
assert(RegexMatch_HaveNumberedCapture(&match, 1));
|
||||||
assert(RegexMatch_HaveNumberedCapture(&match, 2));
|
assert(RegexMatch_HaveNumberedCapture(&match, 2));
|
||||||
|
|
||||||
|
Regex_Destroy(®ex);
|
||||||
|
RegexMatch_Destroy(&match);
|
||||||
|
assert(allocator.reserved == 0);
|
||||||
|
Allocator_DestroySystemAllocator(&allocator);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
void testOption(void)
|
||||||
|
{
|
||||||
|
StringView regex_string = StringView_FromString("abc|def|");
|
||||||
|
StringView match_string_0 = StringView_FromString("abc");
|
||||||
|
StringView match_string_1 = StringView_FromString("def");
|
||||||
|
StringView match_string_2 = StringView_FromString("");
|
||||||
|
allocator_t allocator;
|
||||||
|
Allocator_CreateSystemAllocator(&allocator);
|
||||||
|
|
||||||
|
Regex regex;
|
||||||
|
assert(Regex_Create(®ex, &allocator) == EXIT_SUCCESS);
|
||||||
|
|
||||||
|
assert(Regex_Compile(®ex, regex_string) == EXIT_SUCCESS);
|
||||||
|
|
||||||
|
RegexMatch match;
|
||||||
|
|
||||||
|
assert(Regex_FirstMatch(®ex, match_string_0, &match) == EXIT_SUCCESS);
|
||||||
|
assert(StringView_Equal(match.match, match_string_0));
|
||||||
|
|
||||||
|
assert(Regex_FirstMatch(®ex, match_string_1, &match) == EXIT_SUCCESS);
|
||||||
|
assert(StringView_Equal(match.match, match_string_1));
|
||||||
|
|
||||||
|
assert(Regex_FirstMatch(®ex, match_string_2, &match) == EXIT_SUCCESS);
|
||||||
|
assert(StringView_Equal(match.match, match_string_2));
|
||||||
|
|
||||||
Regex_Destroy(®ex);
|
Regex_Destroy(®ex);
|
||||||
assert(allocator.reserved == 0);
|
assert(allocator.reserved == 0);
|
||||||
Allocator_DestroySystemAllocator(&allocator);
|
Allocator_DestroySystemAllocator(&allocator);
|
||||||
|
@ -118,6 +150,7 @@ int main()
|
||||||
testLiteral();
|
testLiteral();
|
||||||
testBackslash();
|
testBackslash();
|
||||||
testGroup();
|
testGroup();
|
||||||
|
testOption();
|
||||||
testQuantifier();
|
testQuantifier();
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue