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("shutdown(2)");
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_SOCKLEN(val);
00247 break;
00248 }
00249
00250 rb_io_check_closed(fptr);
00251 if (setsockopt(fptr->fd, level, option, v, vlen) < 0)
00252 rsock_sys_fail_path("setsockopt(2)", fptr->pathv);
00253
00254 return INT2FIX(0);
00255 }
00256
00257 #if !defined(__BEOS__)
00258
00259
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 static VALUE
00315 bsock_getsockopt(VALUE sock, VALUE lev, VALUE optname)
00316 {
00317 int level, option;
00318 socklen_t len;
00319 char *buf;
00320 rb_io_t *fptr;
00321 int family;
00322
00323 GetOpenFile(sock, fptr);
00324 family = rsock_getfamily(fptr->fd);
00325 level = rsock_level_arg(family, lev);
00326 option = rsock_optname_arg(family, level, optname);
00327 len = 256;
00328 buf = ALLOCA_N(char,len);
00329
00330 rb_io_check_closed(fptr);
00331
00332 if (getsockopt(fptr->fd, level, option, buf, &len) < 0)
00333 rsock_sys_fail_path("getsockopt(2)", fptr->pathv);
00334
00335 return rsock_sockopt_new(family, level, option, rb_str_new(buf, len));
00336 }
00337 #else
00338 #define bsock_getsockopt rb_f_notimplement
00339 #endif
00340
00341
00342
00343
00344
00345
00346
00347
00348
00349
00350
00351
00352
00353
00354 static VALUE
00355 bsock_getsockname(VALUE sock)
00356 {
00357 union_sockaddr buf;
00358 socklen_t len = (socklen_t)sizeof buf;
00359 socklen_t len0 = len;
00360 rb_io_t *fptr;
00361
00362 GetOpenFile(sock, fptr);
00363 if (getsockname(fptr->fd, &buf.addr, &len) < 0)
00364 rb_sys_fail("getsockname(2)");
00365 if (len0 < len) len = len0;
00366 return rb_str_new((char*)&buf, len);
00367 }
00368
00369
00370
00371
00372
00373
00374
00375
00376
00377
00378
00379
00380
00381
00382
00383
00384
00385 static VALUE
00386 bsock_getpeername(VALUE sock)
00387 {
00388 union_sockaddr buf;
00389 socklen_t len = (socklen_t)sizeof buf;
00390 socklen_t len0 = len;
00391 rb_io_t *fptr;
00392
00393 GetOpenFile(sock, fptr);
00394 if (getpeername(fptr->fd, &buf.addr, &len) < 0)
00395 rb_sys_fail("getpeername(2)");
00396 if (len0 < len) len = len0;
00397 return rb_str_new((char*)&buf, len);
00398 }
00399
00400 #if defined(HAVE_GETPEEREID) || defined(SO_PEERCRED) || defined(HAVE_GETPEERUCRED)
00401
00402
00403
00404
00405
00406
00407
00408
00409
00410
00411
00412
00413
00414
00415
00416
00417
00418
00419
00420
00421
00422
00423 static VALUE
00424 bsock_getpeereid(VALUE self)
00425 {
00426 #if defined(HAVE_GETPEEREID)
00427 rb_io_t *fptr;
00428 uid_t euid;
00429 gid_t egid;
00430 GetOpenFile(self, fptr);
00431 if (getpeereid(fptr->fd, &euid, &egid) == -1)
00432 rb_sys_fail("getpeereid(3)");
00433 return rb_assoc_new(UIDT2NUM(euid), GIDT2NUM(egid));
00434 #elif defined(SO_PEERCRED)
00435 rb_io_t *fptr;
00436 struct ucred cred;
00437 socklen_t len = sizeof(cred);
00438 GetOpenFile(self, fptr);
00439 if (getsockopt(fptr->fd, SOL_SOCKET, SO_PEERCRED, &cred, &len) == -1)
00440 rb_sys_fail("getsockopt(SO_PEERCRED)");
00441 return rb_assoc_new(UIDT2NUM(cred.uid), GIDT2NUM(cred.gid));
00442 #elif defined(HAVE_GETPEERUCRED)
00443 rb_io_t *fptr;
00444 ucred_t *uc = NULL;
00445 VALUE ret;
00446 GetOpenFile(self, fptr);
00447 if (getpeerucred(fptr->fd, &uc) == -1)
00448 rb_sys_fail("getpeerucred(3C)");
00449 ret = rb_assoc_new(UIDT2NUM(ucred_geteuid(uc)), GIDT2NUM(ucred_getegid(uc)));
00450 ucred_free(uc);
00451 return ret;
00452 #endif
00453 }
00454 #else
00455 #define bsock_getpeereid rb_f_notimplement
00456 #endif
00457
00458
00459
00460
00461
00462
00463
00464
00465
00466
00467
00468
00469
00470
00471
00472
00473
00474
00475 static VALUE
00476 bsock_local_address(VALUE sock)
00477 {
00478 union_sockaddr buf;
00479 socklen_t len = (socklen_t)sizeof buf;
00480 socklen_t len0 = len;
00481 rb_io_t *fptr;
00482
00483 GetOpenFile(sock, fptr);
00484 if (getsockname(fptr->fd, &buf.addr, &len) < 0)
00485 rb_sys_fail("getsockname(2)");
00486 if (len0 < len) len = len0;
00487 return rsock_fd_socket_addrinfo(fptr->fd, &buf.addr, len);
00488 }
00489
00490
00491
00492
00493
00494
00495
00496
00497
00498
00499
00500
00501
00502
00503
00504
00505
00506
00507
00508
00509 static VALUE
00510 bsock_remote_address(VALUE sock)
00511 {
00512 union_sockaddr buf;
00513 socklen_t len = (socklen_t)sizeof buf;
00514 socklen_t len0 = len;
00515 rb_io_t *fptr;
00516
00517 GetOpenFile(sock, fptr);
00518 if (getpeername(fptr->fd, &buf.addr, &len) < 0)
00519 rb_sys_fail("getpeername(2)");
00520 if (len0 < len) len = len0;
00521 return rsock_fd_socket_addrinfo(fptr->fd, &buf.addr, len);
00522 }
00523
00524
00525
00526
00527
00528
00529
00530
00531
00532
00533
00534
00535
00536
00537
00538
00539
00540
00541 VALUE
00542 rsock_bsock_send(int argc, VALUE *argv, VALUE sock)
00543 {
00544 struct rsock_send_arg arg;
00545 VALUE flags, to;
00546 rb_io_t *fptr;
00547 int n;
00548 rb_blocking_function_t *func;
00549
00550 rb_scan_args(argc, argv, "21", &arg.mesg, &flags, &to);
00551
00552 StringValue(arg.mesg);
00553 if (!NIL_P(to)) {
00554 SockAddrStringValue(to);
00555 to = rb_str_new4(to);
00556 arg.to = (struct sockaddr *)RSTRING_PTR(to);
00557 arg.tolen = RSTRING_SOCKLEN(to);
00558 func = rsock_sendto_blocking;
00559 }
00560 else {
00561 func = rsock_send_blocking;
00562 }
00563 GetOpenFile(sock, fptr);
00564 arg.fd = fptr->fd;
00565 arg.flags = NUM2INT(flags);
00566 while (rb_thread_fd_writable(arg.fd),
00567 (n = (int)BLOCKING_REGION_FD(func, &arg)) < 0) {
00568 if (rb_io_wait_writable(arg.fd)) {
00569 continue;
00570 }
00571 rb_sys_fail("send(2)");
00572 }
00573 return INT2FIX(n);
00574 }
00575
00576
00577
00578
00579
00580
00581
00582
00583
00584
00585
00586
00587
00588
00589 static VALUE
00590 bsock_do_not_reverse_lookup(VALUE sock)
00591 {
00592 rb_io_t *fptr;
00593
00594 GetOpenFile(sock, fptr);
00595 return (fptr->mode & FMODE_NOREVLOOKUP) ? Qtrue : Qfalse;
00596 }
00597
00598
00599
00600
00601
00602
00603
00604
00605
00606
00607
00608
00609
00610 static VALUE
00611 bsock_do_not_reverse_lookup_set(VALUE sock, VALUE state)
00612 {
00613 rb_io_t *fptr;
00614
00615 GetOpenFile(sock, fptr);
00616 if (RTEST(state)) {
00617 fptr->mode |= FMODE_NOREVLOOKUP;
00618 }
00619 else {
00620 fptr->mode &= ~FMODE_NOREVLOOKUP;
00621 }
00622 return sock;
00623 }
00624
00625
00626
00627
00628
00629
00630
00631
00632
00633
00634
00635
00636
00637
00638
00639
00640
00641
00642
00643
00644 static VALUE
00645 bsock_recv(int argc, VALUE *argv, VALUE sock)
00646 {
00647 return rsock_s_recvfrom(sock, argc, argv, RECV_RECV);
00648 }
00649
00650
00651
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 static VALUE
00696 bsock_recv_nonblock(int argc, VALUE *argv, VALUE sock)
00697 {
00698 return rsock_s_recvfrom_nonblock(sock, argc, argv, RECV_RECV);
00699 }
00700
00701
00702
00703
00704
00705
00706
00707
00708
00709 static VALUE
00710 bsock_do_not_rev_lookup(void)
00711 {
00712 return rsock_do_not_reverse_lookup?Qtrue:Qfalse;
00713 }
00714
00715
00716
00717
00718
00719
00720
00721
00722
00723
00724
00725
00726
00727
00728
00729
00730
00731 static VALUE
00732 bsock_do_not_rev_lookup_set(VALUE self, VALUE val)
00733 {
00734 rsock_do_not_reverse_lookup = RTEST(val);
00735 return val;
00736 }
00737
00738 void
00739 rsock_init_basicsocket(void)
00740 {
00741
00742
00743
00744
00745
00746 rb_cBasicSocket = rb_define_class("BasicSocket", rb_cIO);
00747 rb_undef_method(rb_cBasicSocket, "initialize");
00748
00749 rb_define_singleton_method(rb_cBasicSocket, "do_not_reverse_lookup",
00750 bsock_do_not_rev_lookup, 0);
00751 rb_define_singleton_method(rb_cBasicSocket, "do_not_reverse_lookup=",
00752 bsock_do_not_rev_lookup_set, 1);
00753 rb_define_singleton_method(rb_cBasicSocket, "for_fd", bsock_s_for_fd, 1);
00754
00755 rb_define_method(rb_cBasicSocket, "close_read", bsock_close_read, 0);
00756 rb_define_method(rb_cBasicSocket, "close_write", bsock_close_write, 0);
00757 rb_define_method(rb_cBasicSocket, "shutdown", bsock_shutdown, -1);
00758 rb_define_method(rb_cBasicSocket, "setsockopt", bsock_setsockopt, -1);
00759 rb_define_method(rb_cBasicSocket, "getsockopt", bsock_getsockopt, 2);
00760 rb_define_method(rb_cBasicSocket, "getsockname", bsock_getsockname, 0);
00761 rb_define_method(rb_cBasicSocket, "getpeername", bsock_getpeername, 0);
00762 rb_define_method(rb_cBasicSocket, "getpeereid", bsock_getpeereid, 0);
00763 rb_define_method(rb_cBasicSocket, "local_address", bsock_local_address, 0);
00764 rb_define_method(rb_cBasicSocket, "remote_address", bsock_remote_address, 0);
00765 rb_define_method(rb_cBasicSocket, "send", rsock_bsock_send, -1);
00766 rb_define_method(rb_cBasicSocket, "recv", bsock_recv, -1);
00767 rb_define_method(rb_cBasicSocket, "recv_nonblock", bsock_recv_nonblock, -1);
00768 rb_define_method(rb_cBasicSocket, "do_not_reverse_lookup", bsock_do_not_reverse_lookup, 0);
00769 rb_define_method(rb_cBasicSocket, "do_not_reverse_lookup=", bsock_do_not_reverse_lookup_set, 1);
00770
00771 rb_define_method(rb_cBasicSocket, "sendmsg", rsock_bsock_sendmsg, -1);
00772 rb_define_method(rb_cBasicSocket, "sendmsg_nonblock", rsock_bsock_sendmsg_nonblock, -1);
00773 rb_define_method(rb_cBasicSocket, "recvmsg", rsock_bsock_recvmsg, -1);
00774 rb_define_method(rb_cBasicSocket, "recvmsg_nonblock", rsock_bsock_recvmsg_nonblock, -1);
00775
00776 }
00777