make buffered_io available for c code too.

This commit is contained in:
Tom van Dijck 2017-02-08 11:58:50 -08:00
parent 5a4ad51aec
commit fda1845688
2 changed files with 58 additions and 21 deletions

View File

@ -7,27 +7,24 @@
#include <stdlib.h> #include <stdlib.h>
#include <string.h> #include <string.h>
#include "premake.h" #include "premake.h"
#include "buffered_io.h"
typedef struct struct_Buffer void buffer_init(Buffer* b)
{ {
size_t capacity;
size_t length;
char* data;
} Buffer;
int buffered_new(lua_State* L)
{
Buffer* b = (Buffer*)malloc(sizeof(Buffer));
b->capacity = 0; b->capacity = 0;
b->length = 0; b->length = 0;
b->data = NULL; b->data = NULL;
lua_pushlightuserdata(L, b);
return 1;
} }
void buffer_destroy(Buffer* b)
{
free(b->data);
b->capacity = 0;
b->length = 0;
b->data = NULL;
}
static void do_write(Buffer* b, const char *s, size_t len) void buffer_puts(Buffer* b, const void* ptr, size_t len)
{ {
char* data; char* data;
@ -52,10 +49,29 @@ static void do_write(Buffer* b, const char *s, size_t len)
b->capacity = cap; b->capacity = cap;
} }
memcpy(b->data + b->length, s, len); memcpy(b->data + b->length, ptr, len);
b->length += len; b->length += len;
} }
void buffer_printf(Buffer* b, const char *fmt, ...)
{
char text[2048];
va_list args;
va_start(args, fmt);
int len = vsnprintf(text, sizeof(text) - 1, fmt, args);
va_end(args);
buffer_puts(b, text, len);
}
// -- Lua wrappers ----------------------------------------
int buffered_new(lua_State* L)
{
Buffer* b = (Buffer*)malloc(sizeof(Buffer));
buffer_init(b);
lua_pushlightuserdata(L, b);
return 1;
}
int buffered_write(lua_State* L) int buffered_write(lua_State* L)
{ {
@ -63,11 +79,10 @@ int buffered_write(lua_State* L)
const char *s = luaL_checklstring(L, 2, &len); const char *s = luaL_checklstring(L, 2, &len);
Buffer* b = (Buffer*)lua_touserdata(L, 1); Buffer* b = (Buffer*)lua_touserdata(L, 1);
do_write(b, s, len); buffer_puts(b, s, len);
return 0; return 0;
} }
int buffered_writeln(lua_State* L) int buffered_writeln(lua_State* L)
{ {
size_t len; size_t len;
@ -75,21 +90,19 @@ int buffered_writeln(lua_State* L)
Buffer* b = (Buffer*)lua_touserdata(L, 1); Buffer* b = (Buffer*)lua_touserdata(L, 1);
if (s != NULL) if (s != NULL)
do_write(b, s, len); buffer_puts(b, s, len);
do_write(b, "\r\n", 2); buffer_puts(b, "\r\n", 2);
return 0; return 0;
} }
int buffered_close(lua_State* L) int buffered_close(lua_State* L)
{ {
Buffer* b = (Buffer*)lua_touserdata(L, 1); Buffer* b = (Buffer*)lua_touserdata(L, 1);
free(b->data); buffer_destroy(b);
free(b); free(b);
return 0; return 0;
} }
int buffered_tostring(lua_State* L) int buffered_tostring(lua_State* L)
{ {
Buffer* b = (Buffer*)lua_touserdata(L, 1); Buffer* b = (Buffer*)lua_touserdata(L, 1);

24
src/host/buffered_io.h Normal file
View File

@ -0,0 +1,24 @@
/**
* \file buffered_io.h
* \brief provide buffered io.
* \author Copyright (c) 2014
*/
#ifndef buffered_io_h
#define buffered_io_h
#include <stdio.h>
typedef struct struct_Buffer
{
size_t capacity;
size_t length;
char* data;
} Buffer;
void buffer_init(Buffer* b);
void buffer_destroy(Buffer* b);
void buffer_puts(Buffer* b, const void* ptr, size_t len);
void buffer_printf(Buffer* b, const char* s, ...);
#endif