17#include "internal/array.h"
18#include "internal/bits.h"
19#include "internal/error.h"
20#include "internal/gc.h"
21#include "internal/numeric.h"
22#include "internal/string.h"
23#include "internal/io.h"
24#include "internal/io_buffer.h"
27VALUE rb_eIOBufferLockedError;
28VALUE rb_eIOBufferAllocationError;
29VALUE rb_eIOBufferAccessError;
30VALUE rb_eIOBufferInvalidatedError;
31VALUE rb_eIOBufferMaskError;
33size_t RUBY_IO_BUFFER_PAGE_SIZE;
34size_t RUBY_IO_BUFFER_MAP_ALIGNMENT;
35size_t RUBY_IO_BUFFER_DEFAULT_SIZE;
44 RB_IO_BUFFER_HEXDUMP_DEFAULT_WIDTH = 16,
45 RB_IO_BUFFER_HEXDUMP_MAXIMUM_WIDTH = 1024,
47 RB_IO_BUFFER_INSPECT_HEXDUMP_MAXIMUM_SIZE = 256,
48 RB_IO_BUFFER_INSPECT_HEXDUMP_WIDTH = 16,
51 RB_IO_BUFFER_FLAGS_MASK = RB_IO_BUFFER_EXTERNAL | RB_IO_BUFFER_INTERNAL | RB_IO_BUFFER_MAPPED | RB_IO_BUFFER_SHARED | RB_IO_BUFFER_PRIVATE | RB_IO_BUFFER_READONLY,
53 RB_IO_BUFFER_ALLOCATION_FLAGS = RB_IO_BUFFER_INTERNAL | RB_IO_BUFFER_MAPPED,
54 RB_IO_BUFFER_MAPPING_FLAGS = RB_IO_BUFFER_SHARED | RB_IO_BUFFER_PRIVATE,
56 RB_IO_BUFFER_DEBUG = 0,
71 enum rb_io_buffer_flags flags;
83io_buffer_map_memory(
size_t size,
int flags)
86 void * base = VirtualAlloc(0, size, MEM_COMMIT, PAGE_READWRITE);
89 rb_sys_fail(
"io_buffer_map_memory:VirtualAlloc");
92 int mmap_flags = MAP_ANONYMOUS;
93 if (flags & RB_IO_BUFFER_SHARED) {
94 mmap_flags |= MAP_SHARED;
97 mmap_flags |= MAP_PRIVATE;
100 void * base = mmap(NULL, size, PROT_READ | PROT_WRITE, mmap_flags, -1, 0);
102 if (base == MAP_FAILED) {
103 rb_sys_fail(
"io_buffer_map_memory:mmap");
106 ruby_annotate_mmap(base, size,
"Ruby:io_buffer_map_memory");
113io_buffer_map_file(
struct rb_io_buffer *buffer,
int descriptor,
size_t size, rb_off_t offset,
enum rb_io_buffer_flags flags)
116 HANDLE file = (HANDLE)_get_osfhandle(descriptor);
117 if (!file) rb_sys_fail(
"io_buffer_map_descriptor:_get_osfhandle");
119 DWORD protect = PAGE_READONLY, access = FILE_MAP_READ;
121 if (flags & RB_IO_BUFFER_READONLY) {
122 buffer->flags |= RB_IO_BUFFER_READONLY;
125 protect = PAGE_READWRITE;
126 access = FILE_MAP_WRITE;
129 if (flags & RB_IO_BUFFER_PRIVATE) {
130 protect = PAGE_WRITECOPY;
131 access = FILE_MAP_COPY;
132 buffer->flags |= RB_IO_BUFFER_PRIVATE;
136 buffer->flags |= RB_IO_BUFFER_EXTERNAL;
137 buffer->flags |= RB_IO_BUFFER_SHARED;
140 HANDLE mapping = CreateFileMapping(file, NULL, protect, 0, 0, NULL);
141 if (RB_IO_BUFFER_DEBUG) fprintf(stderr,
"io_buffer_map_file:CreateFileMapping -> %p\n", mapping);
142 if (!mapping) rb_sys_fail(
"io_buffer_map_descriptor:CreateFileMapping");
144 void *base = MapViewOfFile(mapping, access, (DWORD)(offset >> 32), (DWORD)(offset & 0xFFFFFFFF), size);
147 CloseHandle(mapping);
148 rb_sys_fail(
"io_buffer_map_file:MapViewOfFile");
151 buffer->mapping = mapping;
153 int protect = PROT_READ, access = 0;
155 if (flags & RB_IO_BUFFER_READONLY) {
156 buffer->flags |= RB_IO_BUFFER_READONLY;
159 protect |= PROT_WRITE;
162 if (flags & RB_IO_BUFFER_PRIVATE) {
163 buffer->flags |= RB_IO_BUFFER_PRIVATE;
164 access |= MAP_PRIVATE;
168 buffer->flags |= RB_IO_BUFFER_EXTERNAL;
169 buffer->flags |= RB_IO_BUFFER_SHARED;
170 access |= MAP_SHARED;
173 void *base = mmap(NULL, size, protect, access, descriptor, offset);
175 if (base == MAP_FAILED) {
176 rb_sys_fail(
"io_buffer_map_file:mmap");
183 buffer->flags |= RB_IO_BUFFER_MAPPED;
184 buffer->flags |= RB_IO_BUFFER_FILE;
188io_buffer_experimental(
void)
190 static int warned = 0;
198 "IO::Buffer is experimental and both the Ruby and C interface may change in the future!"
209 buffer->lock_count = 0;
211 buffer->mapping = NULL;
213 buffer->source =
Qnil;
217io_buffer_initialize(
VALUE self,
struct rb_io_buffer *buffer,
void *base,
size_t size,
enum rb_io_buffer_flags flags,
VALUE source)
219 if (source !=
Qnil) {
231 if (flags & RB_IO_BUFFER_INTERNAL) {
232 base = calloc(size, 1);
234 else if (flags & RB_IO_BUFFER_MAPPED) {
235 base = io_buffer_map_memory(size, flags);
239 rb_raise(rb_eIOBufferAllocationError,
"Could not allocate buffer!");
249 buffer->flags = flags;
250 buffer->lock_count = 0;
254 buffer->mapping = NULL;
262 if (buffer->flags & RB_IO_BUFFER_INTERNAL) {
266 if (buffer->flags & RB_IO_BUFFER_MAPPED) {
268 if (buffer->flags & RB_IO_BUFFER_FILE) {
269 UnmapViewOfFile(buffer->base);
272 VirtualFree(buffer->base, 0, MEM_RELEASE);
275 munmap(buffer->base, buffer->size);
287 if (buffer->mapping) {
288 if (RB_IO_BUFFER_DEBUG) fprintf(stderr,
"io_buffer_release:CloseHandle -> %p\n", buffer->mapping);
289 if (!CloseHandle(buffer->mapping)) {
290 fprintf(stderr,
"io_buffer_release:GetLastError -> %lu\n", GetLastError());
292 buffer->mapping = NULL;
296 io_buffer_zero(buffer);
300rb_io_buffer_type_mark(
void *_buffer)
303 if (buffer->source !=
Qnil) {
307 rb_gc_mark(buffer->source);
309 rb_gc_mark_movable(buffer->source);
315rb_io_buffer_type_compact(
void *_buffer)
318 if (buffer->source !=
Qnil) {
323 rb_gc_update_moved(&buffer->source);
329rb_io_buffer_type_free(
void *_buffer)
333 io_buffer_release(buffer);
337rb_io_buffer_type_size(
const void *_buffer)
343 total += buffer->size;
352 .dmark = rb_io_buffer_type_mark,
353 .dfree = rb_io_buffer_type_free,
354 .dsize = rb_io_buffer_type_size,
355 .dcompact = rb_io_buffer_type_compact,
358 .flags = RUBY_TYPED_THREAD_SAFE_FREE | RUBY_TYPED_WB_PROTECTED | RUBY_TYPED_EMBEDDABLE,
362get_io_buffer(
VALUE self)
372 return rb_typeddata_is_kind_of(buffer->source, &rb_io_buffer_type);
378io_buffer_slice_offset(
const struct rb_io_buffer *buffer)
380 return (uintptr_t)buffer->base;
388 while (io_buffer_slice_p(buffer)) {
389 buffer = get_io_buffer(buffer->source);
398 return io_buffer_lock_owner(buffer)->lock_count > 0;
401static inline enum rb_io_buffer_flags
402io_buffer_extract_flags(
VALUE argument)
404 if (rb_int_negative_p(argument)) {
405 rb_raise(rb_eArgError,
"Flags can't be negative!");
408 enum rb_io_buffer_flags flags =
RB_NUM2UINT(argument);
411 return flags & RB_IO_BUFFER_FLAGS_MASK;
414static inline enum rb_io_buffer_flags
415io_buffer_flags_for_map(
enum rb_io_buffer_flags flags)
417 if (flags & RB_IO_BUFFER_INTERNAL) {
418 rb_raise(rb_eArgError,
"IO::Buffer::INTERNAL can't be used with IO::Buffer.map!");
421 if (flags & RB_IO_BUFFER_EXTERNAL) {
422 rb_raise(rb_eArgError,
"IO::Buffer::EXTERNAL can't be used with IO::Buffer.map!");
425 if ((flags & RB_IO_BUFFER_MAPPING_FLAGS) == RB_IO_BUFFER_MAPPING_FLAGS) {
426 rb_raise(rb_eArgError,
"Flags can't include both IO::Buffer::SHARED and IO::Buffer::PRIVATE!");
434io_buffer_extract_offset(
VALUE argument)
436 if (rb_int_negative_p(argument)) {
437 rb_raise(rb_eArgError,
"Offset can't be negative!");
447io_buffer_extract_length(
VALUE argument)
449 if (rb_int_negative_p(argument)) {
450 rb_raise(rb_eArgError,
"Length can't be negative!");
459io_buffer_extract_size(
VALUE argument)
461 if (rb_int_negative_p(argument)) {
462 rb_raise(rb_eArgError,
"Size can't be negative!");
470io_buffer_extract_amount(
VALUE argument)
472 if (rb_int_negative_p(argument)) {
473 rb_raise(rb_eArgError,
"Amount can't be negative!");
482io_buffer_extract_width(
VALUE argument,
size_t minimum)
484 if (rb_int_negative_p(argument)) {
485 rb_raise(rb_eArgError,
"Width can't be negative!");
490 if (width < minimum) {
491 rb_raise(rb_eArgError,
"Width must be at least %" PRIuSIZE
"!", minimum);
494 if (width > RB_IO_BUFFER_HEXDUMP_MAXIMUM_WIDTH) {
495 rb_raise(rb_eArgError,
"Width must be at most %" PRIuSIZE
"!", (
size_t)RB_IO_BUFFER_HEXDUMP_MAXIMUM_WIDTH);
506io_buffer_default_length(
const struct rb_io_buffer *buffer,
size_t offset)
508 if (offset > buffer->size) {
509 rb_raise(rb_eArgError,
"The given offset is bigger than the buffer size!");
513 return buffer->size - offset;
521io_buffer_extract_offset_length(
VALUE self,
int argc,
VALUE argv[],
size_t *offset,
size_t *length)
525 if (argc >= 1 && !
NIL_P(argv[0])) {
526 *offset = io_buffer_extract_offset(argv[0]);
532 if (argc >= 2 && !
NIL_P(argv[1])) {
533 *length = io_buffer_extract_length(argv[1]);
536 *length = io_buffer_default_length(buffer, *offset);
543rb_io_buffer_type_allocate(
VALUE self)
545 io_buffer_experimental();
550 io_buffer_zero(buffer);
555static VALUE io_buffer_for_make_instance(
VALUE klass,
VALUE string,
enum rb_io_buffer_flags flags)
557 VALUE instance = rb_io_buffer_type_allocate(klass);
561 flags |= RB_IO_BUFFER_EXTERNAL;
564 flags |= RB_IO_BUFFER_READONLY;
566 if (!(flags & RB_IO_BUFFER_READONLY))
567 rb_str_modify(
string);
571 io_buffer_initialize(instance, buffer, (
void *)0, RSTRING_LEN(
string), flags,
string);
580 enum rb_io_buffer_flags flags;
584io_buffer_for_yield_instance(
VALUE _arguments)
588 arguments->instance = io_buffer_for_make_instance(arguments->klass, arguments->string, arguments->flags);
594 return rb_yield(arguments->instance);
598io_buffer_for_yield_instance_ensure(
VALUE _arguments)
602 if (arguments->instance !=
Qnil) {
603 rb_io_buffer_free(arguments->instance);
617 enum rb_io_buffer_flags flags;
623static VALUE rb_io_buffer_locked_ensure(
VALUE self);
632io_buffer_for_locked_callback_call(
VALUE _arguments)
636 return arguments->callback(arguments->buffer, arguments->argument);
644 .callback = callback,
645 .argument = argument,
648 rb_io_buffer_lock(buffer);
649 return rb_ensure(io_buffer_for_locked_callback_call, (
VALUE)&arguments, rb_io_buffer_locked_ensure, buffer);
653io_buffer_for_callback_call(
VALUE _arguments)
657 arguments->instance = io_buffer_for_make_instance(arguments->klass, arguments->string, arguments->flags);
661 arguments->locked = 1;
664 return io_buffer_for_locked_callback(arguments->instance, arguments->callback, arguments->argument);
668io_buffer_for_callback_ensure(
VALUE _arguments)
672 if (arguments->instance !=
Qnil) {
673 rb_io_buffer_free(arguments->instance);
676 if (arguments->locked) {
687 return io_buffer_for_locked_callback(string_or_buffer, callback, argument);
692 .klass = rb_cIOBuffer,
693 .string = string_or_buffer,
695 .flags = RB_IO_BUFFER_READONLY,
697 .callback = callback,
698 .argument = argument,
701 io_buffer_for_callback_ensure, (
VALUE)&arguments);
704 rb_raise(
rb_eTypeError,
"expected String or IO::Buffer, not %"PRIsVALUE,
710static int io_buffer_readonly_p(
struct rb_io_buffer *buffer);
716 struct rb_io_buffer *buffer = get_io_buffer(string_or_buffer);
717 if (io_buffer_readonly_p(buffer)) {
718 rb_raise(rb_eArgError,
"buffer is read-only");
720 return io_buffer_for_locked_callback(string_or_buffer, callback, argument);
725 .klass = rb_cIOBuffer,
726 .string = string_or_buffer,
730 .callback = callback,
731 .argument = argument,
734 io_buffer_for_callback_ensure, (
VALUE)&arguments);
737 rb_raise(
rb_eTypeError,
"expected String or IO::Buffer, not %"PRIsVALUE,
779rb_io_buffer_type_for(
VALUE klass,
VALUE string)
793 return rb_ensure(io_buffer_for_yield_instance, (
VALUE)&arguments, io_buffer_for_yield_instance_ensure, (
VALUE)&arguments);
799 return io_buffer_for_make_instance(klass,
string, RB_IO_BUFFER_READONLY);
817rb_io_buffer_type_string(
VALUE klass,
VALUE length)
827 rb_ensure(io_buffer_for_yield_instance, (
VALUE)&arguments, io_buffer_for_yield_instance_ensure, (
VALUE)&arguments);
833rb_io_buffer_new(
void *base,
size_t size,
enum rb_io_buffer_flags flags)
835 VALUE instance = rb_io_buffer_type_allocate(rb_cIOBuffer);
839 io_buffer_initialize(instance, buffer, base, size, flags,
Qnil);
845rb_io_buffer_new_locked(
void *base,
size_t size,
enum rb_io_buffer_flags flags)
847 VALUE instance = rb_io_buffer_new(base, size, flags);
849 rb_io_buffer_lock(instance);
855rb_io_buffer_map(
VALUE io,
size_t size, rb_off_t offset,
enum rb_io_buffer_flags flags)
857 if (UNLIKELY(offset < 0)) {
858 rb_raise(rb_eArgError,
859 "Offset (%" PRIsVALUE
") can't be negative!",
863 if (UNLIKELY((uintmax_t)offset % RUBY_IO_BUFFER_MAP_ALIGNMENT != 0)) {
864 rb_raise(rb_eArgError,
865 "Offset (%" PRIsVALUE
") must be a multiple of IO::Buffer::MAP_ALIGNMENT (%" PRIuSIZE
")!",
867 RUBY_IO_BUFFER_MAP_ALIGNMENT);
870 VALUE instance = rb_io_buffer_type_allocate(rb_cIOBuffer);
876 io_buffer_map_file(buffer, descriptor, size, offset, flags);
933io_buffer_map(
int argc,
VALUE *argv,
VALUE klass)
940 rb_off_t file_size = rb_file_size(io);
942 if (UNLIKELY(file_size <= 0)) {
943 rb_raise(rb_eArgError,
"Invalid negative or zero file size!");
946 else if (UNLIKELY((uintmax_t)file_size > SIZE_MAX)) {
947 rb_raise(rb_eArgError,
"File larger than address space!");
951 if (argc >= 2 && !
RB_NIL_P(argv[1])) {
952 size = io_buffer_extract_size(argv[1]);
953 if (UNLIKELY(size == 0)) {
954 rb_raise(rb_eArgError,
"Size can't be zero!");
956 if (UNLIKELY(size > (
size_t)file_size)) {
957 rb_raise(rb_eArgError,
958 "Size (%" PRIuSIZE
") can't be larger than "
959 "file size (%" PRIuSIZE
")",
966 size = (size_t)file_size;
973 if (UNLIKELY(offset < 0)) {
974 rb_raise(rb_eArgError,
975 "Offset (%" PRIsVALUE
") can't be negative!",
978 if (UNLIKELY(offset >= file_size)) {
979 rb_raise(rb_eArgError,
980 "Offset (%" PRIsVALUE
") can't be larger than "
981 "file size (%" PRIuSIZE
")",
987 size = (size_t)(file_size - offset);
989 else if (UNLIKELY((
size_t)(file_size - offset) < size)) {
990 size_t maximum_offset =
991 ((size_t)file_size - size) / RUBY_IO_BUFFER_MAP_ALIGNMENT *
992 RUBY_IO_BUFFER_MAP_ALIGNMENT;
993 rb_raise(rb_eArgError,
994 "Offset (%" PRIsVALUE
") can't be larger than "
995 "%" PRIuSIZE
" for requested size (%" PRIuSIZE
")",
1002 enum rb_io_buffer_flags flags = 0;
1004 flags = io_buffer_extract_flags(argv[3]);
1006 flags = io_buffer_flags_for_map(flags);
1008 return rb_io_buffer_map(io, size, offset, flags);
1012static inline enum rb_io_buffer_flags
1013io_flags_for_size(
size_t size)
1015 if (size >= RUBY_IO_BUFFER_PAGE_SIZE) {
1016 return RB_IO_BUFFER_MAPPED;
1019 return RB_IO_BUFFER_INTERNAL;
1022static inline enum rb_io_buffer_flags
1023io_buffer_flags_for_new(
enum rb_io_buffer_flags flags,
size_t size)
1030 if (!(flags & RB_IO_BUFFER_ALLOCATION_FLAGS)) {
1031 if (flags & RB_IO_BUFFER_MAPPING_FLAGS) {
1033 flags |= RB_IO_BUFFER_MAPPED;
1037 flags |= io_flags_for_size(size);
1041 enum rb_io_buffer_flags allocation = flags & RB_IO_BUFFER_ALLOCATION_FLAGS;
1044 if ((
unsigned int)allocation == RB_IO_BUFFER_ALLOCATION_FLAGS) {
1045 rb_raise(rb_eArgError,
"Flags can't include both IO::Buffer::INTERNAL and IO::Buffer::MAPPED!");
1048 if (flags & RB_IO_BUFFER_EXTERNAL) {
1049 rb_raise(rb_eArgError,
"IO::Buffer::EXTERNAL can't be used with IO::Buffer.new!");
1052 if ((flags & RB_IO_BUFFER_MAPPING_FLAGS) && allocation != RB_IO_BUFFER_MAPPED) {
1053 rb_raise(rb_eArgError,
"IO::Buffer::SHARED and IO::Buffer::PRIVATE require IO::Buffer::MAPPED!");
1056 if ((flags & RB_IO_BUFFER_MAPPING_FLAGS) == RB_IO_BUFFER_MAPPING_FLAGS) {
1057 rb_raise(rb_eArgError,
"Flags can't include both IO::Buffer::SHARED and IO::Buffer::PRIVATE!");
1093rb_io_buffer_initialize(
int argc,
VALUE *argv,
VALUE self)
1101 size = io_buffer_extract_size(argv[0]);
1104 size = RUBY_IO_BUFFER_DEFAULT_SIZE;
1107 enum rb_io_buffer_flags flags = 0;
1109 flags = io_buffer_extract_flags(argv[1]);
1111 flags = io_buffer_flags_for_new(flags, size);
1113 io_buffer_initialize(self, buffer, NULL, size, flags,
Qnil);
1124io_buffer_try_get_bytes(
struct rb_io_buffer *buffer,
void **base,
size_t *size)
1126 size_t length = buffer->size;
1128 void *source_base = NULL;
1130 while (buffer->source !=
Qnil) {
1134 if (io_buffer_slice_p(buffer)) {
1135 source_buffer = get_io_buffer(buffer->source);
1136 source_size = source_buffer->size;
1143 size_t relative_offset = io_buffer_slice_offset(buffer);
1144 if (relative_offset > source_size || buffer->size > source_size - relative_offset ||
1145 relative_offset > SIZE_MAX - offset) {
1151 offset += relative_offset;
1152 if (!source_buffer) {
1153 *base = source_base ? (
char *)source_base + offset : NULL;
1158 buffer = source_buffer;
1163 if (io_buffer_slice_p(buffer)) {
1170 *base = buffer->base ? (
char *)buffer->base + offset : NULL;
1178 if (buffer->source !=
Qnil) {
1182 return io_buffer_try_get_bytes(buffer, &base, &size);
1189enum rb_io_buffer_flags
1190rb_io_buffer_get_bytes(
VALUE self,
void **base,
size_t *size)
1194 if (io_buffer_try_get_bytes(buffer, base, size)) {
1195 enum rb_io_buffer_flags flags = buffer->flags;
1196 if (io_buffer_readonly_p(buffer)) flags |= RB_IO_BUFFER_READONLY;
1205io_buffer_validate_for_writing(
struct rb_io_buffer *buffer)
1207 if (io_buffer_readonly_p(buffer)) {
1208 rb_raise(rb_eIOBufferAccessError,
"Buffer is not writable!");
1211 if (!io_buffer_validate(buffer)) {
1212 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer is invalid!");
1217get_io_buffer_for_writing(
VALUE self)
1219 rb_check_frozen(self);
1222 io_buffer_validate_for_writing(buffer);
1227io_buffer_get_bytes_for_writing(
struct rb_io_buffer *buffer,
void **base,
size_t *size)
1229 io_buffer_validate_for_writing(buffer);
1231 io_buffer_try_get_bytes(buffer, base, size);
1235rb_io_buffer_get_bytes_for_writing(
VALUE self,
void **base,
size_t *size)
1239 io_buffer_get_bytes_for_writing(buffer, base, size);
1243io_buffer_validate_for_reading(
struct rb_io_buffer *buffer)
1245 if (!io_buffer_validate(buffer)) {
1246 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer has been invalidated!");
1251io_buffer_get_bytes_for_reading(
struct rb_io_buffer *buffer,
const void **base,
size_t *size)
1253 io_buffer_validate_for_reading(buffer);
1255 void *writable_base = NULL;
1256 io_buffer_try_get_bytes(buffer, &writable_base, size);
1257 *base = writable_base;
1261rb_io_buffer_get_bytes_for_reading(
VALUE self,
const void **base,
size_t *size)
1265 io_buffer_get_bytes_for_reading(buffer, base, size);
1278rb_io_buffer_to_s(
VALUE self)
1289 io_buffer_try_get_bytes(buffer, &base, &size);
1290 rb_str_catf(result,
" %p+%"PRIdSIZE, base, buffer->size);
1296 if (buffer->flags & RB_IO_BUFFER_EXTERNAL) {
1300 if (buffer->flags & RB_IO_BUFFER_INTERNAL) {
1304 if (buffer->flags & RB_IO_BUFFER_MAPPED) {
1308 if (buffer->flags & RB_IO_BUFFER_FILE) {
1312 if (buffer->flags & RB_IO_BUFFER_SHARED) {
1316 if (io_buffer_locked(buffer)) {
1320 if (buffer->flags & RB_IO_BUFFER_PRIVATE) {
1324 if (io_buffer_readonly_p(buffer)) {
1328 if (buffer->source !=
Qnil) {
1332 if (!io_buffer_validate(buffer)) {
1342io_buffer_hexdump_output_size(
size_t width,
size_t size,
int first)
1345 size_t total = size;
1347 size_t whole_lines = (size / width);
1348 size_t partial_line = (size % width) ? 1 : 0;
1353 total += (whole_lines + partial_line) * (1 + 10 + width*3 + 1 + 1);
1356 if (size && first) total -= 1;
1365io_buffer_hexdump(
VALUE string,
size_t width,
const char *base,
size_t length,
size_t offset,
int first)
1367 char *text = alloca(width+1);
1370 for (; offset < length; offset += width) {
1371 memset(text,
'\0', width);
1373 rb_str_catf(
string,
"0x%08" PRIxSIZE
" ", offset);
1377 rb_str_catf(
string,
"\n0x%08" PRIxSIZE
" ", offset);
1380 for (
size_t i = 0; i < width; i += 1) {
1381 if (offset+i < length) {
1382 unsigned char value = ((
unsigned char*)base)[offset+i];
1384 if (value < 127 && isprint(value)) {
1385 text[i] = (char)value;
1391 rb_str_catf(
string,
" %02x", value);
1398 rb_str_catf(
string,
" %s", text);
1417rb_io_buffer_inspect(
VALUE self)
1421 VALUE result = rb_io_buffer_to_s(self);
1425 if (io_buffer_try_get_bytes(buffer, &base, &total) && base) {
1427 size_t size = buffer->size;
1430 if (size > RB_IO_BUFFER_INSPECT_HEXDUMP_MAXIMUM_SIZE) {
1431 size = RB_IO_BUFFER_INSPECT_HEXDUMP_MAXIMUM_SIZE;
1435 io_buffer_hexdump(result, RB_IO_BUFFER_INSPECT_HEXDUMP_WIDTH, base, size, 0, 0);
1438 rb_str_catf(result,
"\n(and %" PRIuSIZE
" more bytes not printed)", buffer->size - size);
1452rb_io_buffer_size(
VALUE self)
1482rb_io_buffer_source(
VALUE self)
1484 return get_io_buffer(self)->source;
1504rb_io_buffer_valid_p(
VALUE self)
1508 return RBOOL(io_buffer_validate(buffer));
1529rb_io_buffer_null_p(
VALUE self)
1535 io_buffer_try_get_bytes(buffer, &base, &size);
1537 return RBOOL(base == NULL);
1550rb_io_buffer_empty_p(
VALUE self)
1554 return RBOOL(buffer->size == 0);
1569rb_io_buffer_external_p(
VALUE self)
1573 return RBOOL(buffer->flags & RB_IO_BUFFER_EXTERNAL);
1593rb_io_buffer_internal_p(
VALUE self)
1597 return RBOOL(buffer->flags & RB_IO_BUFFER_INTERNAL);
1614rb_io_buffer_mapped_p(
VALUE self)
1618 return RBOOL(buffer->flags & RB_IO_BUFFER_MAPPED);
1645rb_io_buffer_shared_p(
VALUE self)
1649 return RBOOL(buffer->flags & RB_IO_BUFFER_SHARED);
1666rb_io_buffer_locked_p(
VALUE self)
1670 return RBOOL(io_buffer_locked(buffer));
1696rb_io_buffer_private_p(
VALUE self)
1700 return RBOOL(buffer->flags & RB_IO_BUFFER_PRIVATE);
1707 if (buffer->flags & RB_IO_BUFFER_READONLY)
1710 VALUE source = buffer->source;
1720 buffer = get_io_buffer(source);
1737rb_io_buffer_readonly_p(
VALUE self)
1741 return RBOOL(io_buffer_readonly_p(buffer));
1747 struct rb_io_buffer *owner = io_buffer_lock_owner(buffer);
1749 if (owner->lock_count == SIZE_MAX) {
1750 rb_raise(rb_eIOBufferLockedError,
"It's locks all the way down!");
1753 owner->lock_count += 1;
1757rb_io_buffer_lock(
VALUE self)
1761 io_buffer_lock(buffer);
1769 struct rb_io_buffer *owner = io_buffer_lock_owner(buffer);
1771 if (owner->lock_count == 0) {
1772 rb_raise(rb_eIOBufferLockedError,
"Buffer not locked!");
1775 owner->lock_count -= 1;
1779rb_io_buffer_unlock(
VALUE self)
1783 io_buffer_unlock(buffer);
1789rb_io_buffer_try_unlock(
VALUE self)
1792 struct rb_io_buffer *owner = io_buffer_lock_owner(buffer);
1794 if (owner->lock_count > 0) {
1795 owner->lock_count -= 1;
1804rb_io_buffer_locked_ensure(
VALUE self)
1808 io_buffer_unlock(buffer);
1815 VALUE (*callback)(
const void *base,
size_t size,
VALUE argument);
1820io_buffer_readable_bytes_call(
VALUE _arguments)
1826 rb_io_buffer_get_bytes_for_reading(arguments->self, &base, &size);
1828 return arguments->callback(base, size, arguments->argument);
1832rb_io_buffer_locked_for_reading(
VALUE self,
VALUE (*callback)(
const void *base,
size_t size,
VALUE argument),
VALUE argument)
1835 io_buffer_validate_for_reading(buffer);
1839 .callback = callback,
1840 .argument = argument,
1843 rb_io_buffer_lock(self);
1844 return rb_ensure(io_buffer_readable_bytes_call, (
VALUE)&arguments, rb_io_buffer_locked_ensure, self);
1849 VALUE (*callback)(
void *base,
size_t size,
VALUE argument);
1854io_buffer_writable_bytes_call(
VALUE _arguments)
1860 rb_io_buffer_get_bytes_for_writing(arguments->self, &base, &size);
1862 return arguments->callback(base, size, arguments->argument);
1866rb_io_buffer_locked_for_writing(
VALUE self,
VALUE (*callback)(
void *base,
size_t size,
VALUE argument),
VALUE argument)
1868 get_io_buffer_for_writing(self);
1872 .callback = callback,
1873 .argument = argument,
1876 rb_io_buffer_lock(self);
1877 return rb_ensure(io_buffer_writable_bytes_call, (
VALUE)&arguments, rb_io_buffer_locked_ensure, self);
1908rb_io_buffer_locked(
VALUE self)
1914 io_buffer_validate_for_reading(buffer);
1916 io_buffer_lock(buffer);
1922rb_io_buffer_free(
VALUE self)
1926 if (io_buffer_locked(buffer)) {
1927 rb_raise(rb_eIOBufferLockedError,
"Buffer is locked!");
1930 io_buffer_release(buffer);
1969io_buffer_free(
VALUE self)
1971 rb_check_frozen(self);
1973 return rb_io_buffer_free(self);
1979 struct rb_io_buffer *owner = io_buffer_lock_owner(buffer);
1984 if (owner->lock_count != 1) {
1985 rb_bug(
"rb_io_buffer_free_locked: expected lock count 1, got %" PRIuSIZE, owner->lock_count);
1988 io_buffer_unlock(buffer);
1989 io_buffer_release(buffer);
1995size_sum_is_bigger_than(
size_t a,
size_t b,
size_t x)
1997 struct rbimpl_size_overflow_tag size = rbimpl_size_add_overflow(a, b);
1998 return size.overflowed || size.result > x;
2004io_buffer_validate_range(
struct rb_io_buffer *buffer,
size_t offset,
size_t length)
2006 io_buffer_validate_for_reading(buffer);
2008 if (size_sum_is_bigger_than(offset, length, buffer->size)) {
2009 rb_raise(rb_eArgError,
"Specified offset+length is bigger than the buffer size!");
2033rb_io_buffer_hexdump(
int argc,
VALUE *argv,
VALUE self)
2037 size_t offset, length;
2038 struct rb_io_buffer *buffer = io_buffer_extract_offset_length(self, argc, argv, &offset, &length);
2040 size_t width = RB_IO_BUFFER_HEXDUMP_DEFAULT_WIDTH;
2042 width = io_buffer_extract_width(argv[2], 1);
2046 io_buffer_validate_range(buffer, offset, length);
2052 if (io_buffer_try_get_bytes(buffer, &base, &size) && base) {
2053 result =
rb_str_buf_new(io_buffer_hexdump_output_size(width, length, 1));
2055 io_buffer_hexdump(result, width, base, offset+length, offset, 1);
2062rb_io_buffer_slice(
struct rb_io_buffer *buffer,
VALUE self,
size_t offset,
size_t length)
2064 io_buffer_validate_range(buffer, offset, length);
2069 slice->size = length;
2073 slice->base = (
void *)(uintptr_t)offset;
2136io_buffer_slice(
int argc,
VALUE *argv,
VALUE self)
2140 size_t offset, length;
2141 struct rb_io_buffer *buffer = io_buffer_extract_offset_length(self, argc, argv, &offset, &length);
2143 return rb_io_buffer_slice(buffer, self, offset, length);
2147rb_io_buffer_transfer(
VALUE self)
2151 if (io_buffer_locked(buffer)) {
2152 rb_raise(rb_eIOBufferLockedError,
"Cannot transfer ownership of locked buffer!");
2159 *transferred = *buffer;
2160 io_buffer_zero(buffer);
2191io_buffer_transfer(
VALUE self)
2193 rb_check_frozen(self);
2195 return rb_io_buffer_transfer(self);
2199io_buffer_resize_clear(
struct rb_io_buffer *buffer,
void* base,
size_t size)
2201 if (size > buffer->size) {
2202 memset((
unsigned char*)base+buffer->size, 0, size - buffer->size);
2211 enum rb_io_buffer_flags flags = io_flags_for_size(size) | (buffer->flags & RB_IO_BUFFER_READONLY);
2212 io_buffer_initialize(self, &resized, NULL, size, flags,
Qnil);
2215 size_t preserve = buffer->size;
2216 if (preserve > size) preserve = size;
2217 memcpy(resized.base, buffer->base, preserve);
2219 io_buffer_resize_clear(buffer, resized.base, size);
2222 io_buffer_release(buffer);
2227rb_io_buffer_advance(
VALUE self,
size_t amount)
2231 if (buffer->flags & RB_IO_BUFFER_ALLOCATION_FLAGS) {
2232 rb_raise(rb_eIOBufferAccessError,
"Cannot advance an owning buffer!");
2235 if (!io_buffer_validate(buffer)) {
2236 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer has been invalidated!");
2239 if (amount > buffer->size) {
2240 rb_raise(rb_eArgError,
"Advance amount exceeds buffer size!");
2243 if (buffer->source !=
Qnil) {
2246 buffer->base = (
void *)(uintptr_t)(io_buffer_slice_offset(buffer) + amount);
2248 else if (buffer->base) {
2250 buffer->base = (
char *)buffer->base + amount;
2253 buffer->size -= amount;
2257io_buffer_resize_slice(
struct rb_io_buffer *slice,
size_t size)
2259 struct rb_io_buffer *source = get_io_buffer(slice->source);
2261 void *source_base = NULL;
2262 size_t source_size = 0;
2264 if (!io_buffer_try_get_bytes(source, &source_base, &source_size)) {
2265 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer is invalid!");
2268 size_t offset = io_buffer_slice_offset(slice);
2270 if (offset > source_size) {
2271 rb_raise(rb_eIOBufferInvalidatedError,
"Buffer is invalid!");
2274 if (size > source_size - offset) {
2275 rb_raise(rb_eArgError,
"Resized slice exceeds its source buffer!");
2284rb_io_buffer_resize(
VALUE self,
size_t size)
2288 if (io_buffer_slice_p(buffer)) {
2290 io_buffer_resize_slice(buffer, size);
2294 io_buffer_validate_for_reading(buffer);
2296 if (io_buffer_locked(buffer)) {
2297 rb_raise(rb_eIOBufferLockedError,
"Cannot resize locked buffer!");
2304 if (buffer->flags & RB_IO_BUFFER_EXTERNAL) {
2305 rb_raise(rb_eIOBufferAccessError,
"Cannot resize external buffer!");
2308 if (buffer->base == NULL) {
2309 io_buffer_initialize(self, buffer, NULL, size, io_flags_for_size(size),
Qnil);
2314 io_buffer_release(buffer);
2318#if defined(HAVE_MREMAP) && defined(MREMAP_MAYMOVE)
2319 if (buffer->flags & RB_IO_BUFFER_MAPPED) {
2320 void *base = mremap(buffer->base, buffer->size, size, MREMAP_MAYMOVE);
2322 if (base == MAP_FAILED) {
2323 rb_sys_fail(
"rb_io_buffer_resize:mremap");
2326 io_buffer_resize_clear(buffer, base, size);
2328 buffer->base = base;
2329 buffer->size = size;
2335 if (buffer->flags & RB_IO_BUFFER_INTERNAL) {
2336 void *base = realloc(buffer->base, size);
2339 rb_sys_fail(
"rb_io_buffer_resize:realloc");
2342 io_buffer_resize_clear(buffer, base, size);
2344 buffer->base = base;
2345 buffer->size = size;
2350 io_buffer_resize_copy(self, buffer, size);
2380 rb_check_frozen(self);
2382 rb_io_buffer_resize(self, io_buffer_extract_size(size));
2412 rb_check_frozen(self);
2414 size_t size = io_buffer_extract_amount(amount);
2417 rb_check_frozen(self);
2419 rb_io_buffer_advance(self, size);
2441 const void *ptr1, *ptr2;
2442 size_t size1, size2;
2444 rb_io_buffer_get_bytes_for_reading(self, &ptr1, &size1);
2445 rb_io_buffer_get_bytes_for_reading(other, &ptr2, &size2);
2447 if (size1 < size2) {
2451 if (size1 > size2) {
2461 return RB_INT2NUM(memcmp(ptr1, ptr2, size1));
2465io_buffer_validate_type(
size_t size,
size_t offset,
size_t extend)
2467 if (size_sum_is_bigger_than(offset, extend, size)) {
2468 rb_raise(rb_eArgError,
"Type extends beyond end of buffer! (offset=%"PRIdSIZE
" > size=%"PRIdSIZE
")", offset, size);
2493#define ruby_swap8(value) value
2501ruby_swapf32(
float value)
2503 union swapf32 swap = {.value = value};
2504 swap.integral = ruby_swap32(swap.integral);
2514ruby_swapf64(
double value)
2516 union swapf64 swap = {.value = value};
2517 swap.integral = ruby_swap64(swap.integral);
2529#ifdef HAVE_UINT128_T
2530#if __has_builtin(__builtin_bswap128)
2531 result.value = __builtin_bswap128(x.value);
2534 uint64_t low = (uint64_t)x.value;
2535 uint64_t high = (uint64_t)(x.value >> 64);
2536 low = ruby_swap64(low);
2537 high = ruby_swap64(high);
2538 result.value = ((uint128_t)low << 64) | high;
2545 result.parts.low = ruby_swap64(x.parts.high);
2546 result.parts.high = ruby_swap64(x.parts.low);
2557 conversion.uint128 = ruby_swap128_uint(conversion.uint128);
2558 return conversion.int128;
2561#define IO_BUFFER_VALIDATE_TYPE_FOR_WRITING(buffer, base, size, offset, type) \
2562 (io_buffer_get_bytes_for_writing(buffer, &(base), &(size)), \
2563 io_buffer_validate_type(size, offset, sizeof(type)))
2565#define IO_BUFFER_DECLARE_TYPE(name, type, endian, wrap, unwrap, swap) \
2566static ID RB_IO_BUFFER_DATA_TYPE_##name; \
2569io_buffer_read_##name(const void* base, size_t size, size_t *offset) \
2571 io_buffer_validate_type(size, *offset, sizeof(type)); \
2573 memcpy(&value, (char*)base + *offset, sizeof(type)); \
2574 if (endian != RB_IO_BUFFER_HOST_ENDIAN) value = swap(value); \
2575 *offset += sizeof(type); \
2576 return wrap(value); \
2580io_buffer_write_##name(struct rb_io_buffer* buffer, size_t *offset, VALUE _value) \
2582 void* base; size_t size; \
2583 IO_BUFFER_VALIDATE_TYPE_FOR_WRITING(buffer, base, size, *offset, type); \
2584 type value = unwrap(_value); \
2585 IO_BUFFER_VALIDATE_TYPE_FOR_WRITING(buffer, base, size, *offset, type); \
2586 if (endian != RB_IO_BUFFER_HOST_ENDIAN) value = swap(value); \
2587 memcpy((char*)base + *offset, &value, sizeof(type)); \
2588 *offset += sizeof(type); \
2592 RB_IO_BUFFER_DATA_TYPE_##name##_SIZE = sizeof(type) \
2600IO_BUFFER_DECLARE_TYPE(s16, int16_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap16)
2601IO_BUFFER_DECLARE_TYPE(S16, int16_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap16)
2605IO_BUFFER_DECLARE_TYPE(s32, int32_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap32)
2606IO_BUFFER_DECLARE_TYPE(S32, int32_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_INT2NUM,
RB_NUM2INT, ruby_swap32)
2608IO_BUFFER_DECLARE_TYPE(u64, uint64_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_ULL2NUM,
RB_NUM2ULL, ruby_swap64)
2609IO_BUFFER_DECLARE_TYPE(U64, uint64_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_ULL2NUM,
RB_NUM2ULL, ruby_swap64)
2610IO_BUFFER_DECLARE_TYPE(s64, int64_t, RB_IO_BUFFER_LITTLE_ENDIAN,
RB_LL2NUM,
RB_NUM2LL, ruby_swap64)
2611IO_BUFFER_DECLARE_TYPE(S64, int64_t, RB_IO_BUFFER_BIG_ENDIAN,
RB_LL2NUM,
RB_NUM2LL, ruby_swap64)
2613IO_BUFFER_DECLARE_TYPE(u128,
rb_uint128_t, RB_IO_BUFFER_LITTLE_ENDIAN, rb_uint128_to_numeric, rb_numeric_to_uint128, ruby_swap128_uint)
2614IO_BUFFER_DECLARE_TYPE(U128,
rb_uint128_t, RB_IO_BUFFER_BIG_ENDIAN, rb_uint128_to_numeric, rb_numeric_to_uint128, ruby_swap128_uint)
2615IO_BUFFER_DECLARE_TYPE(s128,
rb_int128_t, RB_IO_BUFFER_LITTLE_ENDIAN, rb_int128_to_numeric, rb_numeric_to_int128, ruby_swap128_int)
2616IO_BUFFER_DECLARE_TYPE(S128,
rb_int128_t, RB_IO_BUFFER_BIG_ENDIAN, rb_int128_to_numeric, rb_numeric_to_int128, ruby_swap128_int)
2618IO_BUFFER_DECLARE_TYPE(f32,
float, RB_IO_BUFFER_LITTLE_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf32)
2619IO_BUFFER_DECLARE_TYPE(F32,
float, RB_IO_BUFFER_BIG_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf32)
2620IO_BUFFER_DECLARE_TYPE(f64,
double, RB_IO_BUFFER_LITTLE_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf64)
2621IO_BUFFER_DECLARE_TYPE(F64,
double, RB_IO_BUFFER_BIG_ENDIAN,
DBL2NUM,
NUM2DBL, ruby_swapf64)
2622#undef IO_BUFFER_DECLARE_TYPE
2625io_buffer_buffer_type_size(
ID buffer_type)
2627#define IO_BUFFER_DATA_TYPE_SIZE(name) if (buffer_type == RB_IO_BUFFER_DATA_TYPE_##name) return RB_IO_BUFFER_DATA_TYPE_##name##_SIZE;
2628 IO_BUFFER_DATA_TYPE_SIZE(U8)
2629 IO_BUFFER_DATA_TYPE_SIZE(S8)
2630 IO_BUFFER_DATA_TYPE_SIZE(u16)
2631 IO_BUFFER_DATA_TYPE_SIZE(U16)
2632 IO_BUFFER_DATA_TYPE_SIZE(s16)
2633 IO_BUFFER_DATA_TYPE_SIZE(S16)
2634 IO_BUFFER_DATA_TYPE_SIZE(u32)
2635 IO_BUFFER_DATA_TYPE_SIZE(U32)
2636 IO_BUFFER_DATA_TYPE_SIZE(s32)
2637 IO_BUFFER_DATA_TYPE_SIZE(S32)
2638 IO_BUFFER_DATA_TYPE_SIZE(u64)
2639 IO_BUFFER_DATA_TYPE_SIZE(U64)
2640 IO_BUFFER_DATA_TYPE_SIZE(s64)
2641 IO_BUFFER_DATA_TYPE_SIZE(S64)
2642 IO_BUFFER_DATA_TYPE_SIZE(u128)
2643 IO_BUFFER_DATA_TYPE_SIZE(U128)
2644 IO_BUFFER_DATA_TYPE_SIZE(s128)
2645 IO_BUFFER_DATA_TYPE_SIZE(S128)
2646 IO_BUFFER_DATA_TYPE_SIZE(f32)
2647 IO_BUFFER_DATA_TYPE_SIZE(F32)
2648 IO_BUFFER_DATA_TYPE_SIZE(f64)
2649 IO_BUFFER_DATA_TYPE_SIZE(F64)
2650#undef IO_BUFFER_DATA_TYPE_SIZE
2652 rb_raise(rb_eArgError,
"Invalid type name!");
2656io_buffer_type_id(
VALUE name)
2662#define TYPE_ID(name) io_buffer_type_id(name)
2675io_buffer_size_of(
VALUE klass,
VALUE buffer_type)
2679 for (
long i = 0; i <
RARRAY_LEN(buffer_type); i++) {
2680 total += io_buffer_buffer_type_size(TYPE_ID(
RARRAY_AREF(buffer_type, i)));
2685 return SIZET2NUM(io_buffer_buffer_type_size(TYPE_ID(buffer_type)));
2690rb_io_buffer_get_value(
const void* base,
size_t size,
ID buffer_type,
size_t *offset)
2692#define IO_BUFFER_GET_VALUE(name) if (buffer_type == RB_IO_BUFFER_DATA_TYPE_##name) return io_buffer_read_##name(base, size, offset);
2693 IO_BUFFER_GET_VALUE(U8)
2694 IO_BUFFER_GET_VALUE(S8)
2696 IO_BUFFER_GET_VALUE(u16)
2697 IO_BUFFER_GET_VALUE(U16)
2698 IO_BUFFER_GET_VALUE(s16)
2699 IO_BUFFER_GET_VALUE(S16)
2701 IO_BUFFER_GET_VALUE(u32)
2702 IO_BUFFER_GET_VALUE(U32)
2703 IO_BUFFER_GET_VALUE(s32)
2704 IO_BUFFER_GET_VALUE(S32)
2706 IO_BUFFER_GET_VALUE(u64)
2707 IO_BUFFER_GET_VALUE(U64)
2708 IO_BUFFER_GET_VALUE(s64)
2709 IO_BUFFER_GET_VALUE(S64)
2711 IO_BUFFER_GET_VALUE(u128)
2712 IO_BUFFER_GET_VALUE(U128)
2713 IO_BUFFER_GET_VALUE(s128)
2714 IO_BUFFER_GET_VALUE(S128)
2716 IO_BUFFER_GET_VALUE(f32)
2717 IO_BUFFER_GET_VALUE(F32)
2718 IO_BUFFER_GET_VALUE(f64)
2719 IO_BUFFER_GET_VALUE(F64)
2720#undef IO_BUFFER_GET_VALUE
2722 rb_raise(rb_eArgError,
"Invalid type name!");
2768 size_t offset = io_buffer_extract_offset(_offset);
2770 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2772 return rb_io_buffer_get_value(base, size, TYPE_ID(
type), &offset);
2788 size_t offset = io_buffer_extract_offset(_offset);
2792 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2795 rb_raise(rb_eArgError,
"Argument buffer_types should be an array!");
2800 for (
long i = 0; i <
RARRAY_LEN(buffer_types); i++) {
2802 VALUE value = rb_io_buffer_get_value(base, size, TYPE_ID(
type), &offset);
2812io_buffer_extract_count(
VALUE argument)
2814 if (rb_int_negative_p(argument)) {
2815 rb_raise(rb_eArgError,
"Count can't be negative!");
2822io_buffer_extract_offset_count(
ID buffer_type,
size_t size,
int argc,
VALUE *argv,
size_t *offset,
size_t *count)
2825 *offset = io_buffer_extract_offset(argv[0]);
2832 *count = io_buffer_extract_count(argv[1]);
2835 if (*offset > size) {
2836 rb_raise(rb_eArgError,
"The given offset is bigger than the buffer size!");
2839 *count = (size - *offset) / io_buffer_buffer_type_size(buffer_type);
2850io_buffer_each_locked(
VALUE _arguments)
2853 VALUE self = arguments->self;
2854 int argc = arguments->argc;
2855 VALUE *argv = arguments->argv;
2860 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2864 buffer_type = TYPE_ID(argv[0]);
2867 buffer_type = RB_IO_BUFFER_DATA_TYPE_U8;
2870 size_t offset, count;
2871 io_buffer_extract_offset_count(buffer_type, size, argc-1, argv+1, &offset, &count);
2873 for (
size_t i = 0; i < count; i++) {
2874 size_t current_offset = offset;
2875 VALUE value = rb_io_buffer_get_value(base, size, buffer_type, &offset);
2899io_buffer_each(
int argc,
VALUE *argv,
VALUE self)
2909 rb_io_buffer_lock(self);
2910 return rb_ensure(io_buffer_each_locked, (
VALUE)&arguments, rb_io_buffer_locked_ensure, self);
2924io_buffer_values(
int argc,
VALUE *argv,
VALUE self)
2929 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2933 buffer_type = TYPE_ID(argv[0]);
2936 buffer_type = RB_IO_BUFFER_DATA_TYPE_U8;
2939 size_t offset, count;
2940 io_buffer_extract_offset_count(buffer_type, size, argc-1, argv+1, &offset, &count);
2944 for (
size_t i = 0; i < count; i++) {
2945 VALUE value = rb_io_buffer_get_value(base, size, buffer_type, &offset);
2953io_buffer_each_byte_locked(
VALUE _arguments)
2956 VALUE self = arguments->self;
2957 int argc = arguments->argc;
2958 VALUE *argv = arguments->argv;
2963 rb_io_buffer_get_bytes_for_reading(self, &base, &size);
2965 size_t offset, count;
2966 io_buffer_extract_offset_count(RB_IO_BUFFER_DATA_TYPE_U8, size, argc, argv, &offset, &count);
2968 if (size_sum_is_bigger_than(offset, count, size)) {
2969 rb_raise(rb_eArgError,
"Specified offset+count is bigger than the buffer size!");
2972 for (
size_t i = 0; i < count; i++) {
2973 unsigned char *value = (
unsigned char *)base + i + offset;
2996io_buffer_each_byte(
int argc,
VALUE *argv,
VALUE self)
3006 rb_io_buffer_lock(self);
3007 return rb_ensure(io_buffer_each_byte_locked, (
VALUE)&arguments, rb_io_buffer_locked_ensure, self);
3013 ID type = TYPE_ID(buffer_type);
3014#define IO_BUFFER_SET_VALUE(name) if (type == RB_IO_BUFFER_DATA_TYPE_##name) {io_buffer_write_##name(buffer, offset, value); return;}
3015 IO_BUFFER_SET_VALUE(U8);
3016 IO_BUFFER_SET_VALUE(S8);
3018 IO_BUFFER_SET_VALUE(u16);
3019 IO_BUFFER_SET_VALUE(U16);
3020 IO_BUFFER_SET_VALUE(s16);
3021 IO_BUFFER_SET_VALUE(S16);
3023 IO_BUFFER_SET_VALUE(u32);
3024 IO_BUFFER_SET_VALUE(U32);
3025 IO_BUFFER_SET_VALUE(s32);
3026 IO_BUFFER_SET_VALUE(S32);
3028 IO_BUFFER_SET_VALUE(u64);
3029 IO_BUFFER_SET_VALUE(U64);
3030 IO_BUFFER_SET_VALUE(s64);
3031 IO_BUFFER_SET_VALUE(S64);
3033 IO_BUFFER_SET_VALUE(u128);
3034 IO_BUFFER_SET_VALUE(U128);
3035 IO_BUFFER_SET_VALUE(s128);
3036 IO_BUFFER_SET_VALUE(S128);
3038 IO_BUFFER_SET_VALUE(f32);
3039 IO_BUFFER_SET_VALUE(F32);
3040 IO_BUFFER_SET_VALUE(f64);
3041 IO_BUFFER_SET_VALUE(F64);
3042#undef IO_BUFFER_SET_VALUE
3044 rb_raise(rb_eArgError,
"Invalid type name!");
3087 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
3088 size_t offset = io_buffer_extract_offset(_offset);
3089 rb_io_buffer_set_value(buffer,
type, &offset, value);
3112 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
3115 rb_raise(rb_eArgError,
"Argument buffer_types should be an array!");
3118 size_t offset = io_buffer_extract_offset(_offset);
3121 rb_raise(rb_eArgError,
"Argument values should be an array!");
3125 rb_raise(rb_eArgError,
"Argument buffer_types and values should have the same length!");
3128 for (
long i = 0; i <
RARRAY_LEN(buffer_types); i++) {
3131 rb_io_buffer_set_value(buffer,
type, &offset, value);
3137static size_t IO_BUFFER_BLOCKING_SIZE = 1024*1024;
3140 unsigned char * destination;
3141 const unsigned char * source;
3146io_buffer_memmove_blocking(
void *data)
3150 memmove(arguments->destination, arguments->source, arguments->length);
3156io_buffer_memmove_unblock(
void *data)
3162io_buffer_memmove(
void *base,
size_t size,
size_t offset,
const void *source_base,
size_t source_offset,
size_t source_size,
size_t length)
3164 if (size_sum_is_bigger_than(offset, length, size)) {
3165 rb_raise(rb_eArgError,
"Specified offset+length is bigger than the buffer size!");
3168 if (size_sum_is_bigger_than(source_offset, length, source_size)) {
3169 rb_raise(rb_eArgError,
"The computed source range exceeds the size of the source buffer!");
3172 if (length == 0)
return;
3177 .destination = (
unsigned char*)base+offset,
3178 .source = (
unsigned char*)source_base+source_offset,
3182 if (arguments.length >= IO_BUFFER_BLOCKING_SIZE) {
3184 }
else if (arguments.length != 0) {
3185 memmove(arguments.destination, arguments.source, arguments.length);
3190io_buffer_extract_copy_arguments(
size_t source_size,
int argc,
VALUE *argv,
size_t *offset,
size_t *length,
size_t *source_offset)
3194 *offset = io_buffer_extract_offset(argv[0]);
3202 *source_offset = io_buffer_extract_offset(argv[2]);
3204 if (*source_offset > source_size) {
3205 rb_raise(rb_eArgError,
"The given source offset is bigger than the source itself!");
3213 if (argc >= 2 && !
RB_NIL_P(argv[1])) {
3214 *length = io_buffer_extract_length(argv[1]);
3218 *length = source_size - *source_offset;
3224io_buffer_copy_from(
struct rb_io_buffer *buffer,
const void *source_base,
size_t source_size,
int argc,
VALUE *argv)
3226 size_t offset, length, source_offset;
3227 io_buffer_extract_copy_arguments(source_size, argc, argv, &offset, &length, &source_offset);
3231 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3233 if (length >= IO_BUFFER_BLOCKING_SIZE) {
3234 io_buffer_lock(buffer);
3237 io_buffer_memmove(base, size, offset, source_base, source_offset, source_size, length);
3239 if (length >= IO_BUFFER_BLOCKING_SIZE) {
3240 io_buffer_unlock(buffer);
3248 const void *source_base;
3259io_buffer_copy_to(
void *base,
size_t size,
VALUE _arguments)
3263 size_t offset, length, source_offset;
3264 io_buffer_extract_copy_arguments(arguments->source_size, arguments->argc, arguments->argv, &offset, &length, &source_offset);
3266 io_buffer_memmove(base, size, offset, arguments->source_base, source_offset, arguments->source_size, length);
3275io_buffer_copy_from_readable(
const void *base,
size_t size,
VALUE _arguments)
3279 arguments->source_base = base;
3280 arguments->source_size = size;
3282 return rb_io_buffer_locked_for_writing(arguments->destination, io_buffer_copy_to, _arguments);
3286io_buffer_initialize_copy_from(
const void *base,
size_t size,
VALUE self)
3290 io_buffer_initialize(self, buffer, NULL, size, io_flags_for_size(size),
Qnil);
3293 .destination = self,
3294 .source_base = base,
3295 .source_size = size,
3302 return rb_io_buffer_locked_for_writing(self, io_buffer_copy_to, (
VALUE)&arguments);
3323rb_io_buffer_initialize_copy(
VALUE self,
VALUE source)
3325 return rb_io_buffer_locked_for_reading(source, io_buffer_initialize_copy_from, self);
3397io_buffer_copy(
int argc,
VALUE *argv,
VALUE self)
3401 VALUE source = argv[0];
3403 .destination = self,
3413 return rb_io_buffer_locked_for_reading(source, io_buffer_copy_from_readable, (
VALUE)&arguments);
3423io_buffer_get_string_locked(
const void *base,
size_t size,
VALUE _arguments)
3426 if (size_sum_is_bigger_than(arguments->offset, arguments->length, size)) {
3427 rb_raise(rb_eArgError,
"Specified offset+length is bigger than the buffer size!");
3429 const char *data = base ? (
const char *)base + arguments->offset : NULL;
3430 return rb_enc_str_new(data, arguments->length, arguments->encoding);
3448io_buffer_get_string(
int argc,
VALUE *argv,
VALUE self)
3453 io_buffer_extract_offset_length(self, argc, argv, &arguments.offset, &arguments.length);
3457 arguments.encoding = argc >= 3 ? rb_find_encoding(argv[2]) : rb_ascii8bit_encoding();
3458 return rb_io_buffer_locked_for_reading(self, io_buffer_get_string_locked, (
VALUE)&arguments);
3485io_buffer_set_string(
int argc,
VALUE *argv,
VALUE self)
3489 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
3493 const void *source_base = RSTRING_PTR(
string);
3494 size_t source_size = RSTRING_LEN(
string);
3496 VALUE result = io_buffer_copy_from(buffer, source_base, source_size, argc-1, argv+1);
3502rb_io_buffer_clear(
VALUE self, uint8_t value,
size_t offset,
size_t length)
3504 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
3508 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3510 io_buffer_validate_range(buffer, offset, length);
3512 if (length == 0)
return;
3515 memset((
char*)base + offset, value, length);
3550io_buffer_clear(
int argc,
VALUE *argv,
VALUE self)
3559 size_t offset, length;
3560 io_buffer_extract_offset_length(self, argc-1, argv+1, &offset, &length);
3562 rb_io_buffer_clear(self, value, offset, length);
3568io_buffer_default_size(
size_t page_size)
3571 const size_t platform_agnostic_default_size = 64*1024;
3574 const char *default_size = getenv(
"RUBY_IO_BUFFER_DEFAULT_SIZE");
3577 int value = atoi(default_size);
3585 if (platform_agnostic_default_size < page_size) {
3589 return platform_agnostic_default_size;
3595 rb_blocking_function_t *function;
3600io_buffer_blocking_region_begin(
VALUE _argument)
3604 return rb_io_blocking_region(argument->io, argument->function, argument->data);
3608io_buffer_blocking_region_ensure(
VALUE _argument)
3612 io_buffer_unlock(argument->buffer);
3618io_buffer_blocking_region(
VALUE io,
struct rb_io_buffer *buffer, rb_blocking_function_t *function,
void *data)
3620 struct rb_io *ioptr;
3626 .function = function,
3633 io_buffer_lock(buffer);
3635 return rb_ensure(io_buffer_blocking_region_begin, (
VALUE)&argument, io_buffer_blocking_region_ensure, (
VALUE)&argument);
3648io_buffer_read_internal(
void *_argument)
3651 ssize_t result = read(argument->descriptor, argument->base, argument->length);
3657rb_io_buffer_read(
VALUE self,
VALUE io,
size_t offset,
size_t length)
3660 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
3662 io_buffer_validate_range(buffer, offset, length);
3667 if (scheduler !=
Qnil) {
3670 if (!UNDEF_P(result)) {
3675 buffer = get_io_buffer_for_writing(self);
3676 io_buffer_validate_range(buffer, offset, length);
3681 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3687 .base = (
char*)base + offset,
3691 return io_buffer_blocking_region(io, buffer, io_buffer_read_internal, &argument);
3718io_buffer_read(
int argc,
VALUE *argv,
VALUE self)
3724 size_t offset, length;
3725 io_buffer_extract_offset_length(self, argc-1, argv+1, &offset, &length);
3727 return rb_io_buffer_read(self, io, offset, length);
3742io_buffer_pread_internal(
void *_argument)
3745 ssize_t result = pread(argument->descriptor, argument->base, argument->length, argument->from);
3751rb_io_buffer_pread(
VALUE self,
VALUE io, rb_off_t from,
size_t offset,
size_t length)
3754 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
3756 io_buffer_validate_range(buffer, offset, length);
3761 if (scheduler !=
Qnil) {
3764 if (!UNDEF_P(result)) {
3768 buffer = get_io_buffer_for_writing(self);
3769 io_buffer_validate_range(buffer, offset, length);
3774 io_buffer_get_bytes_for_writing(buffer, &base, &size);
3780 .base = (
char*)base + offset,
3785 return io_buffer_blocking_region(io, buffer, io_buffer_pread_internal, &argument);
3816io_buffer_pread(
int argc,
VALUE *argv,
VALUE self)
3823 size_t offset, length;
3824 io_buffer_extract_offset_length(self, argc-2, argv+2, &offset, &length);
3826 return rb_io_buffer_pread(self, io, from, offset, length);
3839io_buffer_write_internal(
void *_argument)
3842 ssize_t result = write(argument->descriptor, argument->base, argument->length);
3848rb_io_buffer_write(
VALUE self,
VALUE io,
size_t offset,
size_t length)
3853 io_buffer_validate_range(buffer, offset, length);
3858 if (scheduler !=
Qnil) {
3861 if (!UNDEF_P(result)) {
3865 buffer = get_io_buffer(self);
3866 io_buffer_validate_range(buffer, offset, length);
3871 io_buffer_get_bytes_for_reading(buffer, &base, &size);
3877 .base = (
const char*)base + offset,
3881 return io_buffer_blocking_region(io, buffer, io_buffer_write_internal, &argument);
3901io_buffer_write(
int argc,
VALUE *argv,
VALUE self)
3907 size_t offset, length;
3908 io_buffer_extract_offset_length(self, argc-1, argv+1, &offset, &length);
3910 return rb_io_buffer_write(self, io, offset, length);
3925io_buffer_pwrite_internal(
void *_argument)
3928 ssize_t result = pwrite(argument->descriptor, argument->base, argument->length, argument->from);
3934rb_io_buffer_pwrite(
VALUE self,
VALUE io, rb_off_t from,
size_t offset,
size_t length)
3939 io_buffer_validate_range(buffer, offset, length);
3944 if (scheduler !=
Qnil) {
3947 if (!UNDEF_P(result)) {
3951 buffer = get_io_buffer(self);
3952 io_buffer_validate_range(buffer, offset, length);
3957 io_buffer_get_bytes_for_reading(buffer, &base, &size);
3963 .base = (
const char*)base + offset,
3968 return io_buffer_blocking_region(io, buffer, io_buffer_pwrite_internal, &argument);
3993io_buffer_pwrite(
int argc,
VALUE *argv,
VALUE self)
4000 size_t offset, length;
4001 io_buffer_extract_offset_length(self, argc-2, argv+2, &offset, &length);
4003 return rb_io_buffer_pwrite(self, io, from, offset, length);
4007io_buffer_check_mask_size(
size_t size)
4010 rb_raise(rb_eIOBufferMaskError,
"Zero-length mask given!");
4014memory_and(
unsigned char * restrict output,
const unsigned char * restrict base,
size_t size,
const unsigned char * restrict mask,
size_t mask_size)
4016 for (
size_t offset = 0; offset < size; offset += 1) {
4017 output[offset] = base[offset] & mask[offset % mask_size];
4038 struct rb_io_buffer *mask_buffer = get_io_buffer(mask);
4042 io_buffer_get_bytes_for_reading(buffer, &base, &size);
4044 const void *mask_base;
4046 io_buffer_get_bytes_for_reading(mask_buffer, &mask_base, &mask_size);
4048 io_buffer_check_mask_size(mask_size);
4050 VALUE output = rb_io_buffer_new(NULL, size, io_flags_for_size(size));
4051 struct rb_io_buffer *output_buffer = get_io_buffer(output);
4053 memory_and(output_buffer->base, base, size, mask_base, mask_size);
4059memory_or(
unsigned char * restrict output,
const unsigned char * restrict base,
size_t size,
const unsigned char * restrict mask,
size_t mask_size)
4061 for (
size_t offset = 0; offset < size; offset += 1) {
4062 output[offset] = base[offset] | mask[offset % mask_size];
4083 struct rb_io_buffer *mask_buffer = get_io_buffer(mask);
4087 io_buffer_get_bytes_for_reading(buffer, &base, &size);
4089 const void *mask_base;
4091 io_buffer_get_bytes_for_reading(mask_buffer, &mask_base, &mask_size);
4093 io_buffer_check_mask_size(mask_size);
4095 VALUE output = rb_io_buffer_new(NULL, size, io_flags_for_size(size));
4096 struct rb_io_buffer *output_buffer = get_io_buffer(output);
4098 memory_or(output_buffer->base, base, size, mask_base, mask_size);
4104memory_xor(
unsigned char * restrict output,
const unsigned char * restrict base,
size_t size,
const unsigned char * restrict mask,
size_t mask_size)
4106 for (
size_t offset = 0; offset < size; offset += 1) {
4107 output[offset] = base[offset] ^ mask[offset % mask_size];
4128 struct rb_io_buffer *mask_buffer = get_io_buffer(mask);
4132 io_buffer_get_bytes_for_reading(buffer, &base, &size);
4134 const void *mask_base;
4136 io_buffer_get_bytes_for_reading(mask_buffer, &mask_base, &mask_size);
4138 io_buffer_check_mask_size(mask_size);
4140 VALUE output = rb_io_buffer_new(NULL, size, io_flags_for_size(size));
4141 struct rb_io_buffer *output_buffer = get_io_buffer(output);
4143 memory_xor(output_buffer->base, base, size, mask_base, mask_size);
4149memory_not(
unsigned char * restrict output,
const unsigned char * restrict base,
size_t size)
4151 for (
size_t offset = 0; offset < size; offset += 1) {
4152 output[offset] = ~base[offset];
4169io_buffer_not(
VALUE self)
4175 io_buffer_get_bytes_for_reading(buffer, &base, &size);
4177 VALUE output = rb_io_buffer_new(NULL, size, io_flags_for_size(size));
4178 struct rb_io_buffer *output_buffer = get_io_buffer(output);
4180 memory_not(output_buffer->base, base, size);
4189 void *a_base = NULL, *b_base = NULL;
4190 size_t a_size = 0, b_size = 0;
4191 if (!io_buffer_try_get_bytes(a, &a_base, &a_size))
return 0;
4192 if (!io_buffer_try_get_bytes(b, &b_base, &b_size))
return 0;
4194 if (a_size == 0 || b_size == 0 || a_base == NULL || b_base == NULL)
return 0;
4198 uintptr_t a_start = (uintptr_t)a_base;
4199 uintptr_t b_start = (uintptr_t)b_base;
4200 if (a_start <= b_start)
return b_start - a_start < a_size;
4201 return a_start - b_start < b_size;
4207 if (io_buffer_overlaps(a, b))
4208 rb_raise(rb_eIOBufferMaskError,
"Mask overlaps source buffer!");
4212memory_and_inplace(
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
4214 for (
size_t offset = 0; offset < size; offset += 1) {
4215 base[offset] &= mask[offset % mask_size];
4239 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
4241 struct rb_io_buffer *mask_buffer = get_io_buffer(mask);
4243 io_buffer_check_mask_size(mask_buffer->size);
4244 io_buffer_check_overlaps(buffer, mask_buffer);
4248 io_buffer_get_bytes_for_writing(buffer, &base, &size);
4250 const void *mask_base;
4252 io_buffer_get_bytes_for_reading(mask_buffer, &mask_base, &mask_size);
4254 memory_and_inplace(base, size, (
unsigned char *)mask_base, mask_size);
4260memory_or_inplace(
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
4262 for (
size_t offset = 0; offset < size; offset += 1) {
4263 base[offset] |= mask[offset % mask_size];
4287 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
4289 struct rb_io_buffer *mask_buffer = get_io_buffer(mask);
4291 io_buffer_check_mask_size(mask_buffer->size);
4292 io_buffer_check_overlaps(buffer, mask_buffer);
4296 io_buffer_get_bytes_for_writing(buffer, &base, &size);
4298 const void *mask_base;
4300 io_buffer_get_bytes_for_reading(mask_buffer, &mask_base, &mask_size);
4302 memory_or_inplace(base, size, (
unsigned char *)mask_base, mask_size);
4308memory_xor_inplace(
unsigned char * restrict base,
size_t size,
unsigned char * restrict mask,
size_t mask_size)
4310 for (
size_t offset = 0; offset < size; offset += 1) {
4311 base[offset] ^= mask[offset % mask_size];
4335 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
4337 struct rb_io_buffer *mask_buffer = get_io_buffer(mask);
4339 io_buffer_check_mask_size(mask_buffer->size);
4340 io_buffer_check_overlaps(buffer, mask_buffer);
4344 io_buffer_get_bytes_for_writing(buffer, &base, &size);
4346 const void *mask_base;
4348 io_buffer_get_bytes_for_reading(mask_buffer, &mask_base, &mask_size);
4350 memory_xor_inplace(base, size, (
unsigned char *)mask_base, mask_size);
4356memory_not_inplace(
unsigned char * restrict base,
size_t size)
4358 for (
size_t offset = 0; offset < size; offset += 1) {
4359 base[offset] = ~base[offset];
4381io_buffer_not_inplace(
VALUE self)
4383 struct rb_io_buffer *buffer = get_io_buffer_for_writing(self);
4387 io_buffer_get_bytes_for_writing(buffer, &base, &size);
4389 memory_not_inplace(base, size);
4395memory_bit_count(
const unsigned char *base,
size_t size)
4400 const uint64_t *base64 = (
const uint64_t *)base;
4401 size_t count64 = size / 8;
4402 for (
size_t i = 0; i < count64; i += 1) {
4403 count += rb_popcount64(base64[i]);
4407 size_t remaining = size % 8;
4408 const unsigned char *tail = base + (count64 * 8);
4409 for (
size_t i = 0; i < remaining; i += 1) {
4410 count += rb_popcount32(tail[i]);
4430io_buffer_bit_count(
int argc,
VALUE *argv,
VALUE self)
4434 size_t offset, length;
4435 struct rb_io_buffer *buffer = io_buffer_extract_offset_length(self, argc, argv, &offset, &length);
4437 io_buffer_validate_range(buffer, offset, length);
4441 io_buffer_get_bytes_for_reading(buffer, &base, &size);
4446 size_t count = memory_bit_count((
const unsigned char *)base + offset, length);
4458 if (!io_buffer_try_get_bytes(buffer, &base, &size) || base == NULL) {
4462 bool readonly =
true;
4463 if (flags & RUBY_MEMORY_VIEW_WRITABLE) {
4464 if (io_buffer_readonly_p(buffer)) {
4471 if (flags & RUBY_MEMORY_VIEW_FORMAT) {
4474 bool request_multi_dimensional = flags & RUBY_MEMORY_VIEW_MULTI_DIMENSIONAL;
4475 bool request_strides =
4476 (flags & RUBY_MEMORY_VIEW_STRIDES) == RUBY_MEMORY_VIEW_STRIDES;
4477 if (request_multi_dimensional || request_strides) {
4478 size_t n_metadata = 0;
4479 if (request_multi_dimensional)
4481 if (request_strides)
4483 ssize_t *metadata_buffer =
ALLOC_N(ssize_t, n_metadata);
4485 if (request_multi_dimensional) {
4486 ssize_t *shape = &metadata_buffer[i];
4487 shape[0] = buffer->size;
4488 view->
shape = shape;
4491 if (request_strides) {
4492 ssize_t *strides = &metadata_buffer[i];
4499 io_buffer_lock(buffer);
4507 rb_io_buffer_unlock(self);
4515io_buffer_memory_view_available_p(
VALUE self)
4521 return io_buffer_try_get_bytes(buffer, &base, &size) && base != NULL;
4525 .
get_func = io_buffer_memory_view_get,
4526 .release_func = io_buffer_memory_view_release,
4527 .available_p_func = io_buffer_memory_view_available_p,
4633 rb_eIOBufferLockedError = rb_define_class_under(rb_cIOBuffer,
"LockedError",
rb_eRuntimeError);
4636 rb_eIOBufferAllocationError = rb_define_class_under(rb_cIOBuffer,
"AllocationError",
rb_eRuntimeError);
4639 rb_eIOBufferAccessError = rb_define_class_under(rb_cIOBuffer,
"AccessError",
rb_eRuntimeError);
4642 rb_eIOBufferInvalidatedError = rb_define_class_under(rb_cIOBuffer,
"InvalidatedError",
rb_eRuntimeError);
4645 rb_eIOBufferMaskError = rb_define_class_under(rb_cIOBuffer,
"MaskError", rb_eArgError);
4653 GetSystemInfo(&info);
4654 RUBY_IO_BUFFER_PAGE_SIZE = info.dwPageSize;
4655 RUBY_IO_BUFFER_MAP_ALIGNMENT = info.dwAllocationGranularity;
4657 RUBY_IO_BUFFER_PAGE_SIZE = sysconf(_SC_PAGESIZE);
4658 RUBY_IO_BUFFER_MAP_ALIGNMENT = RUBY_IO_BUFFER_PAGE_SIZE;
4661 RUBY_IO_BUFFER_DEFAULT_SIZE = io_buffer_default_size(RUBY_IO_BUFFER_PAGE_SIZE);
4664 rb_define_const(rb_cIOBuffer,
"VERSION",
INT2NUM(RUBY_IO_BUFFER_VERSION));
4667 rb_define_const(rb_cIOBuffer,
"PAGE_SIZE",
SIZET2NUM(RUBY_IO_BUFFER_PAGE_SIZE));
4670 rb_define_const(rb_cIOBuffer,
"MAP_ALIGNMENT",
SIZET2NUM(RUBY_IO_BUFFER_MAP_ALIGNMENT));
4675 rb_define_const(rb_cIOBuffer,
"DEFAULT_SIZE",
SIZET2NUM(RUBY_IO_BUFFER_DEFAULT_SIZE));
4680 rb_define_method(rb_cIOBuffer,
"initialize_copy", rb_io_buffer_initialize_copy, 1);
4691 rb_define_const(rb_cIOBuffer,
"EXTERNAL",
RB_INT2NUM(RB_IO_BUFFER_EXTERNAL));
4694 rb_define_const(rb_cIOBuffer,
"INTERNAL",
RB_INT2NUM(RB_IO_BUFFER_INTERNAL));
4697 rb_define_const(rb_cIOBuffer,
"MAPPED",
RB_INT2NUM(RB_IO_BUFFER_MAPPED));
4700 rb_define_const(rb_cIOBuffer,
"SHARED",
RB_INT2NUM(RB_IO_BUFFER_SHARED));
4703 rb_define_const(rb_cIOBuffer,
"PRIVATE",
RB_INT2NUM(RB_IO_BUFFER_PRIVATE));
4706 rb_define_const(rb_cIOBuffer,
"READONLY",
RB_INT2NUM(RB_IO_BUFFER_READONLY));
4709 rb_define_const(rb_cIOBuffer,
"LITTLE_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_LITTLE_ENDIAN));
4712 rb_define_const(rb_cIOBuffer,
"BIG_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_BIG_ENDIAN));
4715 rb_define_const(rb_cIOBuffer,
"HOST_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_HOST_ENDIAN));
4718 rb_define_const(rb_cIOBuffer,
"NETWORK_ENDIAN",
RB_INT2NUM(RB_IO_BUFFER_NETWORK_ENDIAN));
4745#define IO_BUFFER_DEFINE_DATA_TYPE(name) RB_IO_BUFFER_DATA_TYPE_##name = rb_intern_const(#name)
4746 IO_BUFFER_DEFINE_DATA_TYPE(U8);
4747 IO_BUFFER_DEFINE_DATA_TYPE(S8);
4749 IO_BUFFER_DEFINE_DATA_TYPE(u16);
4750 IO_BUFFER_DEFINE_DATA_TYPE(U16);
4751 IO_BUFFER_DEFINE_DATA_TYPE(s16);
4752 IO_BUFFER_DEFINE_DATA_TYPE(S16);
4754 IO_BUFFER_DEFINE_DATA_TYPE(u32);
4755 IO_BUFFER_DEFINE_DATA_TYPE(U32);
4756 IO_BUFFER_DEFINE_DATA_TYPE(s32);
4757 IO_BUFFER_DEFINE_DATA_TYPE(S32);
4759 IO_BUFFER_DEFINE_DATA_TYPE(u64);
4760 IO_BUFFER_DEFINE_DATA_TYPE(U64);
4761 IO_BUFFER_DEFINE_DATA_TYPE(s64);
4762 IO_BUFFER_DEFINE_DATA_TYPE(S64);
4764 IO_BUFFER_DEFINE_DATA_TYPE(u128);
4765 IO_BUFFER_DEFINE_DATA_TYPE(U128);
4766 IO_BUFFER_DEFINE_DATA_TYPE(s128);
4767 IO_BUFFER_DEFINE_DATA_TYPE(S128);
4769 IO_BUFFER_DEFINE_DATA_TYPE(f32);
4770 IO_BUFFER_DEFINE_DATA_TYPE(F32);
4771 IO_BUFFER_DEFINE_DATA_TYPE(f64);
4772 IO_BUFFER_DEFINE_DATA_TYPE(F64);
4773#undef IO_BUFFER_DEFINE_DATA_TYPE
#define RUBY_ASSERT(...)
Asserts that the given expression is truthy if and only if RUBY_DEBUG is truthy.
#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.
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 xfree
Old name of ruby_xfree.
#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 STATIC_SYM_P
Old name of RB_STATIC_SYM_P.
#define ALLOC_N
Old name of RB_ALLOC_N.
#define NUM2DBL
Old name of rb_num2dbl.
#define INT2NUM
Old name of RB_INT2NUM.
#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 T_SYMBOL
Old name of RUBY_T_SYMBOL.
#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_eTypeError
TypeError exception.
VALUE rb_eRuntimeError
RuntimeError exception.
@ RB_WARN_CATEGORY_EXPERIMENTAL
Warning is for experimental features.
VALUE rb_cObject
Object class.
static VALUE rb_class_of(VALUE obj)
Object to class mapping function.
VALUE rb_obj_class(VALUE obj)
Queries the class of an object.
VALUE rb_obj_is_kind_of(VALUE obj, VALUE klass)
Queries if the given object is an instance (of possibly descendants) of the given class.
VALUE rb_mComparable
Comparable module.
#define RB_OBJ_WRITE(old, slot, young)
Declaration of a "back" pointer.
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_read(VALUE scheduler, VALUE io, VALUE buffer, size_t offset, size_t length)
Non-blocking read from the passed IO.
VALUE rb_fiber_scheduler_io_pwrite(VALUE scheduler, VALUE io, rb_off_t from, VALUE buffer, size_t offset, size_t length)
Non-blocking write to the passed IO at the specified offset.
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 offset, size_t length)
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 offset, size_t length)
Non-blocking write to the passed IO.
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_new_frozen(VALUE str)
Creates a frozen copy of the string, if necessary.
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.
ID rb_sym2id(VALUE obj)
Converts an instance of rb_cSymbol into an ID.
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.
#define RB_GC_GUARD(v)
Prevents premature destruction of local objects.
bool rb_memory_view_register(VALUE klass, const rb_memory_view_entry_t *entry)
Associates the passed class with the passed memory view entry.
bool rb_memory_view_init_as_byte_array(rb_memory_view_t *view, VALUE obj, void *data, const ssize_t len, const bool readonly)
Fill the members of view as an 1-dimensional byte array.
VALUE type(ANYARGS)
ANYARGS-ed function type.
VALUE rb_ensure(type *q, VALUE w, type *e, VALUE r)
An equivalent of ensure clause.
#define OFFT2NUM
Converts a C's off_t into an instance of rb_cInteger.
#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.
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.
Operations applied to a specific kind of a memory view.
rb_memory_view_get_func_t get_func
Exports a memory view from a Ruby object.
A MemoryView structure, rb_memory_view_t, is used for exporting objects' MemoryView.
const ssize_t * strides
ndim size array indicating the number of bytes to skip to go to the next element in each dimension.
const ssize_t * shape
ndim size array indicating the number of elements in each dimension.
void * private_data
The private data for managing this exported memory.
const char * format
A string to describe the format of an element, or NULL for unsigned bytes.
uintptr_t ID
Type that represents a Ruby identifier such as a variable name.
uintptr_t VALUE
Type that represents a Ruby object.
static void Check_Type(VALUE v, enum ruby_value_type t)
Identical to RB_TYPE_P(), except it raises exceptions on predication failure.
static bool RB_TYPE_P(VALUE obj, enum ruby_value_type t)
Queries if the given object is of given type.