00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011 #include "rubysocket.h"
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00023
00024 static VALUE
00025 bsock_s_for_fd(VALUE klass, VALUE fd)
00026 {
00027 rb_io_t *fptr;
00028 VALUE sock = rsock_init_sock(rb_obj_alloc(klass), NUM2INT(fd));
00029
00030 GetOpenFile(sock, fptr);
00031
00032 return sock;
00033 }
00034
00035
00036
00037
00038
00039
00040
00041
00042
00043
00044
00045
00046
00047
00048
00049
00050
00051
00052
00053
00054
00055
00056
00057
00058
00059
00060
00061
00062 static VALUE
00063 bsock_shutdown(int argc, VALUE *argv, VALUE sock)
00064 {
00065 VALUE howto;
00066 int how;
00067 rb_io_t *fptr;
00068
00069 if (rb_safe_level() >= 4 && !OBJ_TAINTED(sock)) {
00070 rb_raise(rb_eSecurityError, "Insecure: can't shutdown socket");
00071 }
00072 rb_scan_args(argc, argv, "01", &howto);
00073 if (howto == Qnil)
00074 how = SHUT_RDWR;
00075 else {
00076 how = rsock_shutdown_how_arg(howto);
00077 if (how != SHUT_WR && how != SHUT_RD && how != SHUT_RDWR) {
00078 rb_raise(rb_eArgError, "`how' should be either :SHUT_RD, :SHUT_WR, :SHUT_RDWR");
00079 }
00080 }
00081 GetOpenFile(sock, fptr);
00082 if (shutdown(fptr->fd, how) == -1)
00083 rb_sys_fail(0);
00084
00085 return INT2FIX(0);
00086 }
00087
00088
00089
00090
00091
00092
00093
00094
00095
00096
00097
00098 static VALUE
00099 bsock_close_read(VALUE sock)
00100 {
00101 rb_io_t *fptr;
00102
00103 if (rb_safe_level() >= 4 && !OBJ_TAINTED(sock)) {
00104 rb_raise(rb_eSecurityError, "Insecure: can't close socket");
00105 }
00106 GetOpenFile(sock, fptr);
00107 shutdown(fptr->fd, 0);
00108 if (!(fptr->mode & FMODE_WRITABLE)) {
00109 return rb_io_close(sock);
00110 }
00111 fptr->mode &= ~FMODE_READABLE;
00112
00113 return Qnil;
00114 }
00115
00116
00117
00118
00119
00120
00121
00122
00123
00124
00125
00126
00127
00128
00129
00130
00131 static VALUE
00132 bsock_close_write(VALUE sock)
00133 {
00134 rb_io_t *fptr;
00135
00136 if (rb_safe_level() >= 4 && !OBJ_TAINTED(sock)) {
00137 rb_raise(rb_eSecurityError, "Insecure: can't close socket");
00138 }
00139 GetOpenFile(sock, fptr);
00140 if (!(fptr->mode & FMODE_READABLE)) {
00141 return rb_io_close(sock);
00142 }
00143 shutdown(fptr->fd, 1);
00144 fptr->mode &= ~FMODE_WRITABLE;
00145
00146 return Qnil;
00147 }
00148
00149
00150
00151
00152
00153
00154
00155
00156
00157
00158
00159
00160
00161
00162
00163
00164
00165
00166
00167
00168
00169
00170
00171
00172
00173
00174
00175
00176
00177
00178
00179
00180
00181
00182
00183
00184
00185
00186
00187
00188
00189
00190
00191
00192
00193
00194
00195
00196
00197
00198
00199
00200
00201
00202
00203
00204
00205
00206 static VALUE
00207 bsock_setsockopt(int argc, VALUE *argv, VALUE sock)
00208 {
00209 VALUE lev, optname, val;
00210 int family, level, option;
00211 rb_io_t *fptr;
00212 int i;
00213 char *v;
00214 int vlen;
00215
00216 if (argc == 1) {
00217 lev = rb_funcall(argv[0], rb_intern("level"), 0);
00218 optname = rb_funcall(argv[0], rb_intern("optname"), 0);
00219 val = rb_funcall(argv[0], rb_intern("data"), 0);
00220 }
00221 else {
00222 rb_scan_args(argc, argv, "30", &lev, &optname, &val);
00223 }
00224
00225 rb_secure(2);
00226 GetOpenFile(sock, fptr);
00227 family = rsock_getfamily(fptr->fd);
00228 level = rsock_level_arg(family, lev);
00229 option = rsock_optname_arg(family, level, optname);
00230
00231 switch (TYPE(val)) {
00232 case T_FIXNUM:
00233 i = FIX2INT(val);
00234 goto numval;
00235 case T_FALSE:
00236 i = 0;
00237 goto numval;
00238 case T_TRUE:
00239 i = 1;
00240 numval:
00241 v = (char*)&i; vlen = (int)sizeof(i);
00242 break;
00243 default:
00244 StringValue(val);
00245 v = RSTRING_PTR(val);
00246 vlen = RSTRING_LENINT(val);
00247 break;
00248 }
00249
00250 #define rb_sys_fail_path(path) rb_sys_fail(NIL_P(path) ? 0 : RSTRING_PTR(path))
00251
00252 rb_io_check_closed(fptr);
00253 if (setsockopt(fptr->fd, level, option, v, vlen) < 0)
00254 rb_sys_fail_path(fptr->pathv);
00255
00256 return INT2FIX(0);
00257 }
00258
00259 #if !defined(__BEOS__)
00260
00261
00262
00263
00264
00265
00266
00267
00268
00269
00270
00271
00272
00273
00274
00275
00276
00277
00278
00279
00280
00281
00282
00283
00284
00285
00286
00287
00288
00289
00290
00291
00292
00293
00294
00295
00296
00297
00298
00299
00300
00301
00302
00303
00304
00305
00306
00307
00308
00309
00310
00311
00312
00313
00314
00315
00316 static VALUE
00317 bsock_getsockopt(VALUE sock, VALUE lev, VALUE optname)
00318 {
00319 int level, option;
00320 socklen_t len;
00321 char *buf;
00322 rb_io_t *fptr;
00323 int family;
00324
00325 GetOpenFile(sock, fptr);
00326 family = rsock_getfamily(fptr->fd);
00327 level = rsock_level_arg(family, lev);
00328 option = rsock_optname_arg(family, level, optname);
00329 len = 256;
00330 buf = ALLOCA_N(char,len);
00331
00332 rb_io_check_closed(fptr);
00333
00334 if (getsockopt(fptr->fd, level, option, buf, &len) < 0)
00335 rb_sys_fail_path(fptr->pathv);
00336
00337 return rsock_sockopt_new(family, level, option, rb_str_new(buf, len));
00338 }
00339 #else
00340 #define bsock_getsockopt rb_f_notimplement
00341 #endif
00342
00343
00344
00345
00346
00347
00348
00349
00350
00351
00352
00353
00354
00355
00356 static VALUE
00357 bsock_getsockname(VALUE sock)
00358 {
00359 struct sockaddr_storage buf;
00360 socklen_t len = (socklen_t)sizeof buf;
00361 socklen_t len0 = len;
00362 rb_io_t *fptr;
00363
00364 GetOpenFile(sock, fptr);
00365 if (getsockname(fptr->fd, (struct sockaddr*)&buf, &len) < 0)
00366 rb_sys_fail("getsockname(2)");
00367 if (len0 < len) len = len0;
00368 return rb_str_new((char*)&buf, len);
00369 }
00370
00371
00372
00373
00374
00375
00376
00377
00378
00379
00380
00381
00382
00383
00384
00385
00386
00387 static VALUE
00388 bsock_getpeername(VALUE sock)
00389 {
00390 struct sockaddr_storage buf;
00391 socklen_t len = (socklen_t)sizeof buf;
00392 socklen_t len0 = len;
00393 rb_io_t *fptr;
00394
00395 GetOpenFile(sock, fptr);
00396 if (getpeername(fptr->fd, (struct sockaddr*)&buf, &len) < 0)
00397 rb_sys_fail("getpeername(2)");
00398 if (len0 < len) len = len0;
00399 return rb_str_new((char*)&buf, len);
00400 }
00401
00402 #if defined(HAVE_GETPEEREID) || defined(SO_PEERCRED) || defined(HAVE_GETPEERUCRED)
00403
00404
00405
00406
00407
00408
00409
00410
00411
00412
00413
00414
00415
00416
00417
00418
00419
00420
00421
00422
00423
00424
00425 static VALUE
00426 bsock_getpeereid(VALUE self)
00427 {
00428 #if defined(HAVE_GETPEEREID)
00429 rb_io_t *fptr;
00430 uid_t euid;
00431 gid_t egid;
00432 GetOpenFile(self, fptr);
00433 if (getpeereid(fptr->fd, &euid, &egid) == -1)
00434 rb_sys_fail("getpeereid");
00435 return rb_assoc_new(UIDT2NUM(euid), GIDT2NUM(egid));
00436 #elif defined(SO_PEERCRED)
00437 rb_io_t *fptr;
00438 struct ucred cred;
00439 socklen_t len = sizeof(cred);
00440 GetOpenFile(self, fptr);
00441 if (getsockopt(fptr->fd, SOL_SOCKET, SO_PEERCRED, &cred, &len) == -1)
00442 rb_sys_fail("getsockopt(SO_PEERCRED)");
00443 return rb_assoc_new(UIDT2NUM(cred.uid), GIDT2NUM(cred.gid));
00444 #elif defined(HAVE_GETPEERUCRED)
00445 rb_io_t *fptr;
00446 ucred_t *uc = NULL;
00447 VALUE ret;
00448 GetOpenFile(self, fptr);
00449 if (getpeerucred(fptr->fd, &uc) == -1)
00450 rb_sys_fail("getpeerucred");
00451 ret = rb_assoc_new(UIDT2NUM(ucred_geteuid(uc)), GIDT2NUM(ucred_getegid(uc)));
00452 ucred_free(uc);
00453 return ret;
00454 #endif
00455 }
00456 #else
00457 #define bsock_getpeereid rb_f_notimplement
00458 #endif
00459
00460
00461
00462
00463
00464
00465
00466
00467
00468
00469
00470
00471
00472
00473
00474
00475
00476
00477 static VALUE
00478 bsock_local_address(VALUE sock)
00479 {
00480 struct sockaddr_storage buf;
00481 socklen_t len = (socklen_t)sizeof buf;
00482 socklen_t len0 = len;
00483 rb_io_t *fptr;
00484
00485 GetOpenFile(sock, fptr);
00486 if (getsockname(fptr->fd, (struct sockaddr*)&buf, &len) < 0)
00487 rb_sys_fail("getsockname(2)");
00488 if (len0 < len) len = len0;
00489 return rsock_fd_socket_addrinfo(fptr->fd, (struct sockaddr *)&buf, len);
00490 }
00491
00492
00493
00494
00495
00496
00497
00498
00499
00500
00501
00502
00503
00504
00505
00506
00507
00508
00509
00510
00511 static VALUE
00512 bsock_remote_address(VALUE sock)
00513 {
00514 struct sockaddr_storage buf;
00515 socklen_t len = (socklen_t)sizeof buf;
00516 socklen_t len0 = len;
00517 rb_io_t *fptr;
00518
00519 GetOpenFile(sock, fptr);
00520 if (getpeername(fptr->fd, (struct sockaddr*)&buf, &len) < 0)
00521 rb_sys_fail("getpeername(2)");
00522 if (len0 < len) len = len0;
00523 return rsock_fd_socket_addrinfo(fptr->fd, (struct sockaddr *)&buf, len);
00524 }
00525
00526
00527
00528
00529
00530
00531
00532
00533
00534
00535
00536
00537
00538
00539
00540
00541
00542
00543 VALUE
00544 rsock_bsock_send(int argc, VALUE *argv, VALUE sock)
00545 {
00546 struct rsock_send_arg arg;
00547 VALUE flags, to;
00548 rb_io_t *fptr;
00549 int n;
00550 rb_blocking_function_t *func;
00551
00552 rb_secure(4);
00553 rb_scan_args(argc, argv, "21", &arg.mesg, &flags, &to);
00554
00555 StringValue(arg.mesg);
00556 if (!NIL_P(to)) {
00557 SockAddrStringValue(to);
00558 to = rb_str_new4(to);
00559 arg.to = (struct sockaddr *)RSTRING_PTR(to);
00560 arg.tolen = (socklen_t)RSTRING_LENINT(to);
00561 func = rsock_sendto_blocking;
00562 }
00563 else {
00564 func = rsock_send_blocking;
00565 }
00566 GetOpenFile(sock, fptr);
00567 arg.fd = fptr->fd;
00568 arg.flags = NUM2INT(flags);
00569 while (rb_thread_fd_writable(arg.fd),
00570 (n = (int)BLOCKING_REGION_FD(func, &arg)) < 0) {
00571 if (rb_io_wait_writable(arg.fd)) {
00572 continue;
00573 }
00574 rb_sys_fail("send(2)");
00575 }
00576 return INT2FIX(n);
00577 }
00578
00579
00580
00581
00582
00583
00584
00585
00586
00587
00588
00589
00590
00591
00592 static VALUE
00593 bsock_do_not_reverse_lookup(VALUE sock)
00594 {
00595 rb_io_t *fptr;
00596
00597 GetOpenFile(sock, fptr);
00598 return (fptr->mode & FMODE_NOREVLOOKUP) ? Qtrue : Qfalse;
00599 }
00600
00601
00602
00603
00604
00605
00606
00607
00608
00609
00610
00611
00612
00613 static VALUE
00614 bsock_do_not_reverse_lookup_set(VALUE sock, VALUE state)
00615 {
00616 rb_io_t *fptr;
00617
00618 rb_secure(4);
00619 GetOpenFile(sock, fptr);
00620 if (RTEST(state)) {
00621 fptr->mode |= FMODE_NOREVLOOKUP;
00622 }
00623 else {
00624 fptr->mode &= ~FMODE_NOREVLOOKUP;
00625 }
00626 return sock;
00627 }
00628
00629
00630
00631
00632
00633
00634
00635
00636
00637
00638
00639
00640
00641
00642
00643
00644
00645
00646
00647
00648 static VALUE
00649 bsock_recv(int argc, VALUE *argv, VALUE sock)
00650 {
00651 return rsock_s_recvfrom(sock, argc, argv, RECV_RECV);
00652 }
00653
00654
00655
00656
00657
00658
00659
00660
00661
00662
00663
00664
00665
00666
00667
00668
00669
00670
00671
00672
00673
00674
00675
00676
00677
00678
00679
00680
00681
00682
00683
00684
00685
00686
00687
00688
00689
00690
00691
00692
00693
00694
00695
00696
00697
00698
00699 static VALUE
00700 bsock_recv_nonblock(int argc, VALUE *argv, VALUE sock)
00701 {
00702 return rsock_s_recvfrom_nonblock(sock, argc, argv, RECV_RECV);
00703 }
00704
00705
00706
00707
00708
00709
00710
00711
00712
00713 static VALUE
00714 bsock_do_not_rev_lookup(void)
00715 {
00716 return rsock_do_not_reverse_lookup?Qtrue:Qfalse;
00717 }
00718
00719
00720
00721
00722
00723
00724
00725
00726
00727
00728
00729
00730
00731
00732
00733
00734
00735 static VALUE
00736 bsock_do_not_rev_lookup_set(VALUE self, VALUE val)
00737 {
00738 rb_secure(4);
00739 rsock_do_not_reverse_lookup = RTEST(val);
00740 return val;
00741 }
00742
00743 void
00744 rsock_init_basicsocket(void)
00745 {
00746
00747
00748
00749
00750
00751 rb_cBasicSocket = rb_define_class("BasicSocket", rb_cIO);
00752 rb_undef_method(rb_cBasicSocket, "initialize");
00753
00754 rb_define_singleton_method(rb_cBasicSocket, "do_not_reverse_lookup",
00755 bsock_do_not_rev_lookup, 0);
00756 rb_define_singleton_method(rb_cBasicSocket, "do_not_reverse_lookup=",
00757 bsock_do_not_rev_lookup_set, 1);
00758 rb_define_singleton_method(rb_cBasicSocket, "for_fd", bsock_s_for_fd, 1);
00759
00760 rb_define_method(rb_cBasicSocket, "close_read", bsock_close_read, 0);
00761 rb_define_method(rb_cBasicSocket, "close_write", bsock_close_write, 0);
00762 rb_define_method(rb_cBasicSocket, "shutdown", bsock_shutdown, -1);
00763 rb_define_method(rb_cBasicSocket, "setsockopt", bsock_setsockopt, -1);
00764 rb_define_method(rb_cBasicSocket, "getsockopt", bsock_getsockopt, 2);
00765 rb_define_method(rb_cBasicSocket, "getsockname", bsock_getsockname, 0);
00766 rb_define_method(rb_cBasicSocket, "getpeername", bsock_getpeername, 0);
00767 rb_define_method(rb_cBasicSocket, "getpeereid", bsock_getpeereid, 0);
00768 rb_define_method(rb_cBasicSocket, "local_address", bsock_local_address, 0);
00769 rb_define_method(rb_cBasicSocket, "remote_address", bsock_remote_address, 0);
00770 rb_define_method(rb_cBasicSocket, "send", rsock_bsock_send, -1);
00771 rb_define_method(rb_cBasicSocket, "recv", bsock_recv, -1);
00772 rb_define_method(rb_cBasicSocket, "recv_nonblock", bsock_recv_nonblock, -1);
00773 rb_define_method(rb_cBasicSocket, "do_not_reverse_lookup", bsock_do_not_reverse_lookup, 0);
00774 rb_define_method(rb_cBasicSocket, "do_not_reverse_lookup=", bsock_do_not_reverse_lookup_set, 1);
00775
00776 rb_define_method(rb_cBasicSocket, "sendmsg", rsock_bsock_sendmsg, -1);
00777 rb_define_method(rb_cBasicSocket, "sendmsg_nonblock", rsock_bsock_sendmsg_nonblock, -1);
00778 rb_define_method(rb_cBasicSocket, "recvmsg", rsock_bsock_recvmsg, -1);
00779 rb_define_method(rb_cBasicSocket, "recvmsg_nonblock", rsock_bsock_recvmsg_nonblock, -1);
00780
00781 }
00782