Create a pip package for pipe_io_server

This commit is contained in:
Kristóf Tóth
2019-08-02 15:32:38 +02:00
parent 499ce8af76
commit c39527d99e
19 changed files with 98 additions and 0 deletions

View File

@ -0,0 +1,29 @@
#include "pipe_io.h"
void run2() {
pipe_io pipeio = pipeio_new_io("in", "out");
const char* line;
while ((line = pipeio_recv_msg(&pipeio.reader))) {
pipeio_send_msg(&pipeio.writer, line);
}
pipeio_close_io(&pipeio);
}
void msg_handler(const char* msg, void* ctx) {
pipe_writer* writer = (pipe_writer*)(ctx);
pipeio_send_msg(writer, msg);
}
void run1() {
pipe_io pipeio = pipeio_new_io("in", "out");
pipeio_set_msg_handler(&pipeio.reader, &msg_handler, &pipeio.writer);
pipeio_run(&pipeio.reader);
pipeio_close_io(&pipeio);
}
int main() {
run1(); // or run2();
// note that run1 and run2 do the same thing
// and you can use whichever API you prefer
}

View File

@ -0,0 +1,13 @@
#include <iostream>
#include <string>
#include <fstream>
#include "pipe_io.hpp"
int main() {
PipeIO pipe_io("in", "out");
pipe_io.reader.set_message_handler(
[&pipe_io](std::string msg){pipe_io.writer.send_message(msg);}
);
pipe_io.reader.run();
}

View File

@ -0,0 +1,147 @@
#pragma once
#include <stdlib.h>
#include <stdio.h>
#include <stdbool.h>
#include <string.h>
typedef struct pipe_reader {
FILE* stream;
char* line_buf;
size_t buf_size;
void (*msg_handler)(const char*, void*);
void* handler_ctx;
} pipe_reader;
static void null_handler(const char* msg, void* ctx) {}
pipe_reader pipeio_new_reader(const char* pipe_path) {
pipe_reader preader;
preader.stream = fopen(pipe_path, "r");
preader.line_buf = NULL;
preader.buf_size = 0;
preader.msg_handler = &null_handler;
return preader;
}
bool pipeio_check_reader(const pipe_reader* preader) {
if (!preader || preader->stream == NULL) {
return false;
}
return true;
}
const char* pipeio_recv_msg(pipe_reader* preader) {
if (!pipeio_check_reader(preader)) {
return NULL;
}
ssize_t read = getline(&preader->line_buf, &preader->buf_size, preader->stream);
if (read > 0) {
preader->line_buf[read - 1] = '\0';
return preader->line_buf;
}
return NULL;
}
void pipeio_set_msg_handler(pipe_reader* preader, void (*msg_handler)(const char*, void*), void* handler_ctx) {
if (msg_handler != NULL && handler_ctx != NULL) {
preader->msg_handler = msg_handler;
preader->handler_ctx = handler_ctx;
}
}
void pipeio_run(pipe_reader* preader) {
const char* line;
while ((line = pipeio_recv_msg(preader))) {
(*preader->msg_handler)(line, preader->handler_ctx);
}
}
void pipeio_close_reader(pipe_reader* preader) {
if (!pipeio_check_reader(preader)) {
return;
}
fclose(preader->stream);
preader->stream = NULL;
if (preader->line_buf) {
free(preader->line_buf);
preader->line_buf = NULL;
}
}
typedef struct pipe_writer {
FILE* stream;
} pipe_writer;
pipe_writer pipeio_new_writer(const char* pipe_path) {
pipe_writer pwriter;
pwriter.stream = fopen(pipe_path, "w");
return pwriter;
}
bool pipeio_check_writer(const pipe_writer* pwriter) {
if (!pwriter || pwriter->stream == NULL) {
return false;
}
return true;
}
bool pipeio_send_msg(pipe_writer* pwriter, const char* msg) {
if (!pipeio_check_writer(pwriter)) {
return false;
}
fprintf(pwriter->stream, "%s\n", msg);
fflush(pwriter->stream);
return true;
}
void pipeio_close_writer(pipe_writer* pwriter) {
if (!pipeio_check_writer(pwriter)) {
return;
}
fclose(pwriter->stream);
pwriter->stream = NULL;
}
typedef struct pipe_io {
pipe_reader reader;
pipe_writer writer;
} pipe_io;
pipe_io pipeio_new_io(const char* in_pipe_path, const char* out_pipe_path) {
pipe_io pipeio;
pipeio.reader = pipeio_new_reader(in_pipe_path);
pipeio.writer = pipeio_new_writer(out_pipe_path);
return pipeio;
}
bool pipeio_check_io(const pipe_io* pipeio) {
if (!pipeio) {
return false;
}
return (pipeio_check_reader(&pipeio->reader) && pipeio_check_writer(&pipeio->writer));
}
void pipeio_close_io(pipe_io* pipeio) {
if (!pipeio_check_io(pipeio)) {
return;
}
pipeio_close_reader(&pipeio->reader);
pipeio_close_writer(&pipeio->writer);
}
// const char* recv_msg_pipe_io(pipe_io* pipeio) {
// if (!check_pipe_io(pipeio)) {
// return NULL;
// }
// return recv_msg_pipe_reader(&pipeio->preader);
// }
// bool send_msg_pipe_io(pipe_io* pipeio, const char* msg) {
// if (!check_pipe_io(pipeio)) {
// return false;
// }
// return send_msg_pipe_writer(&pipeio->pwriter, msg);
// }

View File

@ -0,0 +1,79 @@
extern "C" {
#include "pipe_io.h"
}
#include <string>
#include <iostream>
#include <functional>
class PipeReader {
public:
PipeReader(const std::string& pipe_path)
:preader(pipeio_new_reader(pipe_path.c_str())) {}
virtual ~PipeReader() {
pipeio_close_reader(&this->preader);
}
PipeReader(PipeReader&) = delete;
PipeReader(PipeReader&&) = delete;
PipeReader& operator=(const PipeReader&) = delete;
PipeReader& operator=(const PipeReader&&) = delete;
void set_message_handler(std::function<void(std::string)> message_handler) {
this->message_handler = message_handler;
}
void run() {
while (const char* linebuf = this->recv_message()) {
std::string msg(linebuf);
this->message_handler(msg);
}
}
const char* recv_message() {
return pipeio_recv_msg(&this->preader);
}
private:
pipe_reader preader;
std::function<void(std::string)> message_handler = [](std::string){};
};
class PipeWriter {
public:
PipeWriter(const std::string& pipe_path)
:pwriter(pipeio_new_writer(pipe_path.c_str())) {}
virtual ~PipeWriter() {
pipeio_close_writer(&this->pwriter);
}
PipeWriter(PipeReader&) = delete;
PipeWriter(PipeReader&&) = delete;
PipeWriter& operator=(const PipeReader&) = delete;
PipeWriter& operator=(const PipeReader&&) = delete;
void send_message(std::string msg) {
pipeio_send_msg(&this->pwriter, msg.c_str());
}
private:
pipe_writer pwriter;
};
class PipeIO {
public:
PipeReader reader;
PipeWriter writer;
PipeIO(const std::string& in_pipe_path, const std::string& out_pipe_path)
:reader(in_pipe_path), writer(out_pipe_path) {}
PipeIO(PipeReader&) = delete;
PipeIO(PipeReader&&) = delete;
PipeIO& operator=(const PipeReader&) = delete;
PipeIO& operator=(const PipeReader&&) = delete;
};