factor/native/write.c

133 lines
2.5 KiB
C

#include "factor.h"
/* Return true if write was done */
void write_step(PORT* port)
{
BYTE* chars = (BYTE*)port->buffer + sizeof(STRING);
FIXNUM amount = write(port->fd,chars + port->buf_pos,
port->buf_fill - port->buf_pos);
if(amount == -1)
{
if(errno != EAGAIN)
postpone_io_error(port,__FUNCTION__);
}
else
port->buf_pos += amount;
}
bool can_write(PORT* port, FIXNUM len)
{
CELL buf_capacity;
if(port->type != PORT_WRITE)
general_error(ERROR_INCOMPATIBLE_PORT,tag_object(port));
buf_capacity = port->buffer->capacity * CHARS;
/* Is the string longer than the buffer? */
if(port->buf_fill == 0 && len > buf_capacity)
{
/* Increase the buffer to fit the string */
port->buffer = allot_string(len / CHARS + 1);
return true;
}
else
return (port->buf_fill + len <= buf_capacity);
}
void primitive_can_write(void)
{
PORT* port = untag_port(dpop());
FIXNUM len = to_fixnum(dpop());
pending_io_error(port);
dpush(tag_boolean(can_write(port,len)));
}
void primitive_add_write_io_task(void)
{
CELL callback = dpop();
CELL port = dpop();
add_io_task(IO_TASK_WRITE,port,F,callback,
write_io_tasks,&write_fd_count);
}
bool perform_write_io_task(PORT* port)
{
if(port->buf_pos == port->buf_fill || port->io_error != F)
{
/* Nothing to write */
port->buf_pos = 0;
port->buf_fill = 0;
return true;
}
else
{
write_step(port);
return false;
}
}
void write_char_8(PORT* port, FIXNUM ch)
{
BYTE c = (BYTE)ch;
pending_io_error(port);
if(!can_write(port,1))
io_error(__FUNCTION__);
bput((CELL)port->buffer + sizeof(STRING) + port->buf_fill,c);
port->buf_fill++;
}
/* Caller must ensure buffer is of the right size. */
void write_string_raw(PORT* port, BYTE* str, CELL len)
{
/* Append string to buffer */
memcpy((void*)((CELL)port->buffer + sizeof(STRING)
+ port->buf_fill),str,len);
port->buf_fill += len;
}
void write_string_8(PORT* port, STRING* str)
{
BYTE* c_str;
pending_io_error(port);
/* Note this ensures the buffer is large enough to fit the string */
if(!can_write(port,str->capacity))
io_error(__FUNCTION__);
c_str = to_c_string(str);
write_string_raw(port,c_str,str->capacity);
}
void primitive_write_8(void)
{
PORT* port = untag_port(dpop());
CELL text = dpop();
CELL type = type_of(text);
STRING* str;
pending_io_error(port);
switch(type)
{
case FIXNUM_TYPE:
case BIGNUM_TYPE:
write_char_8(port,to_fixnum(text));
break;
case STRING_TYPE:
str = untag_string(text);
write_string_8(port,str);
break;
default:
type_error(TEXT_TYPE,text);
break;
}
}