xref: /unit/src/ruby/nxt_ruby_stream_io.c (revision 1235)
1584Salexander.borisov@nginx.com 
2584Salexander.borisov@nginx.com /*
3584Salexander.borisov@nginx.com  * Copyright (C) Alexander Borisov
4584Salexander.borisov@nginx.com  * Copyright (C) NGINX, Inc.
5584Salexander.borisov@nginx.com  */
6584Salexander.borisov@nginx.com 
7584Salexander.borisov@nginx.com #include <ruby/nxt_ruby.h>
8743Smax.romanov@nginx.com #include <nxt_unit.h>
9584Salexander.borisov@nginx.com 
10584Salexander.borisov@nginx.com 
11584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_new(VALUE class, VALUE wrap);
12584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_initialize(int argc, VALUE *argv, VALUE self);
13584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_gets(VALUE obj, VALUE args);
14584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_each(VALUE obj, VALUE args);
15584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_read(VALUE obj, VALUE args);
16584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_rewind(VALUE obj, VALUE args);
17584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_puts(VALUE obj, VALUE args);
18584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_write(VALUE obj, VALUE args);
19584Salexander.borisov@nginx.com nxt_inline long nxt_ruby_stream_io_s_write(nxt_ruby_run_ctx_t *run_ctx,
20584Salexander.borisov@nginx.com     VALUE val);
21584Salexander.borisov@nginx.com static VALUE nxt_ruby_stream_io_flush(VALUE obj, VALUE args);
22584Salexander.borisov@nginx.com 
23584Salexander.borisov@nginx.com 
24584Salexander.borisov@nginx.com VALUE
25584Salexander.borisov@nginx.com nxt_ruby_stream_io_input_init(void)
26584Salexander.borisov@nginx.com {
27584Salexander.borisov@nginx.com     VALUE  stream_io;
28584Salexander.borisov@nginx.com 
29584Salexander.borisov@nginx.com     stream_io = rb_define_class("NGINX_Unit_Stream_IO_Read", rb_cData);
30584Salexander.borisov@nginx.com 
31584Salexander.borisov@nginx.com     rb_gc_register_address(&stream_io);
32584Salexander.borisov@nginx.com 
33584Salexander.borisov@nginx.com     rb_define_singleton_method(stream_io, "new", nxt_ruby_stream_io_new, 1);
34*1235Sigor@sysoev.ru     rb_define_method(stream_io, "initialize",
35*1235Sigor@sysoev.ru                      nxt_ruby_stream_io_initialize, -1);
36584Salexander.borisov@nginx.com     rb_define_method(stream_io, "gets", nxt_ruby_stream_io_gets, 0);
37584Salexander.borisov@nginx.com     rb_define_method(stream_io, "each", nxt_ruby_stream_io_each, 0);
38584Salexander.borisov@nginx.com     rb_define_method(stream_io, "read", nxt_ruby_stream_io_read, -2);
39584Salexander.borisov@nginx.com     rb_define_method(stream_io, "rewind", nxt_ruby_stream_io_rewind, 0);
40584Salexander.borisov@nginx.com 
41584Salexander.borisov@nginx.com     return stream_io;
42584Salexander.borisov@nginx.com }
43584Salexander.borisov@nginx.com 
44584Salexander.borisov@nginx.com 
45584Salexander.borisov@nginx.com VALUE
46584Salexander.borisov@nginx.com nxt_ruby_stream_io_error_init(void)
47584Salexander.borisov@nginx.com {
48584Salexander.borisov@nginx.com     VALUE  stream_io;
49584Salexander.borisov@nginx.com 
50584Salexander.borisov@nginx.com     stream_io = rb_define_class("NGINX_Unit_Stream_IO_Error", rb_cData);
51584Salexander.borisov@nginx.com 
52584Salexander.borisov@nginx.com     rb_gc_register_address(&stream_io);
53584Salexander.borisov@nginx.com 
54584Salexander.borisov@nginx.com     rb_define_singleton_method(stream_io, "new", nxt_ruby_stream_io_new, 1);
55*1235Sigor@sysoev.ru     rb_define_method(stream_io, "initialize",
56*1235Sigor@sysoev.ru                      nxt_ruby_stream_io_initialize, -1);
57584Salexander.borisov@nginx.com     rb_define_method(stream_io, "puts", nxt_ruby_stream_io_puts, -2);
58584Salexander.borisov@nginx.com     rb_define_method(stream_io, "write", nxt_ruby_stream_io_write, -2);
59584Salexander.borisov@nginx.com     rb_define_method(stream_io, "flush", nxt_ruby_stream_io_flush, 0);
60584Salexander.borisov@nginx.com 
61584Salexander.borisov@nginx.com     return stream_io;
62584Salexander.borisov@nginx.com }
63584Salexander.borisov@nginx.com 
64584Salexander.borisov@nginx.com 
65584Salexander.borisov@nginx.com static VALUE
66584Salexander.borisov@nginx.com nxt_ruby_stream_io_new(VALUE class, VALUE wrap)
67584Salexander.borisov@nginx.com {
68584Salexander.borisov@nginx.com     VALUE               self;
69584Salexander.borisov@nginx.com     nxt_ruby_run_ctx_t  *run_ctx;
70584Salexander.borisov@nginx.com 
71584Salexander.borisov@nginx.com     Data_Get_Struct(wrap, nxt_ruby_run_ctx_t, run_ctx);
72584Salexander.borisov@nginx.com     self = Data_Wrap_Struct(class, 0, 0, run_ctx);
73584Salexander.borisov@nginx.com 
74584Salexander.borisov@nginx.com     rb_obj_call_init(self, 0, NULL);
75584Salexander.borisov@nginx.com 
76584Salexander.borisov@nginx.com     return self;
77584Salexander.borisov@nginx.com }
78584Salexander.borisov@nginx.com 
79584Salexander.borisov@nginx.com 
80584Salexander.borisov@nginx.com static VALUE
81584Salexander.borisov@nginx.com nxt_ruby_stream_io_initialize(int argc, VALUE *argv, VALUE self)
82584Salexander.borisov@nginx.com {
83584Salexander.borisov@nginx.com     return self;
84584Salexander.borisov@nginx.com }
85584Salexander.borisov@nginx.com 
86584Salexander.borisov@nginx.com 
87584Salexander.borisov@nginx.com static VALUE
88584Salexander.borisov@nginx.com nxt_ruby_stream_io_gets(VALUE obj, VALUE args)
89584Salexander.borisov@nginx.com {
90743Smax.romanov@nginx.com     VALUE                    buf;
91743Smax.romanov@nginx.com     char                     *p;
92743Smax.romanov@nginx.com     size_t                   size, b_size;
93743Smax.romanov@nginx.com     nxt_unit_buf_t           *b;
94743Smax.romanov@nginx.com     nxt_ruby_run_ctx_t       *run_ctx;
95743Smax.romanov@nginx.com     nxt_unit_request_info_t  *req;
96584Salexander.borisov@nginx.com 
97584Salexander.borisov@nginx.com     Data_Get_Struct(obj, nxt_ruby_run_ctx_t, run_ctx);
98584Salexander.borisov@nginx.com 
99743Smax.romanov@nginx.com     req = run_ctx->req;
100743Smax.romanov@nginx.com 
101743Smax.romanov@nginx.com     if (req->content_length == 0) {
102584Salexander.borisov@nginx.com         return Qnil;
103584Salexander.borisov@nginx.com     }
104584Salexander.borisov@nginx.com 
105743Smax.romanov@nginx.com     size = 0;
106743Smax.romanov@nginx.com 
107743Smax.romanov@nginx.com     for (b = req->content_buf; b; b = nxt_unit_buf_next(b)) {
108743Smax.romanov@nginx.com         b_size = b->end - b->free;
109743Smax.romanov@nginx.com         p = memchr(b->free, '\n', b_size);
110743Smax.romanov@nginx.com 
111743Smax.romanov@nginx.com         if (p != NULL) {
112743Smax.romanov@nginx.com             p++;
113743Smax.romanov@nginx.com             size += p - b->free;
114743Smax.romanov@nginx.com             break;
115743Smax.romanov@nginx.com         }
116743Smax.romanov@nginx.com 
117743Smax.romanov@nginx.com         size += b_size;
118743Smax.romanov@nginx.com     }
119743Smax.romanov@nginx.com 
120743Smax.romanov@nginx.com     buf = rb_str_buf_new(size);
121584Salexander.borisov@nginx.com 
122584Salexander.borisov@nginx.com     if (buf == Qnil) {
123584Salexander.borisov@nginx.com         return Qnil;
124584Salexander.borisov@nginx.com     }
125584Salexander.borisov@nginx.com 
126743Smax.romanov@nginx.com     size = nxt_unit_request_read(req, RSTRING_PTR(buf), size);
127743Smax.romanov@nginx.com 
128743Smax.romanov@nginx.com     rb_str_set_len(buf, size);
129584Salexander.borisov@nginx.com 
130584Salexander.borisov@nginx.com     return buf;
131584Salexander.borisov@nginx.com }
132584Salexander.borisov@nginx.com 
133584Salexander.borisov@nginx.com 
134584Salexander.borisov@nginx.com static VALUE
135584Salexander.borisov@nginx.com nxt_ruby_stream_io_each(VALUE obj, VALUE args)
136584Salexander.borisov@nginx.com {
137584Salexander.borisov@nginx.com     VALUE  chunk;
138584Salexander.borisov@nginx.com 
139584Salexander.borisov@nginx.com     if (rb_block_given_p() == 0) {
140584Salexander.borisov@nginx.com         rb_raise(rb_eArgError, "Expected block on rack.input 'each' method");
141584Salexander.borisov@nginx.com     }
142584Salexander.borisov@nginx.com 
143584Salexander.borisov@nginx.com     for ( ;; ) {
144584Salexander.borisov@nginx.com         chunk = nxt_ruby_stream_io_gets(obj, Qnil);
145584Salexander.borisov@nginx.com 
146584Salexander.borisov@nginx.com         if (chunk == Qnil) {
147584Salexander.borisov@nginx.com             return Qnil;
148584Salexander.borisov@nginx.com         }
149584Salexander.borisov@nginx.com 
150584Salexander.borisov@nginx.com         rb_yield(chunk);
151584Salexander.borisov@nginx.com     }
152584Salexander.borisov@nginx.com 
153584Salexander.borisov@nginx.com     return Qnil;
154584Salexander.borisov@nginx.com }
155584Salexander.borisov@nginx.com 
156584Salexander.borisov@nginx.com 
157584Salexander.borisov@nginx.com static VALUE
158584Salexander.borisov@nginx.com nxt_ruby_stream_io_read(VALUE obj, VALUE args)
159584Salexander.borisov@nginx.com {
160584Salexander.borisov@nginx.com     VALUE                buf;
161584Salexander.borisov@nginx.com     long                 copy_size, u_size;
162584Salexander.borisov@nginx.com     nxt_ruby_run_ctx_t  *run_ctx;
163584Salexander.borisov@nginx.com 
164584Salexander.borisov@nginx.com     Data_Get_Struct(obj, nxt_ruby_run_ctx_t, run_ctx);
165584Salexander.borisov@nginx.com 
166743Smax.romanov@nginx.com     copy_size = run_ctx->req->content_length;
167584Salexander.borisov@nginx.com 
168584Salexander.borisov@nginx.com     if (RARRAY_LEN(args) > 0 && TYPE(RARRAY_PTR(args)[0]) == T_FIXNUM) {
169584Salexander.borisov@nginx.com         u_size = NUM2LONG(RARRAY_PTR(args)[0]);
170584Salexander.borisov@nginx.com 
171584Salexander.borisov@nginx.com         if (u_size < 0 || copy_size == 0) {
172584Salexander.borisov@nginx.com             return Qnil;
173584Salexander.borisov@nginx.com         }
174584Salexander.borisov@nginx.com 
175584Salexander.borisov@nginx.com         if (copy_size > u_size) {
176584Salexander.borisov@nginx.com             copy_size = u_size;
177584Salexander.borisov@nginx.com         }
178584Salexander.borisov@nginx.com     }
179584Salexander.borisov@nginx.com 
180584Salexander.borisov@nginx.com     if (copy_size == 0) {
181584Salexander.borisov@nginx.com         return rb_str_new_cstr("");
182584Salexander.borisov@nginx.com     }
183584Salexander.borisov@nginx.com 
184584Salexander.borisov@nginx.com     buf = rb_str_buf_new(copy_size);
185584Salexander.borisov@nginx.com 
186584Salexander.borisov@nginx.com     if (nxt_slow_path(buf == Qnil)) {
187584Salexander.borisov@nginx.com         return Qnil;
188584Salexander.borisov@nginx.com     }
189584Salexander.borisov@nginx.com 
190743Smax.romanov@nginx.com     copy_size = nxt_unit_request_read(run_ctx->req, RSTRING_PTR(buf),
191743Smax.romanov@nginx.com                                       copy_size);
192584Salexander.borisov@nginx.com 
193584Salexander.borisov@nginx.com     if (RARRAY_LEN(args) > 1 && TYPE(RARRAY_PTR(args)[1]) == T_STRING) {
194584Salexander.borisov@nginx.com 
195584Salexander.borisov@nginx.com         rb_str_set_len(RARRAY_PTR(args)[1], 0);
196584Salexander.borisov@nginx.com         rb_str_cat(RARRAY_PTR(args)[1], RSTRING_PTR(buf), copy_size);
197584Salexander.borisov@nginx.com     }
198584Salexander.borisov@nginx.com 
199743Smax.romanov@nginx.com     rb_str_set_len(buf, copy_size);
200584Salexander.borisov@nginx.com 
201584Salexander.borisov@nginx.com     return buf;
202584Salexander.borisov@nginx.com }
203584Salexander.borisov@nginx.com 
204584Salexander.borisov@nginx.com 
205584Salexander.borisov@nginx.com static VALUE
206584Salexander.borisov@nginx.com nxt_ruby_stream_io_rewind(VALUE obj, VALUE args)
207584Salexander.borisov@nginx.com {
208584Salexander.borisov@nginx.com     return Qnil;
209584Salexander.borisov@nginx.com }
210584Salexander.borisov@nginx.com 
211584Salexander.borisov@nginx.com 
212584Salexander.borisov@nginx.com static VALUE
213584Salexander.borisov@nginx.com nxt_ruby_stream_io_puts(VALUE obj, VALUE args)
214584Salexander.borisov@nginx.com {
215584Salexander.borisov@nginx.com     nxt_ruby_run_ctx_t  *run_ctx;
216584Salexander.borisov@nginx.com 
217584Salexander.borisov@nginx.com     if (RARRAY_LEN(args) != 1) {
218584Salexander.borisov@nginx.com         return Qnil;
219584Salexander.borisov@nginx.com     }
220584Salexander.borisov@nginx.com 
221584Salexander.borisov@nginx.com     Data_Get_Struct(obj, nxt_ruby_run_ctx_t, run_ctx);
222584Salexander.borisov@nginx.com 
223584Salexander.borisov@nginx.com     nxt_ruby_stream_io_s_write(run_ctx, RARRAY_PTR(args)[0]);
224584Salexander.borisov@nginx.com 
225584Salexander.borisov@nginx.com     return Qnil;
226584Salexander.borisov@nginx.com }
227584Salexander.borisov@nginx.com 
228584Salexander.borisov@nginx.com 
229584Salexander.borisov@nginx.com static VALUE
230584Salexander.borisov@nginx.com nxt_ruby_stream_io_write(VALUE obj, VALUE args)
231584Salexander.borisov@nginx.com {
232584Salexander.borisov@nginx.com     long                len;
233584Salexander.borisov@nginx.com     nxt_ruby_run_ctx_t  *run_ctx;
234584Salexander.borisov@nginx.com 
235584Salexander.borisov@nginx.com     if (RARRAY_LEN(args) != 1) {
236584Salexander.borisov@nginx.com         return Qnil;
237584Salexander.borisov@nginx.com     }
238584Salexander.borisov@nginx.com 
239584Salexander.borisov@nginx.com     Data_Get_Struct(obj, nxt_ruby_run_ctx_t, run_ctx);
240584Salexander.borisov@nginx.com 
241584Salexander.borisov@nginx.com     len = nxt_ruby_stream_io_s_write(run_ctx, RARRAY_PTR(args)[0]);
242584Salexander.borisov@nginx.com 
243584Salexander.borisov@nginx.com     return LONG2FIX(len);
244584Salexander.borisov@nginx.com }
245584Salexander.borisov@nginx.com 
246584Salexander.borisov@nginx.com 
247584Salexander.borisov@nginx.com nxt_inline long
248584Salexander.borisov@nginx.com nxt_ruby_stream_io_s_write(nxt_ruby_run_ctx_t *run_ctx, VALUE val)
249584Salexander.borisov@nginx.com {
250584Salexander.borisov@nginx.com     if (nxt_slow_path(val == Qnil)) {
251584Salexander.borisov@nginx.com         return 0;
252584Salexander.borisov@nginx.com     }
253584Salexander.borisov@nginx.com 
254584Salexander.borisov@nginx.com     if (TYPE(val) != T_STRING) {
255584Salexander.borisov@nginx.com         val = rb_funcall(val, rb_intern("to_s"), 0);
256584Salexander.borisov@nginx.com 
257584Salexander.borisov@nginx.com         if (TYPE(val) != T_STRING) {
258584Salexander.borisov@nginx.com             return 0;
259584Salexander.borisov@nginx.com         }
260584Salexander.borisov@nginx.com     }
261584Salexander.borisov@nginx.com 
262743Smax.romanov@nginx.com     nxt_unit_req_error(run_ctx->req, "Ruby: %s", RSTRING_PTR(val));
263584Salexander.borisov@nginx.com 
264584Salexander.borisov@nginx.com     return RSTRING_LEN(val);
265584Salexander.borisov@nginx.com }
266584Salexander.borisov@nginx.com 
267584Salexander.borisov@nginx.com 
268584Salexander.borisov@nginx.com static VALUE
269584Salexander.borisov@nginx.com nxt_ruby_stream_io_flush(VALUE obj, VALUE args)
270584Salexander.borisov@nginx.com {
271584Salexander.borisov@nginx.com     return Qnil;
272584Salexander.borisov@nginx.com }
273