16#include "internal/array.h"
17#include "internal/bits.h"
18#include "internal/error.h"
19#include "internal/gc.h"
20#include "internal/numeric.h"
21#include "internal/string.h"
22#include "internal/io.h"
25VALUE rb_eIOBufferLockedError;
26VALUE rb_eIOBufferAllocationError;
27VALUE rb_eIOBufferAccessError;
28VALUE rb_eIOBufferInvalidatedError;
29VALUE rb_eIOBufferMaskError;
31size_t RUBY_IO_BUFFER_PAGE_SIZE;
32size_t RUBY_IO_BUFFER_DEFAULT_SIZE;
41 RB_IO_BUFFER_HEXDUMP_DEFAULT_WIDTH = 16,
43 RB_IO_BUFFER_INSPECT_HEXDUMP_MAXIMUM_SIZE = 256,
44 RB_IO_BUFFER_INSPECT_HEXDUMP_WIDTH = 16,
47 RB_IO_BUFFER_FLAGS_MASK = RB_IO_BUFFER_EXTERNAL | RB_IO_BUFFER_INTERNAL | RB_IO_BUFFER_MAPPED | RB_IO_BUFFER_SHARED | RB_IO_BUFFER_LOCKED | RB_IO_BUFFER_PRIVATE | RB_IO_BUFFER_READONLY,
49 RB_IO_BUFFER_DEBUG = 0,
55 enum rb_io_buffer_flags flags;
65io_buffer_map_memory(
size_t size,
int flags)
68 void * base = VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
71 rb_sys_fail(
"io_buffer_map_memory:VirtualAlloc");
74 int mmap_flags = MAP_ANONYMOUS;
75 if (flags & RB_IO_BUFFER_SHARED) {
76 mmap_flags |= MAP_SHARED;
79 mmap_flags |= MAP_PRIVATE;
82 void * base = mmap(NULL, size, PROT_READ | PROT_WRITE, mmap_flags, -1, 0);
84 if (base == MAP_FAILED) {
85 rb_sys_fail(
"io_buffer_map_memory:mmap");
88 ruby_annotate_mmap(base, size,
"Ruby:io_buffer_map_memory");
95io_buffer_map_file(
struct rb_io_buffer *buffer,
int descriptor,
size_t size, rb_off_t offset,
enum rb_io_buffer_flags flags)
98 HANDLE file = (HANDLE)_get_osfhandle(descriptor);
99 if (!file) rb_sys_fail(
"io_buffer_map_descriptor:_get_osfhandle");
101 DWORD protect = PAGE_READONLY, access = FILE_MAP_READ;
103 if (flags & RB_IO_BUFFER_READONLY) {
104 buffer->flags |= RB_IO_BUFFER_READONLY;
107 protect = PAGE_READWRITE;
108 access = FILE_MAP_WRITE;
111 if (flags & RB_IO_BUFFER_PRIVATE) {
112 protect = PAGE_WRITECOPY;
113 access = FILE_MAP_COPY;
114 buffer->flags |= RB_IO_BUFFER_PRIVATE;
118 buffer->flags |= RB_IO_BUFFER_EXTERNAL;
119 buffer->flags |= RB_IO_BUFFER_SHARED;
122 HANDLE mapping = CreateFileMapping(file, NULL, protect, 0, 0, NULL);
123 if (RB_IO_BUFFER_DEBUG) fprintf(stderr,
"io_buffer_map_file:CreateFileMapping -> %p\n", mapping);
124 if (!mapping) rb_sys_fail(
"io_buffer_map_descriptor:CreateFileMapping");
126 void *base = MapViewOfFile(mapping, access, (DWORD)(offset >> 32), (DWORD)(offset & 0xFFFFFFFF), size);
129 CloseHandle(mapping);
130 rb_sys_fail(
"io_buffer_map_file:MapViewOfFile");
133 buffer->mapping = mapping;
135 int protect = PROT_READ, access = 0;
137 if (flags & RB_IO_BUFFER_READONLY) {
138 buffer->flags |= RB_IO_BUFFER_READONLY;
141 protect |= PROT_WRITE;
144 if (flags & RB_IO_BUFFER_PRIVATE) {
145 buffer->flags |= RB_IO_BUFFER_PRIVATE;
146 access |= MAP_PRIVATE;
150 buffer->flags |= RB_IO_BUFFER_EXTERNAL;
151 buffer->flags |= RB_IO_BUFFER_SHARED;
152 access |= MAP_SHARED;
155 void *base = mmap(NULL, size, protect, access, descriptor, offset);
157 if (base == MAP_FAILED) {
158 rb_sys_fail(
"io_buffer_map_file:mmap");
165 buffer->flags |= RB_IO_BUFFER_MAPPED;
166 buffer->flags |= RB_IO_BUFFER_FILE;
170io_buffer_experimental(
void)
172 static int warned = 0;
180 "IO::Buffer is experimental and both the Ruby and C interface may change in the future!"
191 buffer->mapping = NULL;
193 buffer->source =
Qnil;
197io_buffer_initialize(
VALUE self,
struct rb_io_buffer *buffer,
void *base,
size_t size,
enum rb_io_buffer_flags flags,
VALUE source)
204 if (flags & RB_IO_BUFFER_INTERNAL) {
205 base = calloc(size, 1);
207 else if (flags & RB_IO_BUFFER_MAPPED) {
208 base = io_buffer_map_memory(size, flags);
212 rb_raise(rb_eIOBufferAllocationError,
"Could not allocate buffer!");
222 buffer->flags = flags;
226 buffer->mapping = NULL;
234 if (buffer->flags & RB_IO_BUFFER_INTERNAL) {
238 if (buffer->flags & RB_IO_BUFFER_MAPPED) {
240 if (buffer->flags & RB_IO_BUFFER_FILE) {
241 UnmapViewOfFile(buffer->base);
244 VirtualFree(buffer->base, 0, MEM_RELEASE);
247 munmap(buffer->base, buffer->size);
261 buffer->source =
Qnil;
265 if (buffer->mapping) {
266 if (RB_IO_BUFFER_DEBUG) fprintf(stderr,
"io_buffer_free:CloseHandle -> %p\n", buffer->mapping);
267 if (!CloseHandle(buffer->mapping)) {
268 fprintf(stderr,
"io_buffer_free:GetLastError -> %lu\n", GetLastError());
270 buffer->mapping = NULL;
276rb_io_buffer_type_mark(
void *_buffer)
279 if (buffer->source !=
Qnil) {
283 rb_gc_mark(buffer->source);
285 rb_gc_mark_movable(buffer->source);
291rb_io_buffer_type_compact(
void *_buffer)
294 if (buffer->source !=
Qnil) {
299 buffer->source = rb_gc_location(buffer->source);
305rb_io_buffer_type_free(
void *_buffer)
309 io_buffer_free(buffer);
313rb_io_buffer_type_size(
const void *_buffer)
319 total += buffer->size;
328 .dmark = rb_io_buffer_type_mark,
329 .dfree = rb_io_buffer_type_free,
330 .dsize = rb_io_buffer_type_size,
331 .dcompact = rb_io_buffer_type_compact,
334 .flags = RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED | RUBY_TYPED_EMBEDDABLE,
337static inline enum rb_io_buffer_flags
338io_buffer_extract_flags(
VALUE argument)
340 if (rb_int_negative_p(argument)) {
341 rb_raise(rb_eArgError,
"Flags can't be negative!");
344 enum rb_io_buffer_flags flags =
RB_NUM2UINT(argument);
347 return flags & RB_IO_BUFFER_FLAGS_MASK;
352io_buffer_extract_offset(
VALUE argument)
354 if (rb_int_negative_p(argument)) {
355 rb_raise(rb_eArgError,
"Offset can't be negative!");
365io_buffer_extract_length(
VALUE argument)
367 if (rb_int_negative_p(argument)) {
368 rb_raise(rb_eArgError,
"Length can't be negative!");
377io_buffer_extract_size(
VALUE argument)
379 if (rb_int_negative_p(argument)) {
380 rb_raise(rb_eArgError,
"Size can't be negative!");
389io_buffer_extract_width(
VALUE argument,
size_t minimum)
391 if (rb_int_negative_p(argument)) {
392 rb_raise(rb_eArgError,
"Width can't be negative!");
397 if (width < minimum) {
398 rb_raise(rb_eArgError,
"Width must be at least %" PRIuSIZE
"!", minimum);
409io_buffer_default_length(
const struct rb_io_buffer *buffer,
size_t offset)
411 if (offset > buffer->size) {
412 rb_raise(rb_eArgError,
"The given offset is bigger than the buffer size!");
416 return buffer->size - offset;
425io_buffer_extract_length_offset(
VALUE self,
int argc,
VALUE argv[],
size_t *length,
size_t *offset)
430 if (argc >= 2 && !
NIL_P(argv[1])) {
431 *offset = io_buffer_extract_offset(argv[1]);
437 if (argc >= 1 && !
NIL_P(argv[0])) {
438 *length = io_buffer_extract_length(argv[0]);
441 *length = io_buffer_default_length(buffer, *offset);
457io_buffer_extract_offset_length(
VALUE self,
int argc,
VALUE argv[],
size_t *offset,
size_t *length)
462 if (argc >= 1 && !
NIL_P(argv[0])) {
463 *offset = io_buffer_extract_offset(argv[0]);
469 if (argc >= 2 && !
NIL_P(argv[1])) {
470 *length = io_buffer_extract_length(argv[1]);
473 *length = io_buffer_default_length(buffer, *offset);
480rb_io_buffer_type_allocate(
VALUE self)
485 io_buffer_zero(buffer);
490static VALUE io_buffer_for_make_instance(
VALUE klass,
VALUE string,
enum rb_io_buffer_flags flags)
492 VALUE instance = rb_io_buffer_type_allocate(klass);
497 flags |= RB_IO_BUFFER_EXTERNAL;
500 flags |= RB_IO_BUFFER_READONLY;
502 if (!(flags & RB_IO_BUFFER_READONLY))
503 rb_str_modify(
string);
505 io_buffer_initialize(instance, buffer, RSTRING_PTR(
string), RSTRING_LEN(
string), flags,
string);
514 enum rb_io_buffer_flags flags;
518io_buffer_for_yield_instance(
VALUE _arguments)
522 arguments->instance = io_buffer_for_make_instance(arguments->klass, arguments->string, arguments->flags);
528 return rb_yield(arguments->instance);
532io_buffer_for_yield_instance_ensure(
VALUE _arguments)
536 if (arguments->instance !=
Qnil) {
537 rb_io_buffer_free(arguments->instance);
584rb_io_buffer_type_for(
VALUE klass,
VALUE string)
598 return rb_ensure(io_buffer_for_yield_instance, (
VALUE)&arguments, io_buffer_for_yield_instance_ensure, (
VALUE)&arguments);
602 string = rb_str_tmp_frozen_acquire(
string);
603 return io_buffer_for_make_instance(klass,
string, RB_IO_BUFFER_READONLY);
621rb_io_buffer_type_string(
VALUE klass,
VALUE length)
631 rb_ensure(io_buffer_for_yield_instance, (
VALUE)&arguments, io_buffer_for_yield_instance_ensure, (
VALUE)&arguments);
637rb_io_buffer_new(
void *base,
size_t size,
enum rb_io_buffer_flags flags)
639 VALUE instance = rb_io_buffer_type_allocate(rb_cIOBuffer);
644 io_buffer_initialize(instance, buffer, base, size, flags,
Qnil);
650rb_io_buffer_map(
VALUE io,
size_t size, rb_off_t offset,
enum rb_io_buffer_flags flags)
652 io_buffer_experimental();
654 VALUE instance = rb_io_buffer_type_allocate(rb_cIOBuffer);
661 io_buffer_map_file(buffer, descriptor, size, offset, flags);
704io_buffer_map(
int argc,
VALUE *argv,
VALUE klass)
712 if (argc >= 2 && !
RB_NIL_P(argv[1])) {
713 size = io_buffer_extract_size(argv[1]);
716 rb_off_t file_size = rb_file_size(io);
720 rb_raise(rb_eArgError,
"Invalid negative file size!");
723 else if ((uintmax_t)file_size > SIZE_MAX) {
724 rb_raise(rb_eArgError,
"File larger than address space!");
728 size = (size_t)file_size;
738 enum rb_io_buffer_flags flags = 0;
740 flags = io_buffer_extract_flags(argv[3]);
743 return rb_io_buffer_map(io, size, offset, flags);
747static inline enum rb_io_buffer_flags
748io_flags_for_size(
size_t size)
750 if (size >= RUBY_IO_BUFFER_PAGE_SIZE) {
751 return RB_IO_BUFFER_MAPPED;
754 return RB_IO_BUFFER_INTERNAL;
782rb_io_buffer_initialize(
int argc,
VALUE *argv,
VALUE self)
784 io_buffer_experimental();
793 size = io_buffer_extract_size(argv[0]);
796 size = RUBY_IO_BUFFER_DEFAULT_SIZE;
799 enum rb_io_buffer_flags flags = 0;
801 flags = io_buffer_extract_flags(argv[1]);
804 flags |= io_flags_for_size(size);
807 io_buffer_initialize(self, buffer, NULL, size, flags,
Qnil);
813io_buffer_validate_slice(
VALUE source,
void *base,
size_t size)
815 void *source_base = NULL;
816 size_t source_size = 0;
822 rb_io_buffer_get_bytes(source, &source_base, &source_size);
826 if (source_base == NULL)
return 0;
829 if (base < source_base)
return 0;
831 const void *source_end = (
char*)source_base + source_size;
832 const void *end = (
char*)base + size;
835 if (end > source_end)
return 0;
844 if (buffer->source !=
Qnil) {
846 return io_buffer_validate_slice(buffer->source, buffer->base, buffer->size);
853enum rb_io_buffer_flags
854rb_io_buffer_get_bytes(
VALUE self,
void **base,
size_t *size)
859 if (io_buffer_validate(buffer)) {
861 *base = buffer->base;
862 *size = buffer->size;
864 return buffer->flags;
876io_buffer_get_bytes_for_writing(
struct rb_io_buffer *buffer,
void **base,
size_t *size)
878 if (buffer->flags & RB_IO_BUFFER_READONLY ||
880 rb_raise(rb_eIOBufferAccessError,
"Buffer is not writable!");
883 if (!io_buffer_validate(buffer)) {
884 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer is invalid!");
888 *base = buffer->base;
889 *size = buffer->size;
897rb_io_buffer_get_bytes_for_writing(
VALUE self,
void **base,
size_t *size)
902 io_buffer_get_bytes_for_writing(buffer, base, size);
906io_buffer_get_bytes_for_reading(
struct rb_io_buffer *buffer,
const void **base,
size_t *size)
908 if (!io_buffer_validate(buffer)) {
909 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer has been invalidated!");
913 *base = buffer->base;
914 *size = buffer->size;
922rb_io_buffer_get_bytes_for_reading(
VALUE self,
const void **base,
size_t *size)
927 io_buffer_get_bytes_for_reading(buffer, base, size);
940rb_io_buffer_to_s(
VALUE self)
948 rb_str_catf(result,
" %p+%"PRIdSIZE, buffer->base, buffer->size);
950 if (buffer->base == NULL) {
954 if (buffer->flags & RB_IO_BUFFER_EXTERNAL) {
958 if (buffer->flags & RB_IO_BUFFER_INTERNAL) {
962 if (buffer->flags & RB_IO_BUFFER_MAPPED) {
966 if (buffer->flags & RB_IO_BUFFER_FILE) {
970 if (buffer->flags & RB_IO_BUFFER_SHARED) {
974 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
978 if (buffer->flags & RB_IO_BUFFER_PRIVATE) {
982 if (buffer->flags & RB_IO_BUFFER_READONLY) {
986 if (buffer->source !=
Qnil) {
990 if (!io_buffer_validate(buffer)) {
1000io_buffer_hexdump_output_size(
size_t width,
size_t size,
int first)
1003 size_t total = size;
1005 size_t whole_lines = (size / width);
1006 size_t partial_line = (size % width) ? 1 : 0;
1011 total += (whole_lines + partial_line) * (1 + 10 + width*3 + 1 + 1);
1014 if (size && first) total -= 1;
1023io_buffer_hexdump(
VALUE string,
size_t width,
const char *base,
size_t length,
size_t offset,
int first)
1025 char *text = alloca(width+1);
1028 for (; offset < length; offset += width) {
1029 memset(text,
'\0', width);
1031 rb_str_catf(
string,
"0x%08" PRIxSIZE
" ", offset);
1035 rb_str_catf(
string,
"\n0x%08" PRIxSIZE
" ", offset);
1038 for (
size_t i = 0; i < width; i += 1) {
1039 if (offset+i < length) {
1040 unsigned char value = ((
unsigned char*)base)[offset+i];
1042 if (value < 127 && isprint(value)) {
1043 text[i] = (char)value;
1049 rb_str_catf(
string,
" %02x", value);
1056 rb_str_catf(
string,
" %s", text);
1075rb_io_buffer_inspect(
VALUE self)
1080 VALUE result = rb_io_buffer_to_s(self);
1082 if (io_buffer_validate(buffer)) {
1084 size_t size = buffer->size;
1087 if (size > RB_IO_BUFFER_INSPECT_HEXDUMP_MAXIMUM_SIZE) {
1088 size = RB_IO_BUFFER_INSPECT_HEXDUMP_MAXIMUM_SIZE;
1092 io_buffer_hexdump(result, RB_IO_BUFFER_INSPECT_HEXDUMP_WIDTH, buffer->base, size, 0, 0);
1095 rb_str_catf(result,
"\n(and %" PRIuSIZE
" more bytes not printed)", buffer->size - size);
1109rb_io_buffer_size(
VALUE self)
1126rb_io_buffer_valid_p(
VALUE self)
1131 return RBOOL(io_buffer_validate(buffer));
1149rb_io_buffer_null_p(
VALUE self)
1154 return RBOOL(buffer->base == NULL);
1165rb_io_buffer_empty_p(
VALUE self)
1170 return RBOOL(buffer->size == 0);
1185rb_io_buffer_external_p(
VALUE self)
1190 return RBOOL(buffer->flags & RB_IO_BUFFER_EXTERNAL);
1210rb_io_buffer_internal_p(
VALUE self)
1215 return RBOOL(buffer->flags & RB_IO_BUFFER_INTERNAL);
1232rb_io_buffer_mapped_p(
VALUE self)
1237 return RBOOL(buffer->flags & RB_IO_BUFFER_MAPPED);
1264rb_io_buffer_shared_p(
VALUE self)
1269 return RBOOL(buffer->flags & RB_IO_BUFFER_SHARED);
1287rb_io_buffer_locked_p(
VALUE self)
1292 return RBOOL(buffer->flags & RB_IO_BUFFER_LOCKED);
1318rb_io_buffer_private_p(
VALUE self)
1323 return RBOOL(buffer->flags & RB_IO_BUFFER_PRIVATE);
1327rb_io_buffer_readonly_p(
VALUE self)
1332 return buffer->flags & RB_IO_BUFFER_READONLY;
1344io_buffer_readonly_p(
VALUE self)
1346 return RBOOL(rb_io_buffer_readonly_p(self));
1352 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
1353 rb_raise(rb_eIOBufferLockedError,
"Buffer already locked!");
1356 buffer->flags |= RB_IO_BUFFER_LOCKED;
1360rb_io_buffer_lock(
VALUE self)
1365 io_buffer_lock(buffer);
1373 if (!(buffer->flags & RB_IO_BUFFER_LOCKED)) {
1374 rb_raise(rb_eIOBufferLockedError,
"Buffer not locked!");
1377 buffer->flags &= ~RB_IO_BUFFER_LOCKED;
1381rb_io_buffer_unlock(
VALUE self)
1386 io_buffer_unlock(buffer);
1392rb_io_buffer_try_unlock(
VALUE self)
1397 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
1398 buffer->flags &= ~RB_IO_BUFFER_LOCKED;
1436rb_io_buffer_locked(
VALUE self)
1441 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
1442 rb_raise(rb_eIOBufferLockedError,
"Buffer already locked!");
1445 buffer->flags |= RB_IO_BUFFER_LOCKED;
1449 buffer->flags &= ~RB_IO_BUFFER_LOCKED;
1480rb_io_buffer_free(
VALUE self)
1485 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
1486 rb_raise(rb_eIOBufferLockedError,
"Buffer is locked!");
1489 io_buffer_free(buffer);
1499 io_buffer_unlock(buffer);
1500 io_buffer_free(buffer);
1508io_buffer_validate_range(
struct rb_io_buffer *buffer,
size_t offset,
size_t length)
1511 if (offset + length > buffer->size) {
1512 rb_raise(rb_eArgError,
"Specified offset+length is bigger than the buffer size!");
1533rb_io_buffer_hexdump(
int argc,
VALUE *argv,
VALUE self)
1537 size_t offset, length;
1538 struct rb_io_buffer *buffer = io_buffer_extract_offset_length(self, argc, argv, &offset, &length);
1540 size_t width = RB_IO_BUFFER_HEXDUMP_DEFAULT_WIDTH;
1542 width = io_buffer_extract_width(argv[2], 1);
1546 io_buffer_validate_range(buffer, offset, length);
1550 if (io_buffer_validate(buffer) && buffer->base) {
1551 result =
rb_str_buf_new(io_buffer_hexdump_output_size(width, length, 1));
1553 io_buffer_hexdump(result, width, buffer->base, offset+length, offset, 1);
1560rb_io_buffer_slice(
struct rb_io_buffer *buffer,
VALUE self,
size_t offset,
size_t length)
1562 io_buffer_validate_range(buffer, offset, length);
1568 slice->flags |= (buffer->flags & RB_IO_BUFFER_READONLY);
1569 slice->base = (
char*)buffer->base + offset;
1570 slice->size = length;
1573 if (buffer->source !=
Qnil) {
1574 RB_OBJ_WRITE(instance, &slice->source, buffer->source);
1634io_buffer_slice(
int argc,
VALUE *argv,
VALUE self)
1638 size_t offset, length;
1639 struct rb_io_buffer *buffer = io_buffer_extract_offset_length(self, argc, argv, &offset, &length);
1641 return rb_io_buffer_slice(buffer, self, offset, length);
1663rb_io_buffer_transfer(
VALUE self)
1668 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
1669 rb_raise(rb_eIOBufferLockedError,
"Cannot transfer ownership of locked buffer!");
1676 *transferred = *buffer;
1677 io_buffer_zero(buffer);
1683io_buffer_resize_clear(
struct rb_io_buffer *buffer,
void* base,
size_t size)
1685 if (size > buffer->size) {
1686 memset((
unsigned char*)base+buffer->size, 0, size - buffer->size);
1695 io_buffer_initialize(self, &resized, NULL, size, io_flags_for_size(size),
Qnil);
1698 size_t preserve = buffer->size;
1699 if (preserve > size) preserve = size;
1700 memcpy(resized.base, buffer->base, preserve);
1702 io_buffer_resize_clear(buffer, resized.base, size);
1705 io_buffer_free(buffer);
1710rb_io_buffer_resize(
VALUE self,
size_t size)
1715 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
1716 rb_raise(rb_eIOBufferLockedError,
"Cannot resize locked buffer!");
1719 if (buffer->base == NULL) {
1720 io_buffer_initialize(self, buffer, NULL, size, io_flags_for_size(size),
Qnil);
1724 if (buffer->flags & RB_IO_BUFFER_EXTERNAL) {
1725 rb_raise(rb_eIOBufferAccessError,
"Cannot resize external buffer!");
1728#if defined(HAVE_MREMAP) && defined(MREMAP_MAYMOVE)
1729 if (buffer->flags & RB_IO_BUFFER_MAPPED) {
1730 void *base = mremap(buffer->base, buffer->size, size, MREMAP_MAYMOVE);
1732 if (base == MAP_FAILED) {
1733 rb_sys_fail(
"rb_io_buffer_resize:mremap");
1736 io_buffer_resize_clear(buffer, base, size);
1738 buffer->base = base;
1739 buffer->size = size;
1745 if (buffer->flags & RB_IO_BUFFER_INTERNAL) {
1747 io_buffer_free(buffer);
1751 void *base = realloc(buffer->base, size);
1754 rb_sys_fail(
"rb_io_buffer_resize:realloc");
1757 io_buffer_resize_clear(buffer, base, size);
1759 buffer->base = base;
1760 buffer->size = size;
1765 io_buffer_resize_copy(self, buffer, size);
1789 rb_io_buffer_resize(self, io_buffer_extract_size(size));
1803 const void *ptr1, *ptr2;
1804 size_t size1, size2;
1806 rb_io_buffer_get_bytes_for_reading(self, &ptr1, &size1);
1807 rb_io_buffer_get_bytes_for_reading(other, &ptr2, &size2);
1809 if (size1 < size2) {
1813 if (size1 > size2) {
1817 return RB_INT2NUM(memcmp(ptr1, ptr2, size1));
1821io_buffer_validate_type(
size_t size,
size_t offset)
1823 if (offset > size) {
1824 rb_raise(rb_eArgError,
"Type extends beyond end of buffer! (offset=%"PRIdSIZE
" > size=%"PRIdSIZE
")", offset, size);
1846#define ruby_swap8(value) value
1854ruby_swapf32(
float value)
1856 union swapf32 swap = {.value = value};
1857 swap.integral = ruby_swap32(swap.integral);
1867ruby_swapf64(
double value)
1869 union swapf64 swap = {.value = value};
1870 swap.integral = ruby_swap64(swap.integral);
1874#define IO_BUFFER_DECLARE_TYPE(name, type, endian, wrap, unwrap, swap) \
1875static ID RB_IO_BUFFER_DATA_TYPE_##name; \
1878io_buffer_read_##name(const void* base, size_t size, size_t *offset) \
1880 io_buffer_validate_type(size, *offset + sizeof(type)); \
1882 memcpy(&value, (char*)base + *offset, sizeof(type)); \
1883 if (endian != RB_IO_BUFFER_HOST_ENDIAN) value = swap(value); \
1884 *offset += sizeof(type); \
1885 return wrap(value); \
1889io_buffer_write_##name(const void* base, size_t size, size_t *offset, VALUE _value) \
1891 io_buffer_validate_type(size, *offset + sizeof(type)); \
1892 type value = unwrap(_value); \
1893 if (endian != RB_IO_BUFFER_HOST_ENDIAN) value = swap(value); \
1894 memcpy((char*)base + *offset, &value, sizeof(type)); \
1895 *offset += sizeof(type); \
1899 RB_IO_BUFFER_DATA_TYPE_##name##_SIZE = sizeof(type) \
1907IO_BUFFER_DECLARE_TYPE(s16, int16_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap16)
1908IO_BUFFER_DECLARE_TYPE(S16, int16_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap16)
1912IO_BUFFER_DECLARE_TYPE(s32, int32_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap32)
1913IO_BUFFER_DECLARE_TYPE(S32, int32_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap32)
1915IO_BUFFER_DECLARE_TYPE(u64, uint64_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_ULL2NUM,
RB_NUM2ULL, ruby_swap64)
1916IO_BUFFER_DECLARE_TYPE(U64, uint64_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_ULL2NUM,
RB_NUM2ULL, ruby_swap64)
1917IO_BUFFER_DECLARE_TYPE(s64, int64_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_LL2NUM,
RB_NUM2LL, ruby_swap64)
1918IO_BUFFER_DECLARE_TYPE(S64, int64_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_LL2NUM,
RB_NUM2LL, ruby_swap64)
1920IO_BUFFER_DECLARE_TYPE(f32,
float, RB_IO_BUFFER_LITTLE_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf32)
1921IO_BUFFER_DECLARE_TYPE(F32,
float, RB_IO_BUFFER_BIG_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf32)
1922IO_BUFFER_DECLARE_TYPE(f64,
double, RB_IO_BUFFER_LITTLE_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf64)
1923IO_BUFFER_DECLARE_TYPE(F64,
double, RB_IO_BUFFER_BIG_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf64)
1924#undef IO_BUFFER_DECLARE_TYPE
1927io_buffer_buffer_type_size(
ID buffer_type)
1929#define IO_BUFFER_DATA_TYPE_SIZE(name) if (buffer_type == RB_IO_BUFFER_DATA_TYPE_##name) return RB_IO_BUFFER_DATA_TYPE_##name##_SIZE;
1930 IO_BUFFER_DATA_TYPE_SIZE(U8)
1931 IO_BUFFER_DATA_TYPE_SIZE(S8)
1932 IO_BUFFER_DATA_TYPE_SIZE(u16)
1933 IO_BUFFER_DATA_TYPE_SIZE(U16)
1934 IO_BUFFER_DATA_TYPE_SIZE(s16)
1935 IO_BUFFER_DATA_TYPE_SIZE(S16)
1936 IO_BUFFER_DATA_TYPE_SIZE(u32)
1937 IO_BUFFER_DATA_TYPE_SIZE(U32)
1938 IO_BUFFER_DATA_TYPE_SIZE(s32)
1939 IO_BUFFER_DATA_TYPE_SIZE(S32)
1940 IO_BUFFER_DATA_TYPE_SIZE(u64)
1941 IO_BUFFER_DATA_TYPE_SIZE(U64)
1942 IO_BUFFER_DATA_TYPE_SIZE(s64)
1943 IO_BUFFER_DATA_TYPE_SIZE(S64)
1944 IO_BUFFER_DATA_TYPE_SIZE(f32)
1945 IO_BUFFER_DATA_TYPE_SIZE(F32)
1946 IO_BUFFER_DATA_TYPE_SIZE(f64)
1947 IO_BUFFER_DATA_TYPE_SIZE(F64)
1948#undef IO_BUFFER_DATA_TYPE_SIZE
1950 rb_raise(rb_eArgError,
"Invalid type name!");
1964io_buffer_size_of(
VALUE klass,
VALUE buffer_type)
1968 for (
long i = 0; i <
RARRAY_LEN(buffer_type); i++) {
1979rb_io_buffer_get_value(
const void* base,
size_t size,
ID buffer_type,
size_t *offset)
1981#define IO_BUFFER_GET_VALUE(name) if (buffer_type == RB_IO_BUFFER_DATA_TYPE_##name) return io_buffer_read_##name(base, size, offset);
1982 IO_BUFFER_GET_VALUE(U8)
1983 IO_BUFFER_GET_VALUE(S8)
1985 IO_BUFFER_GET_VALUE(u16)
1986 IO_BUFFER_GET_VALUE(U16)
1987 IO_BUFFER_GET_VALUE(s16)
1988 IO_BUFFER_GET_VALUE(S16)
1990 IO_BUFFER_GET_VALUE(u32)
1991 IO_BUFFER_GET_VALUE(U32)
1992 IO_BUFFER_GET_VALUE(s32)
1993 IO_BUFFER_GET_VALUE(S32)
1995 IO_BUFFER_GET_VALUE(u64)
1996 IO_BUFFER_GET_VALUE(U64)
1997 IO_BUFFER_GET_VALUE(s64)
1998 IO_BUFFER_GET_VALUE(S64)
2000 IO_BUFFER_GET_VALUE(f32)
2001 IO_BUFFER_GET_VALUE(F32)
2002 IO_BUFFER_GET_VALUE(f64)
2003 IO_BUFFER_GET_VALUE(F64)
2004#undef IO_BUFFER_GET_VALUE
2006 rb_raise(rb_eArgError,
"Invalid type name!");
2048 size_t offset = io_buffer_extract_offset(_offset);
2050 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2052 return rb_io_buffer_get_value(base, size,
RB_SYM2ID(
type), &offset);
2068 size_t offset = io_buffer_extract_offset(_offset);
2072 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2075 rb_raise(rb_eArgError,
"Argument buffer_types should be an array!");
2080 for (
long i = 0; i <
RARRAY_LEN(buffer_types); i++) {
2092io_buffer_extract_count(
VALUE argument)
2094 if (rb_int_negative_p(argument)) {
2095 rb_raise(rb_eArgError,
"Count can't be negative!");
2102io_buffer_extract_offset_count(
ID buffer_type,
size_t size,
int argc,
VALUE *argv,
size_t *offset,
size_t *count)
2105 *offset = io_buffer_extract_offset(argv[0]);
2112 *count = io_buffer_extract_count(argv[1]);
2115 if (*offset > size) {
2116 rb_raise(rb_eArgError,
"The given offset is bigger than the buffer size!");
2119 *count = (size - *offset) / io_buffer_buffer_type_size(buffer_type);
2140io_buffer_each(
int argc,
VALUE *argv,
VALUE self)
2147 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2154 buffer_type = RB_IO_BUFFER_DATA_TYPE_U8;
2157 size_t offset, count;
2158 io_buffer_extract_offset_count(buffer_type, size, argc-1, argv+1, &offset, &count);
2160 for (
size_t i = 0; i < count; i++) {
2161 size_t current_offset = offset;
2162 VALUE value = rb_io_buffer_get_value(base, size, buffer_type, &offset);
2180io_buffer_values(
int argc,
VALUE *argv,
VALUE self)
2185 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2192 buffer_type = RB_IO_BUFFER_DATA_TYPE_U8;
2195 size_t offset, count;
2196 io_buffer_extract_offset_count(buffer_type, size, argc-1, argv+1, &offset, &count);
2200 for (
size_t i = 0; i < count; i++) {
2201 VALUE value = rb_io_buffer_get_value(base, size, buffer_type, &offset);
2224io_buffer_each_byte(
int argc,
VALUE *argv,
VALUE self)
2231 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2233 size_t offset, count;
2234 io_buffer_extract_offset_count(RB_IO_BUFFER_DATA_TYPE_U8, size, argc-1, argv+1, &offset, &count);
2236 for (
size_t i = 0; i < count; i++) {
2237 unsigned char *value = (
unsigned char *)base + i + offset;
2245rb_io_buffer_set_value(
const void* base,
size_t size,
ID buffer_type,
size_t *offset,
VALUE value)
2247#define IO_BUFFER_SET_VALUE(name) if (buffer_type == RB_IO_BUFFER_DATA_TYPE_##name) {io_buffer_write_##name(base, size, offset, value); return;}
2248 IO_BUFFER_SET_VALUE(U8);
2249 IO_BUFFER_SET_VALUE(S8);
2251 IO_BUFFER_SET_VALUE(u16);
2252 IO_BUFFER_SET_VALUE(U16);
2253 IO_BUFFER_SET_VALUE(s16);
2254 IO_BUFFER_SET_VALUE(S16);
2256 IO_BUFFER_SET_VALUE(u32);
2257 IO_BUFFER_SET_VALUE(U32);
2258 IO_BUFFER_SET_VALUE(s32);
2259 IO_BUFFER_SET_VALUE(S32);
2261 IO_BUFFER_SET_VALUE(u64);
2262 IO_BUFFER_SET_VALUE(U64);
2263 IO_BUFFER_SET_VALUE(s64);
2264 IO_BUFFER_SET_VALUE(S64);
2266 IO_BUFFER_SET_VALUE(f32);
2267 IO_BUFFER_SET_VALUE(F32);
2268 IO_BUFFER_SET_VALUE(f64);
2269 IO_BUFFER_SET_VALUE(F64);
2270#undef IO_BUFFER_SET_VALUE
2272 rb_raise(rb_eArgError,
"Invalid type name!");
2310 size_t offset = io_buffer_extract_offset(_offset);
2312 rb_io_buffer_get_bytes_for_writing(self, &base, &size);
2314 rb_io_buffer_set_value(base, size,
RB_SYM2ID(
type), &offset, value);
2337 rb_raise(rb_eArgError,
"Argument buffer_types should be an array!");
2341 rb_raise(rb_eArgError,
"Argument values should be an array!");
2345 rb_raise(rb_eArgError,
"Argument buffer_types and values should have the same length!");
2348 size_t offset = io_buffer_extract_offset(_offset);
2352 rb_io_buffer_get_bytes_for_writing(self, &base, &size);
2354 for (
long i = 0; i <
RARRAY_LEN(buffer_types); i++) {
2357 rb_io_buffer_set_value(base, size,
RB_SYM2ID(
type), &offset, value);
2363static size_t IO_BUFFER_BLOCKING_SIZE = 1024*1024;
2366 unsigned char * destination;
2367 const unsigned char * source;
2372io_buffer_memmove_blocking(
void *data)
2376 memmove(arguments->destination, arguments->source, arguments->length);
2382io_buffer_memmove_unblock(
void *data)
2388io_buffer_memmove(
struct rb_io_buffer *buffer,
size_t offset,
const void *source_base,
size_t source_offset,
size_t source_size,
size_t length)
2392 io_buffer_get_bytes_for_writing(buffer, &base, &size);
2394 io_buffer_validate_range(buffer, offset, length);
2396 if (source_offset + length > source_size) {
2397 rb_raise(rb_eArgError,
"The computed source range exceeds the size of the source buffer!");
2401 .destination = (
unsigned char*)base+offset,
2402 .source = (
unsigned char*)source_base+source_offset,
2406 if (arguments.length >= IO_BUFFER_BLOCKING_SIZE) {
2408 }
else if (arguments.length != 0) {
2409 memmove(arguments.destination, arguments.source, arguments.length);
2415io_buffer_copy_from(
struct rb_io_buffer *buffer,
const void *source_base,
size_t source_size,
int argc,
VALUE *argv)
2419 size_t source_offset;
2423 offset = io_buffer_extract_offset(argv[0]);
2428 source_offset = io_buffer_extract_offset(argv[2]);
2430 if (source_offset > source_size) {
2431 rb_raise(rb_eArgError,
"The given source offset is bigger than the source itself!");
2439 if (argc >= 2 && !
RB_NIL_P(argv[1])) {
2440 length = io_buffer_extract_length(argv[1]);
2444 length = source_size - source_offset;
2447 io_buffer_memmove(buffer, offset, source_base, source_offset, source_size, length);
2470rb_io_buffer_initialize_copy(
VALUE self,
VALUE source)
2475 const void *source_base;
2478 rb_io_buffer_get_bytes_for_reading(source, &source_base, &source_size);
2480 io_buffer_initialize(self, buffer, NULL, source_size, io_flags_for_size(source_size),
Qnil);
2482 return io_buffer_copy_from(buffer, source_base, source_size, 0, NULL);
2554io_buffer_copy(
int argc,
VALUE *argv,
VALUE self)
2561 VALUE source = argv[0];
2562 const void *source_base;
2565 rb_io_buffer_get_bytes_for_reading(source, &source_base, &source_size);
2567 return io_buffer_copy_from(buffer, source_base, source_size, argc-1, argv+1);
2585io_buffer_get_string(
int argc,
VALUE *argv,
VALUE self)
2589 size_t offset, length;
2590 struct rb_io_buffer *buffer = io_buffer_extract_offset_length(self, argc, argv, &offset, &length);
2594 io_buffer_get_bytes_for_reading(buffer, &base, &size);
2598 encoding = rb_find_encoding(argv[2]);
2601 encoding = rb_ascii8bit_encoding();
2604 io_buffer_validate_range(buffer, offset, length);
2606 return rb_enc_str_new((
const char*)base + offset, length, encoding);
2633io_buffer_set_string(
int argc,
VALUE *argv,
VALUE self)
2642 const void *source_base = RSTRING_PTR(
string);
2643 size_t source_size = RSTRING_LEN(
string);
2645 return io_buffer_copy_from(buffer, source_base, source_size, argc-1, argv+1);
2649rb_io_buffer_clear(
VALUE self, uint8_t value,
size_t offset,
size_t length)
2656 io_buffer_get_bytes_for_writing(buffer, &base, &size);
2658 io_buffer_validate_range(buffer, offset, length);
2660 memset((
char*)base + offset, value, length);
2695io_buffer_clear(
int argc,
VALUE *argv,
VALUE self)
2704 size_t offset, length;
2705 io_buffer_extract_offset_length(self, argc-1, argv+1, &offset, &length);
2707 rb_io_buffer_clear(self, value, offset, length);
2713io_buffer_default_size(
size_t page_size)
2716 const size_t platform_agnostic_default_size = 64*1024;
2719 const char *default_size = getenv(
"RUBY_IO_BUFFER_DEFAULT_SIZE");
2722 int value = atoi(default_size);
2730 if (platform_agnostic_default_size < page_size) {
2734 return platform_agnostic_default_size;
2740 rb_blocking_function_t *function;
2745io_buffer_blocking_region_begin(
VALUE _argument)
2749 return rb_io_blocking_region(argument->io, argument->function, argument->data);
2753io_buffer_blocking_region_ensure(
VALUE _argument)
2757 io_buffer_unlock(argument->buffer);
2763io_buffer_blocking_region(
VALUE io,
struct rb_io_buffer *buffer, rb_blocking_function_t *function,
void *data)
2765 struct rb_io *ioptr;
2771 .function = function,
2776 if (buffer->flags & RB_IO_BUFFER_LOCKED) {
2777 return io_buffer_blocking_region_begin((
VALUE)&argument);
2781 io_buffer_lock(buffer);
2783 return rb_ensure(io_buffer_blocking_region_begin, (
VALUE)&argument, io_buffer_blocking_region_ensure, (
VALUE)&argument);
2799io_buffer_read_internal(
void *_argument)
2805 ssize_t result = read(argument->descriptor, argument->base, argument->size);
2810 else if (result == 0) {
2816 if (total >= argument->length) {
2820 argument->base = argument->base + result;
2821 argument->size = argument->size - result;
2827rb_io_buffer_read(
VALUE self,
VALUE io,
size_t length,
size_t offset)
2832 if (scheduler !=
Qnil) {
2835 if (!UNDEF_P(result)) {
2843 io_buffer_validate_range(buffer, offset, length);
2849 io_buffer_get_bytes_for_writing(buffer, &base, &size);
2851 base = (
unsigned char*)base + offset;
2852 size = size - offset;
2855 .descriptor = descriptor,
2861 return io_buffer_blocking_region(io, buffer, io_buffer_read_internal, &argument);
2891io_buffer_read(
int argc,
VALUE *argv,
VALUE self)
2897 size_t length, offset;
2898 io_buffer_extract_length_offset(self, argc-1, argv+1, &length, &offset);
2900 return rb_io_buffer_read(self, io, length, offset);
2917io_buffer_pread_internal(
void *_argument)
2923 ssize_t result = pread(argument->descriptor, argument->base, argument->size, argument->offset);
2928 else if (result == 0) {
2934 if (total >= argument->length) {
2938 argument->base = argument->base + result;
2939 argument->size = argument->size - result;
2940 argument->offset = argument->offset + result;
2946rb_io_buffer_pread(
VALUE self,
VALUE io, rb_off_t from,
size_t length,
size_t offset)
2951 if (scheduler !=
Qnil) {
2954 if (!UNDEF_P(result)) {
2962 io_buffer_validate_range(buffer, offset, length);
2968 io_buffer_get_bytes_for_writing(buffer, &base, &size);
2970 base = (
unsigned char*)base + offset;
2971 size = size - offset;
2974 .descriptor = descriptor,
2981 return io_buffer_blocking_region(io, buffer, io_buffer_pread_internal, &argument);
3015io_buffer_pread(
int argc,
VALUE *argv,
VALUE self)
3022 size_t length, offset;
3023 io_buffer_extract_length_offset(self, argc-2, argv+2, &length, &offset);
3025 return rb_io_buffer_pread(self, io, from, length, offset);
3040io_buffer_write_internal(
void *_argument)
3046 ssize_t result = write(argument->descriptor, argument->base, argument->size);
3051 else if (result == 0) {
3057 if (total >= argument->length) {
3061 argument->base = argument->base + result;
3062 argument->size = argument->size - result;
3068rb_io_buffer_write(
VALUE self,
VALUE io,
size_t length,
size_t offset)
3073 if (scheduler !=
Qnil) {
3076 if (!UNDEF_P(result)) {
3084 io_buffer_validate_range(buffer, offset, length);
3090 io_buffer_get_bytes_for_reading(buffer, &base, &size);
3092 base = (
unsigned char*)base + offset;
3093 size = size - offset;
3096 .descriptor = descriptor,
3102 return io_buffer_blocking_region(io, buffer, io_buffer_write_internal, &argument);
3125io_buffer_write(
int argc,
VALUE *argv,
VALUE self)
3131 size_t length, offset;
3132 io_buffer_extract_length_offset(self, argc-1, argv+1, &length, &offset);
3134 return rb_io_buffer_write(self, io, length, offset);
3151io_buffer_pwrite_internal(
void *_argument)
3157 ssize_t result = pwrite(argument->descriptor, argument->base, argument->size, argument->offset);
3162 else if (result == 0) {
3168 if (total >= argument->length) {
3172 argument->base = argument->base + result;
3173 argument->size = argument->size - result;
3174 argument->offset = argument->offset + result;
3180rb_io_buffer_pwrite(
VALUE self,
VALUE io, rb_off_t from,
size_t length,
size_t offset)
3185 if (scheduler !=
Qnil) {
3188 if (!UNDEF_P(result)) {
3196 io_buffer_validate_range(buffer, offset, length);
3202 io_buffer_get_bytes_for_reading(buffer, &base, &size);
3204 base = (
unsigned char*)base + offset;
3205 size = size - offset;
3208 .descriptor = descriptor,
3223 return io_buffer_blocking_region(io, buffer, io_buffer_pwrite_internal, &argument);
3251io_buffer_pwrite(
int argc,
VALUE *argv,
VALUE self)
3258 size_t length, offset;
3259 io_buffer_extract_length_offset(self, argc-2, argv+2, &length, &offset);
3261 return rb_io_buffer_pwrite(self, io, from, length, offset);
3267 if (buffer->size == 0)
3268 rb_raise(rb_eIOBufferMaskError,
"Zero-length mask given!");
3272memory_and(
unsigned char * restrict output,
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
3274 for (
size_t offset = 0; offset < size; offset += 1) {
3275 output[offset] = base[offset] & mask[offset % mask_size];
3300 io_buffer_check_mask(mask_buffer);
3302 VALUE output = rb_io_buffer_new(NULL, buffer->size, io_flags_for_size(buffer->size));
3306 memory_and(output_buffer->base, buffer->base, buffer->size, mask_buffer->base, mask_buffer->size);
3312memory_or(
unsigned char * restrict output,
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
3314 for (
size_t offset = 0; offset < size; offset += 1) {
3315 output[offset] = base[offset] | mask[offset % mask_size];
3340 io_buffer_check_mask(mask_buffer);
3342 VALUE output = rb_io_buffer_new(NULL, buffer->size, io_flags_for_size(buffer->size));
3346 memory_or(output_buffer->base, buffer->base, buffer->size, mask_buffer->base, mask_buffer->size);
3352memory_xor(
unsigned char * restrict output,
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
3354 for (
size_t offset = 0; offset < size; offset += 1) {
3355 output[offset] = base[offset] ^ mask[offset % mask_size];
3380 io_buffer_check_mask(mask_buffer);
3382 VALUE output = rb_io_buffer_new(NULL, buffer->size, io_flags_for_size(buffer->size));
3386 memory_xor(output_buffer->base, buffer->base, buffer->size, mask_buffer->base, mask_buffer->size);
3392memory_not(
unsigned char * restrict output,
unsigned char * restrict base,
size_t size)
3394 for (
size_t offset = 0; offset < size; offset += 1) {
3395 output[offset] = ~base[offset];
3412io_buffer_not(
VALUE self)
3417 VALUE output = rb_io_buffer_new(NULL, buffer->size, io_flags_for_size(buffer->size));
3421 memory_not(output_buffer->base, buffer->base, buffer->size);
3429 if (a->base > b->base) {
3430 return io_buffer_overlaps(b, a);
3433 return (b->base >= a->base) && (b->base < (
void*)((
unsigned char *)a->base + a->size));
3439 if (io_buffer_overlaps(a, b))
3440 rb_raise(rb_eIOBufferMaskError,
"Mask overlaps source buffer!");
3444memory_and_inplace(
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
3446 for (
size_t offset = 0; offset < size; offset += 1) {
3447 base[offset] &= mask[offset % mask_size];
3477 io_buffer_check_mask(mask_buffer);
3478 io_buffer_check_overlaps(buffer, mask_buffer);
3482 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3484 memory_and_inplace(base, size, mask_buffer->base, mask_buffer->size);
3490memory_or_inplace(
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
3492 for (
size_t offset = 0; offset < size; offset += 1) {
3493 base[offset] |= mask[offset % mask_size];
3523 io_buffer_check_mask(mask_buffer);
3524 io_buffer_check_overlaps(buffer, mask_buffer);
3528 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3530 memory_or_inplace(base, size, mask_buffer->base, mask_buffer->size);
3536memory_xor_inplace(
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
3538 for (
size_t offset = 0; offset < size; offset += 1) {
3539 base[offset] ^= mask[offset % mask_size];
3569 io_buffer_check_mask(mask_buffer);
3570 io_buffer_check_overlaps(buffer, mask_buffer);
3574 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3576 memory_xor_inplace(base, size, mask_buffer->base, mask_buffer->size);
3582memory_not_inplace(
unsigned char * restrict base,
size_t size)
3584 for (
size_t offset = 0; offset < size; offset += 1) {
3585 base[offset] = ~base[offset];
3607io_buffer_not_inplace(
VALUE self)
3614 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3616 memory_not_inplace(base, size);
3734 GetSystemInfo(&info);
3735 RUBY_IO_BUFFER_PAGE_SIZE = info.dwPageSize;
3737 RUBY_IO_BUFFER_PAGE_SIZE = sysconf(_SC_PAGESIZE);
3740 RUBY_IO_BUFFER_DEFAULT_SIZE = io_buffer_default_size(RUBY_IO_BUFFER_PAGE_SIZE);
3743 rb_define_const(rb_cIOBuffer,
"PAGE_SIZE",
SIZET2NUM(RUBY_IO_BUFFER_PAGE_SIZE));
3748 rb_define_const(rb_cIOBuffer,
"DEFAULT_SIZE",
SIZET2NUM(RUBY_IO_BUFFER_DEFAULT_SIZE));
3753 rb_define_method(rb_cIOBuffer,
"initialize_copy", rb_io_buffer_initialize_copy, 1);
3763 rb_define_const(rb_cIOBuffer,
"EXTERNAL",
RB_INT2NUM(RB_IO_BUFFER_EXTERNAL));
3766 rb_define_const(rb_cIOBuffer,
"INTERNAL",
RB_INT2NUM(RB_IO_BUFFER_INTERNAL));
3769 rb_define_const(rb_cIOBuffer,
"MAPPED",
RB_INT2NUM(RB_IO_BUFFER_MAPPED));
3772 rb_define_const(rb_cIOBuffer,
"SHARED",
RB_INT2NUM(RB_IO_BUFFER_SHARED));
3775 rb_define_const(rb_cIOBuffer,
"LOCKED",
RB_INT2NUM(RB_IO_BUFFER_LOCKED));
3778 rb_define_const(rb_cIOBuffer,
"PRIVATE",
RB_INT2NUM(RB_IO_BUFFER_PRIVATE));
3781 rb_define_const(rb_cIOBuffer,
"READONLY",
RB_INT2NUM(RB_IO_BUFFER_READONLY));
3784 rb_define_const(rb_cIOBuffer,
"LITTLE_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_LITTLE_ENDIAN));
3787 rb_define_const(rb_cIOBuffer,
"BIG_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_BIG_ENDIAN));
3790 rb_define_const(rb_cIOBuffer,
"HOST_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_HOST_ENDIAN));
3793 rb_define_const(rb_cIOBuffer,
"NETWORK_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_NETWORK_ENDIAN));
3819#define IO_BUFFER_DEFINE_DATA_TYPE(name) RB_IO_BUFFER_DATA_TYPE_##name = rb_intern_const(#name)
3820 IO_BUFFER_DEFINE_DATA_TYPE(U8);
3821 IO_BUFFER_DEFINE_DATA_TYPE(S8);
3823 IO_BUFFER_DEFINE_DATA_TYPE(u16);
3824 IO_BUFFER_DEFINE_DATA_TYPE(U16);
3825 IO_BUFFER_DEFINE_DATA_TYPE(s16);
3826 IO_BUFFER_DEFINE_DATA_TYPE(S16);
3828 IO_BUFFER_DEFINE_DATA_TYPE(u32);
3829 IO_BUFFER_DEFINE_DATA_TYPE(U32);
3830 IO_BUFFER_DEFINE_DATA_TYPE(s32);
3831 IO_BUFFER_DEFINE_DATA_TYPE(S32);
3833 IO_BUFFER_DEFINE_DATA_TYPE(u64);
3834 IO_BUFFER_DEFINE_DATA_TYPE(U64);
3835 IO_BUFFER_DEFINE_DATA_TYPE(s64);
3836 IO_BUFFER_DEFINE_DATA_TYPE(S64);
3838 IO_BUFFER_DEFINE_DATA_TYPE(f32);
3839 IO_BUFFER_DEFINE_DATA_TYPE(F32);
3840 IO_BUFFER_DEFINE_DATA_TYPE(f64);
3841 IO_BUFFER_DEFINE_DATA_TYPE(F64);
3842#undef IO_BUFFER_DEFINE_DATA_TYPE
#define rb_define_method(klass, mid, func, arity)
Defines klass#mid.
#define rb_define_singleton_method(klass, mid, func, arity)
Defines klass.mid.
static bool RB_OBJ_FROZEN(VALUE obj)
Checks if an object is frozen.
void rb_include_module(VALUE klass, VALUE module)
Includes a module to a class.
VALUE rb_define_class_under(VALUE outer, const char *name, VALUE super)
Defines a class under the namespace of outer.
int rb_block_given_p(void)
Determines if the current method is given a block.
#define T_STRING
Old name of RUBY_T_STRING.
#define OBJ_FROZEN
Old name of RB_OBJ_FROZEN.
#define rb_str_cat2
Old name of rb_str_cat_cstr.
#define CLASS_OF
Old name of rb_class_of.
#define SIZET2NUM
Old name of RB_SIZE2NUM.
#define NUM2UINT
Old name of RB_NUM2UINT.
#define NUM2DBL
Old name of rb_num2dbl.
#define Qnil
Old name of RUBY_Qnil.
#define T_ARRAY
Old name of RUBY_T_ARRAY.
#define NIL_P
Old name of RB_NIL_P.
#define DBL2NUM
Old name of rb_float_new.
#define NUM2SIZET
Old name of RB_NUM2SIZE.
void rb_category_warn(rb_warning_category_t category, const char *fmt,...)
Identical to rb_category_warning(), except it reports unless $VERBOSE is nil.
VALUE rb_eRuntimeError
RuntimeError exception.
@ RB_WARN_CATEGORY_EXPERIMENTAL
Warning is for experimental features.
static VALUE rb_class_of(VALUE obj)
Object to class mapping function.
VALUE rb_mComparable
Comparable module.
#define RB_OBJ_WRITE(old, slot, young)
Declaration of a "back" pointer.
VALUE rb_ary_new_capa(long capa)
Identical to rb_ary_new(), except it additionally specifies how many rooms of objects it should alloc...
VALUE rb_ary_push(VALUE ary, VALUE elem)
Special case of rb_ary_cat() that it adds only one element.
VALUE rb_ary_entry(VALUE ary, long off)
Queries an element of an array.
#define RETURN_ENUMERATOR_KW(obj, argc, argv, kw_splat)
Identical to RETURN_SIZED_ENUMERATOR_KW(), except its size is unknown.
static int rb_check_arity(int argc, int min, int max)
Ensures that the passed integer is in the passed range.
VALUE rb_str_append(VALUE dst, VALUE src)
Identical to rb_str_buf_append(), except it converts the right hand side before concatenating.
#define rb_str_new(str, len)
Allocates an instance of rb_cString.
VALUE rb_str_locktmp(VALUE str)
Obtains a "temporary lock" of the string.
VALUE rb_str_unlocktmp(VALUE str)
Releases a lock formerly obtained by rb_str_locktmp().
VALUE rb_str_buf_new(long capa)
Allocates a "string buffer".
#define rb_str_new_cstr(str)
Identical to rb_str_new, except it assumes the passed pointer is a pointer to a C string.
VALUE rb_class_name(VALUE obj)
Queries the name of the given object's class.
void rb_define_alloc_func(VALUE klass, rb_alloc_func_t func)
Sets the allocator function of a class.
#define RB_SYM2ID
Just another name of rb_sym2id.
VALUE rb_io_get_io(VALUE io)
Identical to rb_io_check_io(), except it raises exceptions on conversion failures.
int rb_io_descriptor(VALUE io)
Returns an integer representing the numeric file descriptor for io.
#define RB_IO_POINTER(obj, fp)
Queries the underlying IO pointer.
VALUE rb_io_get_write_io(VALUE io)
Queries the tied IO for writing.
void * rb_nogvl(void *(*func)(void *), void *data1, rb_unblock_function_t *ubf, void *data2, int flags)
Identical to rb_thread_call_without_gvl(), except it additionally takes "flags" that change the behav...
#define RB_NOGVL_OFFLOAD_SAFE
Passing this flag to rb_nogvl() indicates that the passed function is safe to offload to a background...
#define RB_NUM2INT
Just another name of rb_num2int_inline.
#define RB_UINT2NUM
Just another name of rb_uint2num_inline.
#define RB_INT2NUM
Just another name of rb_int2num_inline.
static unsigned int RB_NUM2UINT(VALUE x)
Converts an instance of rb_cNumeric into C's unsigned int.
#define RB_LL2NUM
Just another name of rb_ll2num_inline.
#define RB_ULL2NUM
Just another name of rb_ull2num_inline.
#define RB_NUM2ULL
Just another name of rb_num2ull_inline.
#define RB_NUM2LL
Just another name of rb_num2ll_inline.
VALUE rb_yield_values(int n,...)
Identical to rb_yield(), except it takes variadic number of parameters and pass them to the block.
VALUE rb_yield(VALUE val)
Yields the block.
static VALUE RB_INT2FIX(long i)
Converts a C's long into an instance of rb_cInteger.
#define RB_NUM2LONG
Just another name of rb_num2long_inline.
VALUE type(ANYARGS)
ANYARGS-ed function type.
VALUE rb_ensure(type *q, VALUE w, type *e, VALUE r)
An equivalent of ensure clause.
#define NUM2OFFT
Converts an instance of rb_cNumeric into C's off_t.
#define RARRAY_LEN
Just another name of rb_array_len.
#define RARRAY_AREF(a, i)
#define StringValue(v)
Ensures that the parameter object is a String.
#define RSTRING_GETMEM(str, ptrvar, lenvar)
Convenient macro to obtain the contents and length at once.
VALUE rb_str_to_str(VALUE obj)
Identical to rb_check_string_type(), except it raises exceptions in case of conversion failures.
#define TypedData_Get_Struct(obj, type, data_type, sval)
Obtains a C struct from inside of a wrapper Ruby object.
#define TypedData_Make_Struct(klass, type, data_type, sval)
Identical to TypedData_Wrap_Struct, except it allocates a new data region internally instead of takin...
#define errno
Ractor-aware version of errno.
#define RB_NO_KEYWORDS
Do not pass keywords.
VALUE rb_fiber_scheduler_current(void)
Identical to rb_fiber_scheduler_get(), except it also returns RUBY_Qnil in case of a blocking fiber.
VALUE rb_fiber_scheduler_io_pwrite(VALUE scheduler, VALUE io, rb_off_t from, VALUE buffer, size_t length, size_t offset)
Non-blocking write to the passed IO at the specified offset.
VALUE rb_fiber_scheduler_io_read(VALUE scheduler, VALUE io, VALUE buffer, size_t length, size_t offset)
Non-blocking read from the passed IO.
static VALUE rb_fiber_scheduler_io_result(ssize_t result, int error)
Wrap a ssize_t and int errno into a single VALUE.
VALUE rb_fiber_scheduler_io_pread(VALUE scheduler, VALUE io, rb_off_t from, VALUE buffer, size_t length, size_t offset)
Non-blocking read from the passed IO at the specified offset.
VALUE rb_fiber_scheduler_io_write(VALUE scheduler, VALUE io, VALUE buffer, size_t length, size_t offset)
Non-blocking write to the passed IO.
static bool RB_NIL_P(VALUE obj)
Checks if the given object is nil.
This is the struct that holds necessary info for a struct.
const char * wrap_struct_name
Name of structs of this kind.
Ruby's IO, metadata and buffers.
uintptr_t ID
Type that represents a Ruby identifier such as a variable name.
uintptr_t VALUE
Type that represents a Ruby object.
static bool RB_TYPE_P(VALUE obj, enum ruby_value_type t)
Queries if the given object is of given type.