1 /* 2 * Copyright 2006-2009, Haiku, Inc. All Rights Reserved. 3 * Distributed under the terms of the MIT License. 4 * 5 * Authors: 6 * Axel Dörfler, axeld@pinc-software.de 7 * Andrew Galante, haiku.galante@gmail.com 8 * Hugo Santos, hugosantos@gmail.com 9 */ 10 11 12 #include "EndpointManager.h" 13 #include "TCPEndpoint.h" 14 #include "tcp.h" 15 16 #include <net_protocol.h> 17 #include <net_stat.h> 18 19 #include <KernelExport.h> 20 #include <util/list.h> 21 22 #include <netinet/in.h> 23 #include <netinet/ip.h> 24 #include <new> 25 #include <stdlib.h> 26 #include <string.h> 27 28 #include <lock.h> 29 #include <util/AutoLock.h> 30 31 #include <NetBufferUtilities.h> 32 #include <NetUtilities.h> 33 34 35 //#define TRACE_TCP 36 #ifdef TRACE_TCP 37 # define TRACE(x) dprintf x 38 #else 39 # define TRACE(x) 40 #endif 41 42 43 typedef NetBufferField<uint16, offsetof(tcp_header, checksum)> TCPChecksumField; 44 45 46 net_buffer_module_info *gBufferModule; 47 net_datalink_module_info *gDatalinkModule; 48 net_socket_module_info *gSocketModule; 49 net_stack_module_info *gStackModule; 50 51 52 static EndpointManager* sEndpointManagers[AF_MAX]; 53 static rw_lock sEndpointManagersLock; 54 55 56 // The TCP header length is at most 60 bytes (0xf * 4). 57 static const int kMaxOptionSize = 60 - sizeof(tcp_header); 58 59 60 /*! Returns an endpoint manager for the specified domain, if any. 61 You need to hold the sEndpointManagersLock when calling this function. 62 */ 63 static inline EndpointManager* 64 endpoint_manager_for_locked(int family) 65 { 66 if (family >= AF_MAX || family < 0) 67 return NULL; 68 69 return sEndpointManagers[family]; 70 } 71 72 73 /*! Returns an endpoint manager for the specified domain, if any */ 74 static inline EndpointManager* 75 endpoint_manager_for(net_domain* domain) 76 { 77 ReadLocker _(sEndpointManagersLock); 78 79 return endpoint_manager_for_locked(domain->family); 80 } 81 82 83 static inline void 84 bump_option(tcp_option *&option, size_t &length) 85 { 86 if (option->kind <= TCP_OPTION_NOP) { 87 length++; 88 option = (tcp_option *)((uint8 *)option + 1); 89 } else { 90 length += option->length; 91 option = (tcp_option *)((uint8 *)option + option->length); 92 } 93 } 94 95 96 static inline size_t 97 add_options(tcp_segment_header &segment, uint8 *buffer, size_t bufferSize) 98 { 99 // Some network devices can be very sensitive to the ordering of TCP options 100 // https://github.com/torvalds/linux/blob/9e9fb7655ed585da8f468e29221f0ba194a5f613/net/ipv4/tcp_output.c#L598 101 102 tcp_option *option = (tcp_option *)buffer; 103 size_t length = 0; 104 105 if (segment.max_segment_size > 0 && length + 8 <= bufferSize) { 106 option->kind = TCP_OPTION_MAX_SEGMENT_SIZE; 107 option->length = 4; 108 option->max_segment_size = htons(segment.max_segment_size); 109 bump_option(option, length); 110 } 111 112 if ((segment.options & TCP_HAS_TIMESTAMPS) != 0 113 && length + 12 <= bufferSize) { 114 if ((segment.options & TCP_SACK_PERMITTED) != 0) { 115 // combine with timestamp 116 option->kind = TCP_OPTION_SACK_PERMITTED; 117 option->length = 2; 118 bump_option(option, length); 119 } else { 120 // two NOPs so the timestamps get aligned to a 4 byte boundary 121 option->kind = TCP_OPTION_NOP; 122 bump_option(option, length); 123 option->kind = TCP_OPTION_NOP; 124 bump_option(option, length); 125 } 126 option->kind = TCP_OPTION_TIMESTAMP; 127 option->length = 10; 128 option->timestamp.value = htonl(segment.timestamp_value); 129 option->timestamp.reply = htonl(segment.timestamp_reply); 130 bump_option(option, length); 131 } else if ((segment.options & TCP_SACK_PERMITTED) != 0 132 && length + 4 <= bufferSize) { 133 // two NOPs so that the subsequent data is aligned on a 4 byte boundary 134 option->kind = TCP_OPTION_NOP; 135 bump_option(option, length); 136 option->kind = TCP_OPTION_NOP; 137 bump_option(option, length); 138 option->kind = TCP_OPTION_SACK_PERMITTED; 139 option->length = 2; 140 bump_option(option, length); 141 } 142 143 if ((segment.options & TCP_HAS_WINDOW_SCALE) != 0 144 && length + 4 <= bufferSize) { 145 // one NOP so that the subsequent data is aligned on a 4 byte boundary 146 option->kind = TCP_OPTION_NOP; 147 bump_option(option, length); 148 149 option->kind = TCP_OPTION_WINDOW_SHIFT; 150 option->length = 3; 151 option->window_shift = segment.window_shift; 152 bump_option(option, length); 153 } 154 155 if (segment.sackCount > 0) { 156 int sackCount = ((int)(bufferSize - length) - 4) / sizeof(tcp_sack); 157 if (sackCount > segment.sackCount) 158 sackCount = segment.sackCount; 159 160 if (sackCount > 0) { 161 option->kind = TCP_OPTION_NOP; 162 bump_option(option, length); 163 option->kind = TCP_OPTION_NOP; 164 bump_option(option, length); 165 option->kind = TCP_OPTION_SACK; 166 option->length = 2 + sackCount * sizeof(tcp_sack); 167 memcpy(option->sack, segment.sacks, sackCount * sizeof(tcp_sack)); 168 bump_option(option, length); 169 } 170 } 171 172 if ((length & 3) == 0) { 173 // options completely fill out the option space 174 return length; 175 } 176 177 option->kind = TCP_OPTION_END; 178 return (length + 3) & ~3; 179 // bump to a multiple of 4 length 180 } 181 182 183 static void 184 process_options(tcp_segment_header &segment, net_buffer *buffer, size_t size) 185 { 186 if (size == 0) 187 return; 188 189 tcp_option *option; 190 191 uint8 optionsBuffer[kMaxOptionSize]; 192 if (gBufferModule->direct_access(buffer, sizeof(tcp_header), size, 193 (void **)&option) != B_OK) { 194 if ((size_t)size > sizeof(optionsBuffer)) { 195 dprintf("Ignoring TCP options larger than expected.\n"); 196 return; 197 } 198 199 gBufferModule->read(buffer, sizeof(tcp_header), optionsBuffer, size); 200 option = (tcp_option *)optionsBuffer; 201 } 202 203 while (size > 0) { 204 int32 length = -1; 205 206 switch (option->kind) { 207 case TCP_OPTION_END: 208 case TCP_OPTION_NOP: 209 length = 1; 210 break; 211 case TCP_OPTION_MAX_SEGMENT_SIZE: 212 if (option->length == 4 && size >= 4) 213 segment.max_segment_size = ntohs(option->max_segment_size); 214 break; 215 case TCP_OPTION_WINDOW_SHIFT: 216 if (option->length == 3 && size >= 3) { 217 segment.options |= TCP_HAS_WINDOW_SCALE; 218 segment.window_shift = option->window_shift; 219 } 220 break; 221 case TCP_OPTION_TIMESTAMP: 222 if (option->length == 10 && size >= 10) { 223 segment.options |= TCP_HAS_TIMESTAMPS; 224 segment.timestamp_value = ntohl(option->timestamp.value); 225 segment.timestamp_reply = 226 ntohl(option->timestamp.reply); 227 } 228 break; 229 case TCP_OPTION_SACK_PERMITTED: 230 if (option->length == 2 && size >= 2) 231 segment.options |= TCP_SACK_PERMITTED; 232 break; 233 case TCP_OPTION_SACK: 234 if (size >= option->length) { 235 segment.options |= TCP_HAS_SACK; 236 segment.sackCount = min_c((option->length - 2) 237 / sizeof(tcp_sack), MAX_SACK_BLKS); 238 for(int i = 0; i < segment.sackCount; ++i) { 239 segment.sacks[i].left_edge = ntohl( 240 option->sack[i].left_edge); 241 segment.sacks[i].right_edge = ntohl( 242 option->sack[i].right_edge); 243 } 244 } 245 break; 246 } 247 248 if (length < 0) { 249 length = option->length; 250 if (length == 0 || length > (ssize_t)size) 251 break; 252 } 253 254 option = (tcp_option *)((uint8 *)option + length); 255 size -= length; 256 } 257 } 258 259 260 #if 0 261 static void 262 dump_tcp_header(tcp_header &header) 263 { 264 dprintf(" source port: %u\n", ntohs(header.source_port)); 265 dprintf(" dest port: %u\n", ntohs(header.destination_port)); 266 dprintf(" sequence: %lu\n", header.Sequence()); 267 dprintf(" ack: %lu\n", header.Acknowledge()); 268 dprintf(" flags: %s%s%s%s%s%s\n", (header.flags & TCP_FLAG_FINISH) ? "FIN " : "", 269 (header.flags & TCP_FLAG_SYNCHRONIZE) ? "SYN " : "", 270 (header.flags & TCP_FLAG_RESET) ? "RST " : "", 271 (header.flags & TCP_FLAG_PUSH) ? "PUSH " : "", 272 (header.flags & TCP_FLAG_ACKNOWLEDGE) ? "ACK " : "", 273 (header.flags & TCP_FLAG_URGENT) ? "URG " : ""); 274 dprintf(" window: %u\n", header.AdvertisedWindow()); 275 dprintf(" urgent offset: %u\n", header.UrgentOffset()); 276 } 277 #endif 278 279 280 static int 281 dump_endpoints(int argc, char** argv) 282 { 283 for (int i = 0; i < AF_MAX; i++) { 284 EndpointManager* manager = sEndpointManagers[i]; 285 if (manager != NULL) 286 manager->Dump(); 287 } 288 289 return 0; 290 } 291 292 293 static int 294 dump_endpoint(int argc, char** argv) 295 { 296 if (argc < 2) { 297 kprintf("usage: tcp_endpoint [address]\n"); 298 return 0; 299 } 300 301 TCPEndpoint* endpoint = (TCPEndpoint*)parse_expression(argv[1]); 302 endpoint->Dump(); 303 304 return 0; 305 } 306 307 308 // #pragma mark - internal API 309 310 311 /*! Creates a new endpoint manager for the specified domain, or returns 312 an existing one for this domain. 313 */ 314 EndpointManager* 315 get_endpoint_manager(net_domain* domain) 316 { 317 // See if there is one already 318 EndpointManager* endpointManager = endpoint_manager_for(domain); 319 if (endpointManager != NULL) 320 return endpointManager; 321 322 WriteLocker _(sEndpointManagersLock); 323 324 endpointManager = endpoint_manager_for_locked(domain->family); 325 if (endpointManager != NULL) 326 return endpointManager; 327 328 // There is no endpoint manager for this domain yet, so we need 329 // to create one. 330 331 endpointManager = new(std::nothrow) EndpointManager(domain); 332 if (endpointManager == NULL) 333 return NULL; 334 335 if (endpointManager->Init() != B_OK) { 336 delete endpointManager; 337 return NULL; 338 } 339 340 sEndpointManagers[domain->family] = endpointManager; 341 return endpointManager; 342 } 343 344 345 void 346 put_endpoint_manager(EndpointManager* endpointManager) 347 { 348 // TODO: we may want to use reference counting instead of only discarding 349 // them on unload. But since there is likely only IPv4/v6 there is not much 350 // point to it. 351 } 352 353 354 const char* 355 name_for_state(tcp_state state) 356 { 357 switch (state) { 358 case CLOSED: 359 return "closed"; 360 case LISTEN: 361 return "listen"; 362 case SYNCHRONIZE_SENT: 363 return "syn-sent"; 364 case SYNCHRONIZE_RECEIVED: 365 return "syn-received"; 366 case ESTABLISHED: 367 return "established"; 368 369 // peer closes the connection 370 case FINISH_RECEIVED: 371 return "close-wait"; 372 case WAIT_FOR_FINISH_ACKNOWLEDGE: 373 return "last-ack"; 374 375 // we close the connection 376 case FINISH_SENT: 377 return "fin-wait1"; 378 case FINISH_ACKNOWLEDGED: 379 return "fin-wait2"; 380 case CLOSING: 381 return "closing"; 382 383 case TIME_WAIT: 384 return "time-wait"; 385 } 386 387 return "-"; 388 } 389 390 391 /*! Constructs a TCP header on \a buffer with the specified values 392 for \a flags, \a seq \a ack and \a advertisedWindow. 393 */ 394 status_t 395 add_tcp_header(net_address_module_info* addressModule, 396 tcp_segment_header& segment, net_buffer* buffer) 397 { 398 buffer->protocol = IPPROTO_TCP; 399 400 uint8 optionsBuffer[kMaxOptionSize]; 401 uint32 optionsLength = add_options(segment, optionsBuffer, 402 sizeof(optionsBuffer)); 403 404 NetBufferPrepend<tcp_header> bufferHeader(buffer, 405 sizeof(tcp_header) + optionsLength); 406 if (bufferHeader.Status() != B_OK) 407 return bufferHeader.Status(); 408 409 tcp_header& header = bufferHeader.Data(); 410 411 header.source_port = addressModule->get_port(buffer->source); 412 header.destination_port = addressModule->get_port(buffer->destination); 413 header.sequence = htonl(segment.sequence); 414 header.acknowledge = (segment.flags & TCP_FLAG_ACKNOWLEDGE) 415 ? htonl(segment.acknowledge) : 0; 416 header.reserved = 0; 417 header.header_length = (sizeof(tcp_header) + optionsLength) >> 2; 418 header.flags = segment.flags; 419 header.advertised_window = htons(segment.advertised_window); 420 header.checksum = 0; 421 header.urgent_offset = htons(segment.urgent_offset); 422 423 // we must detach before calculating the checksum as we may 424 // not have a contiguous buffer. 425 bufferHeader.Sync(); 426 427 if (optionsLength > 0) { 428 gBufferModule->write(buffer, sizeof(tcp_header), optionsBuffer, 429 optionsLength); 430 } 431 432 TRACE(("add_tcp_header(): buffer %p, flags 0x%x, seq %" B_PRIu32 ", ack %" B_PRIu32 ", up %u, " 433 "win %u\n", buffer, segment.flags, segment.sequence, 434 segment.acknowledge, segment.urgent_offset, segment.advertised_window)); 435 436 *TCPChecksumField(buffer) = Checksum::PseudoHeader(addressModule, 437 gBufferModule, buffer, IPPROTO_TCP); 438 439 return B_OK; 440 } 441 442 443 size_t 444 tcp_options_length(tcp_segment_header& segment) 445 { 446 size_t length = 0; 447 448 if (segment.max_segment_size > 0) 449 length += 4; 450 451 if ((segment.options & TCP_HAS_TIMESTAMPS) != 0) 452 length += 12; 453 else if ((segment.options & TCP_SACK_PERMITTED) != 0) 454 length += 4; 455 456 if ((segment.options & TCP_HAS_WINDOW_SCALE) != 0) 457 length += 4; 458 459 if (segment.sackCount > 0) { 460 int sackCount = min_c((int)((kMaxOptionSize - length - 4) 461 / sizeof(tcp_sack)), segment.sackCount); 462 if (sackCount > 0) 463 length += 4 + sackCount * sizeof(tcp_sack); 464 } 465 466 if ((length & 3) == 0) 467 return length; 468 469 return (length + 3) & ~3; 470 } 471 472 473 // #pragma mark - protocol API 474 475 476 net_protocol* 477 tcp_init_protocol(net_socket* socket) 478 { 479 TCPEndpoint* protocol = new (std::nothrow) TCPEndpoint(socket); 480 if (protocol == NULL) 481 return NULL; 482 483 if (protocol->InitCheck() != B_OK) { 484 delete protocol; 485 return NULL; 486 } 487 488 TRACE(("Creating new TCPEndpoint: %p\n", protocol)); 489 socket->protocol = IPPROTO_TCP; 490 return protocol; 491 } 492 493 494 status_t 495 tcp_uninit_protocol(net_protocol* protocol) 496 { 497 TRACE(("Deleting TCPEndpoint: %p\n", protocol)); 498 delete (TCPEndpoint*)protocol; 499 return B_OK; 500 } 501 502 503 status_t 504 tcp_open(net_protocol* protocol) 505 { 506 return ((TCPEndpoint*)protocol)->Open(); 507 } 508 509 510 status_t 511 tcp_close(net_protocol* protocol) 512 { 513 return ((TCPEndpoint*)protocol)->Close(); 514 } 515 516 517 status_t 518 tcp_free(net_protocol* protocol) 519 { 520 ((TCPEndpoint*)protocol)->Free(); 521 return B_OK; 522 } 523 524 525 status_t 526 tcp_connect(net_protocol* protocol, const struct sockaddr* address) 527 { 528 return ((TCPEndpoint*)protocol)->Connect(address); 529 } 530 531 532 status_t 533 tcp_accept(net_protocol* protocol, struct net_socket** _acceptedSocket) 534 { 535 return ((TCPEndpoint*)protocol)->Accept(_acceptedSocket); 536 } 537 538 539 status_t 540 tcp_control(net_protocol* _protocol, int level, int option, void* value, 541 size_t* _length) 542 { 543 TCPEndpoint* protocol = (TCPEndpoint*)_protocol; 544 545 if ((level & LEVEL_MASK) == IPPROTO_TCP) { 546 if (option == NET_STAT_SOCKET) 547 return protocol->FillStat((net_stat*)value); 548 } 549 550 return protocol->next->module->control(protocol->next, level, option, 551 value, _length); 552 } 553 554 555 status_t 556 tcp_getsockopt(net_protocol* _protocol, int level, int option, void* value, 557 int* _length) 558 { 559 TCPEndpoint* protocol = (TCPEndpoint*)_protocol; 560 561 if (level == IPPROTO_TCP) 562 return protocol->GetOption(option, value, _length); 563 564 return protocol->next->module->getsockopt(protocol->next, level, option, 565 value, _length); 566 } 567 568 569 status_t 570 tcp_setsockopt(net_protocol* _protocol, int level, int option, 571 const void* _value, int length) 572 { 573 TCPEndpoint* protocol = (TCPEndpoint*)_protocol; 574 575 if (level == SOL_SOCKET) { 576 if (option == SO_SNDBUF || option == SO_RCVBUF) { 577 if (length != sizeof(int)) 578 return B_BAD_VALUE; 579 580 status_t status; 581 const int* value = (const int*)_value; 582 583 if (option == SO_SNDBUF) 584 status = protocol->SetSendBufferSize(*value); 585 else 586 status = protocol->SetReceiveBufferSize(*value); 587 588 if (status < B_OK) 589 return status; 590 } 591 } else if (level == IPPROTO_TCP) 592 return protocol->SetOption(option, _value, length); 593 594 return protocol->next->module->setsockopt(protocol->next, level, option, 595 _value, length); 596 } 597 598 599 status_t 600 tcp_bind(net_protocol* protocol, const struct sockaddr* address) 601 { 602 return ((TCPEndpoint*)protocol)->Bind(address); 603 } 604 605 606 status_t 607 tcp_unbind(net_protocol* protocol, struct sockaddr* address) 608 { 609 return ((TCPEndpoint*)protocol)->Unbind(address); 610 } 611 612 613 status_t 614 tcp_listen(net_protocol* protocol, int count) 615 { 616 return ((TCPEndpoint*)protocol)->Listen(count); 617 } 618 619 620 status_t 621 tcp_shutdown(net_protocol* protocol, int direction) 622 { 623 return ((TCPEndpoint*)protocol)->Shutdown(direction); 624 } 625 626 627 status_t 628 tcp_send_data(net_protocol* protocol, net_buffer* buffer) 629 { 630 return ((TCPEndpoint*)protocol)->SendData(buffer); 631 } 632 633 634 status_t 635 tcp_send_routed_data(net_protocol* protocol, struct net_route* route, 636 net_buffer* buffer) 637 { 638 // TCP never sends routed data 639 return B_ERROR; 640 } 641 642 643 ssize_t 644 tcp_send_avail(net_protocol* protocol) 645 { 646 return ((TCPEndpoint*)protocol)->SendAvailable(); 647 } 648 649 650 status_t 651 tcp_read_data(net_protocol* protocol, size_t numBytes, uint32 flags, 652 net_buffer** _buffer) 653 { 654 return ((TCPEndpoint*)protocol)->ReadData(numBytes, flags, _buffer); 655 } 656 657 658 ssize_t 659 tcp_read_avail(net_protocol* protocol) 660 { 661 return ((TCPEndpoint*)protocol)->ReadAvailable(); 662 } 663 664 665 struct net_domain* 666 tcp_get_domain(net_protocol* protocol) 667 { 668 return protocol->next->module->get_domain(protocol->next); 669 } 670 671 672 size_t 673 tcp_get_mtu(net_protocol* protocol, const struct sockaddr* address) 674 { 675 return protocol->next->module->get_mtu(protocol->next, address); 676 } 677 678 679 status_t 680 tcp_receive_data(net_buffer* buffer) 681 { 682 TRACE(("TCP: Received buffer %p\n", buffer)); 683 684 if (buffer->interface_address == NULL 685 || buffer->interface_address->domain == NULL) 686 return B_ERROR; 687 688 net_domain* domain = buffer->interface_address->domain; 689 net_address_module_info* addressModule = domain->address_module; 690 691 NetBufferHeaderReader<tcp_header> bufferHeader(buffer); 692 if (bufferHeader.Status() < B_OK) 693 return bufferHeader.Status(); 694 695 tcp_header& header = bufferHeader.Data(); 696 697 uint16 headerLength = header.HeaderLength(); 698 if (headerLength < sizeof(tcp_header)) 699 return B_BAD_DATA; 700 701 if (Checksum::PseudoHeader(addressModule, gBufferModule, buffer, 702 IPPROTO_TCP) != 0) 703 return B_BAD_DATA; 704 705 addressModule->set_port(buffer->source, header.source_port); 706 addressModule->set_port(buffer->destination, header.destination_port); 707 708 TRACE((" Looking for: peer %s, local %s\n", 709 AddressString(domain, buffer->source, true).Data(), 710 AddressString(domain, buffer->destination, true).Data())); 711 //dump_tcp_header(header); 712 //gBufferModule->dump(buffer); 713 714 tcp_segment_header segment(header.flags); 715 segment.sequence = header.Sequence(); 716 segment.acknowledge = header.Acknowledge(); 717 segment.advertised_window = header.AdvertisedWindow(); 718 segment.urgent_offset = header.UrgentOffset(); 719 process_options(segment, buffer, headerLength - sizeof(tcp_header)); 720 721 bufferHeader.Remove(headerLength); 722 // we no longer need to keep the header around 723 724 EndpointManager* endpointManager = endpoint_manager_for(domain); 725 if (endpointManager == NULL) { 726 TRACE((" No endpoint manager!\n")); 727 return B_ERROR; 728 } 729 730 int32 segmentAction = DROP; 731 732 TCPEndpoint* endpoint = endpointManager->FindConnection( 733 buffer->destination, buffer->source); 734 if (endpoint != NULL) { 735 segmentAction = endpoint->SegmentReceived(segment, buffer); 736 737 // There are some states in which the socket could have been deleted 738 // while handling a segment. If this flag is set in segmentAction 739 // then we know the socket has been freed and can skip releasing 740 // the reference acquired in EndpointManager::FindConnection() 741 // above. 742 if ((segmentAction & DELETED_ENDPOINT) == 0) 743 gSocketModule->release_socket(endpoint->socket); 744 } else if ((segment.flags & TCP_FLAG_RESET) == 0) 745 segmentAction = DROP | RESET; 746 747 if ((segmentAction & RESET) != 0) { 748 // send reset 749 endpointManager->ReplyWithReset(segment, buffer); 750 } 751 if ((segmentAction & DROP) != 0) 752 gBufferModule->free(buffer); 753 754 return B_OK; 755 } 756 757 758 status_t 759 tcp_error_received(net_error error, net_buffer* data) 760 { 761 return B_ERROR; 762 } 763 764 765 status_t 766 tcp_error_reply(net_protocol* protocol, net_buffer* cause, net_error error, 767 net_error_data* errorData) 768 { 769 return B_ERROR; 770 } 771 772 773 // #pragma mark - 774 775 776 static status_t 777 tcp_init() 778 { 779 rw_lock_init(&sEndpointManagersLock, "endpoint managers"); 780 781 status_t status = gStackModule->register_domain_protocols(AF_INET, 782 SOCK_STREAM, 0, 783 "network/protocols/tcp/v1", 784 "network/protocols/ipv4/v1", 785 NULL); 786 if (status < B_OK) 787 return status; 788 status = gStackModule->register_domain_protocols(AF_INET6, 789 SOCK_STREAM, 0, 790 "network/protocols/tcp/v1", 791 "network/protocols/ipv6/v1", 792 NULL); 793 if (status < B_OK) 794 return status; 795 796 status = gStackModule->register_domain_protocols(AF_INET, SOCK_STREAM, 797 IPPROTO_TCP, 798 "network/protocols/tcp/v1", 799 "network/protocols/ipv4/v1", 800 NULL); 801 if (status < B_OK) 802 return status; 803 status = gStackModule->register_domain_protocols(AF_INET6, SOCK_STREAM, 804 IPPROTO_TCP, 805 "network/protocols/tcp/v1", 806 "network/protocols/ipv6/v1", 807 NULL); 808 if (status < B_OK) 809 return status; 810 811 status = gStackModule->register_domain_receiving_protocol(AF_INET, 812 IPPROTO_TCP, "network/protocols/tcp/v1"); 813 if (status < B_OK) 814 return status; 815 status = gStackModule->register_domain_receiving_protocol(AF_INET6, 816 IPPROTO_TCP, "network/protocols/tcp/v1"); 817 if (status < B_OK) 818 return status; 819 820 add_debugger_command("tcp_endpoints", dump_endpoints, 821 "lists all open TCP endpoints"); 822 add_debugger_command("tcp_endpoint", dump_endpoint, 823 "dumps a TCP endpoint internal state"); 824 825 return B_OK; 826 } 827 828 829 static status_t 830 tcp_uninit() 831 { 832 remove_debugger_command("tcp_endpoint", dump_endpoint); 833 remove_debugger_command("tcp_endpoints", dump_endpoints); 834 835 rw_lock_destroy(&sEndpointManagersLock); 836 837 for (int i = 0; i < AF_MAX; i++) { 838 delete sEndpointManagers[i]; 839 } 840 841 return B_OK; 842 } 843 844 845 static status_t 846 tcp_std_ops(int32 op, ...) 847 { 848 switch (op) { 849 case B_MODULE_INIT: 850 return tcp_init(); 851 852 case B_MODULE_UNINIT: 853 return tcp_uninit(); 854 855 default: 856 return B_ERROR; 857 } 858 } 859 860 861 net_protocol_module_info sTCPModule = { 862 { 863 "network/protocols/tcp/v1", 864 0, 865 tcp_std_ops 866 }, 867 0, 868 869 tcp_init_protocol, 870 tcp_uninit_protocol, 871 tcp_open, 872 tcp_close, 873 tcp_free, 874 tcp_connect, 875 tcp_accept, 876 tcp_control, 877 tcp_getsockopt, 878 tcp_setsockopt, 879 tcp_bind, 880 tcp_unbind, 881 tcp_listen, 882 tcp_shutdown, 883 tcp_send_data, 884 tcp_send_routed_data, 885 tcp_send_avail, 886 tcp_read_data, 887 tcp_read_avail, 888 tcp_get_domain, 889 tcp_get_mtu, 890 tcp_receive_data, 891 NULL, // deliver_data() 892 tcp_error_received, 893 tcp_error_reply, 894 NULL, // add_ancillary_data() 895 NULL, // process_ancillary_data() 896 NULL, // process_ancillary_data_no_container() 897 NULL, // send_data_no_buffer() 898 NULL // read_data_no_buffer() 899 }; 900 901 module_dependency module_dependencies[] = { 902 {NET_STACK_MODULE_NAME, (module_info **)&gStackModule}, 903 {NET_BUFFER_MODULE_NAME, (module_info **)&gBufferModule}, 904 {NET_DATALINK_MODULE_NAME, (module_info **)&gDatalinkModule}, 905 {NET_SOCKET_MODULE_NAME, (module_info **)&gSocketModule}, 906 {} 907 }; 908 909 module_info *modules[] = { 910 (module_info *)&sTCPModule, 911 NULL 912 }; 913