TCPSocket.cpp 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671
  1. /*
  2. * Copyright (c) 2018-2020, Andreas Kling <kling@serenityos.org>
  3. *
  4. * SPDX-License-Identifier: BSD-2-Clause
  5. */
  6. #include <AK/Singleton.h>
  7. #include <AK/Time.h>
  8. #include <Kernel/Debug.h>
  9. #include <Kernel/Devices/Generic/RandomDevice.h>
  10. #include <Kernel/FileSystem/OpenFileDescription.h>
  11. #include <Kernel/Locking/MutexProtected.h>
  12. #include <Kernel/Net/EthernetFrameHeader.h>
  13. #include <Kernel/Net/IPv4.h>
  14. #include <Kernel/Net/NetworkAdapter.h>
  15. #include <Kernel/Net/NetworkingManagement.h>
  16. #include <Kernel/Net/Routing.h>
  17. #include <Kernel/Net/TCP.h>
  18. #include <Kernel/Net/TCPSocket.h>
  19. #include <Kernel/Security/Random.h>
  20. #include <Kernel/Tasks/Process.h>
  21. #include <Kernel/Time/TimeManagement.h>
  22. namespace Kernel {
  23. void TCPSocket::for_each(Function<void(TCPSocket const&)> callback)
  24. {
  25. sockets_by_tuple().for_each_shared([&](auto const& it) {
  26. callback(*it.value);
  27. });
  28. }
  29. ErrorOr<void> TCPSocket::try_for_each(Function<ErrorOr<void>(TCPSocket const&)> callback)
  30. {
  31. return sockets_by_tuple().with_shared([&](auto const& sockets) -> ErrorOr<void> {
  32. for (auto& it : sockets)
  33. TRY(callback(*it.value));
  34. return {};
  35. });
  36. }
  37. bool TCPSocket::unref() const
  38. {
  39. bool did_hit_zero = sockets_by_tuple().with_exclusive([&](auto& table) {
  40. if (deref_base())
  41. return false;
  42. table.remove(tuple());
  43. const_cast<TCPSocket&>(*this).revoke_weak_ptrs();
  44. return true;
  45. });
  46. if (did_hit_zero) {
  47. const_cast<TCPSocket&>(*this).will_be_destroyed();
  48. delete this;
  49. }
  50. return did_hit_zero;
  51. }
  52. void TCPSocket::set_state(State new_state)
  53. {
  54. dbgln_if(TCP_SOCKET_DEBUG, "TCPSocket({}) state moving from {} to {}", this, to_string(m_state), to_string(new_state));
  55. auto was_disconnected = protocol_is_disconnected();
  56. auto previous_role = m_role;
  57. m_state = new_state;
  58. if (new_state == State::Established && m_direction == Direction::Outgoing) {
  59. set_role(Role::Connected);
  60. clear_so_error();
  61. }
  62. if (new_state == State::TimeWait) {
  63. // Once we hit TimeWait, we are only holding the socket in case there
  64. // are packets on the way which we wouldn't want a new socket to get hit
  65. // with, so there's no point in keeping the receive buffer around.
  66. drop_receive_buffer();
  67. }
  68. if (new_state == State::Closed) {
  69. closing_sockets().with_exclusive([&](auto& table) {
  70. table.remove(tuple());
  71. });
  72. if (m_originator)
  73. release_to_originator();
  74. }
  75. if (previous_role != m_role || was_disconnected != protocol_is_disconnected())
  76. evaluate_block_conditions();
  77. }
  78. static Singleton<MutexProtected<HashMap<IPv4SocketTuple, RefPtr<TCPSocket>>>> s_socket_closing;
  79. MutexProtected<HashMap<IPv4SocketTuple, RefPtr<TCPSocket>>>& TCPSocket::closing_sockets()
  80. {
  81. return *s_socket_closing;
  82. }
  83. static Singleton<MutexProtected<HashMap<IPv4SocketTuple, TCPSocket*>>> s_socket_tuples;
  84. MutexProtected<HashMap<IPv4SocketTuple, TCPSocket*>>& TCPSocket::sockets_by_tuple()
  85. {
  86. return *s_socket_tuples;
  87. }
  88. RefPtr<TCPSocket> TCPSocket::from_tuple(IPv4SocketTuple const& tuple)
  89. {
  90. return sockets_by_tuple().with_shared([&](auto const& table) -> RefPtr<TCPSocket> {
  91. auto exact_match = table.get(tuple);
  92. if (exact_match.has_value())
  93. return { *exact_match.value() };
  94. auto address_tuple = IPv4SocketTuple(tuple.local_address(), tuple.local_port(), IPv4Address(), 0);
  95. auto address_match = table.get(address_tuple);
  96. if (address_match.has_value())
  97. return { *address_match.value() };
  98. auto wildcard_tuple = IPv4SocketTuple(IPv4Address(), tuple.local_port(), IPv4Address(), 0);
  99. auto wildcard_match = table.get(wildcard_tuple);
  100. if (wildcard_match.has_value())
  101. return { *wildcard_match.value() };
  102. return {};
  103. });
  104. }
  105. ErrorOr<NonnullRefPtr<TCPSocket>> TCPSocket::try_create_client(IPv4Address const& new_local_address, u16 new_local_port, IPv4Address const& new_peer_address, u16 new_peer_port)
  106. {
  107. auto tuple = IPv4SocketTuple(new_local_address, new_local_port, new_peer_address, new_peer_port);
  108. return sockets_by_tuple().with_exclusive([&](auto& table) -> ErrorOr<NonnullRefPtr<TCPSocket>> {
  109. if (table.contains(tuple))
  110. return EEXIST;
  111. auto receive_buffer = TRY(try_create_receive_buffer());
  112. auto client = TRY(TCPSocket::try_create(protocol(), move(receive_buffer)));
  113. client->set_setup_state(SetupState::InProgress);
  114. client->set_local_address(new_local_address);
  115. client->set_local_port(new_local_port);
  116. client->set_peer_address(new_peer_address);
  117. client->set_peer_port(new_peer_port);
  118. client->set_bound(true);
  119. client->set_direction(Direction::Incoming);
  120. client->set_originator(*this);
  121. m_pending_release_for_accept.set(tuple, client);
  122. table.set(tuple, client);
  123. return { move(client) };
  124. });
  125. }
  126. void TCPSocket::release_to_originator()
  127. {
  128. VERIFY(!!m_originator);
  129. m_originator.strong_ref()->release_for_accept(*this);
  130. m_originator.clear();
  131. }
  132. void TCPSocket::release_for_accept(NonnullRefPtr<TCPSocket> socket)
  133. {
  134. VERIFY(m_pending_release_for_accept.contains(socket->tuple()));
  135. m_pending_release_for_accept.remove(socket->tuple());
  136. // FIXME: Should we observe this error somehow?
  137. [[maybe_unused]] auto rc = queue_connection_from(move(socket));
  138. }
  139. TCPSocket::TCPSocket(int protocol, NonnullOwnPtr<DoubleBuffer> receive_buffer, NonnullOwnPtr<KBuffer> scratch_buffer)
  140. : IPv4Socket(SOCK_STREAM, protocol, move(receive_buffer), move(scratch_buffer))
  141. , m_last_ack_sent_time(TimeManagement::the().monotonic_time())
  142. , m_last_retransmit_time(TimeManagement::the().monotonic_time())
  143. {
  144. }
  145. TCPSocket::~TCPSocket()
  146. {
  147. dequeue_for_retransmit();
  148. dbgln_if(TCP_SOCKET_DEBUG, "~TCPSocket in state {}", to_string(state()));
  149. }
  150. ErrorOr<NonnullRefPtr<TCPSocket>> TCPSocket::try_create(int protocol, NonnullOwnPtr<DoubleBuffer> receive_buffer)
  151. {
  152. // Note: Scratch buffer is only used for SOCK_STREAM sockets.
  153. auto scratch_buffer = TRY(KBuffer::try_create_with_size("TCPSocket: Scratch buffer"sv, 65536));
  154. return adopt_nonnull_ref_or_enomem(new (nothrow) TCPSocket(protocol, move(receive_buffer), move(scratch_buffer)));
  155. }
  156. ErrorOr<size_t> TCPSocket::protocol_size(ReadonlyBytes raw_ipv4_packet)
  157. {
  158. auto& ipv4_packet = *reinterpret_cast<IPv4Packet const*>(raw_ipv4_packet.data());
  159. auto& tcp_packet = *static_cast<TCPPacket const*>(ipv4_packet.payload());
  160. return raw_ipv4_packet.size() - sizeof(IPv4Packet) - tcp_packet.header_size();
  161. }
  162. ErrorOr<size_t> TCPSocket::protocol_receive(ReadonlyBytes raw_ipv4_packet, UserOrKernelBuffer& buffer, size_t buffer_size, [[maybe_unused]] int flags)
  163. {
  164. auto& ipv4_packet = *reinterpret_cast<IPv4Packet const*>(raw_ipv4_packet.data());
  165. auto& tcp_packet = *static_cast<TCPPacket const*>(ipv4_packet.payload());
  166. size_t payload_size = raw_ipv4_packet.size() - sizeof(IPv4Packet) - tcp_packet.header_size();
  167. dbgln_if(TCP_SOCKET_DEBUG, "payload_size {}, will it fit in {}?", payload_size, buffer_size);
  168. VERIFY(buffer_size >= payload_size);
  169. SOCKET_TRY(buffer.write(tcp_packet.payload(), payload_size));
  170. return payload_size;
  171. }
  172. ErrorOr<size_t> TCPSocket::protocol_send(UserOrKernelBuffer const& data, size_t data_length)
  173. {
  174. auto adapter = bound_interface().with([](auto& bound_device) -> RefPtr<NetworkAdapter> { return bound_device; });
  175. RoutingDecision routing_decision = route_to(peer_address(), local_address(), adapter);
  176. if (routing_decision.is_zero())
  177. return set_so_error(EHOSTUNREACH);
  178. size_t mss = routing_decision.adapter->mtu() - sizeof(IPv4Packet) - sizeof(TCPPacket);
  179. data_length = min(data_length, mss);
  180. TRY(send_tcp_packet(TCPFlags::PSH | TCPFlags::ACK, &data, data_length, &routing_decision));
  181. return data_length;
  182. }
  183. ErrorOr<void> TCPSocket::send_ack(bool allow_duplicate)
  184. {
  185. if (!allow_duplicate && m_last_ack_number_sent == m_ack_number)
  186. return {};
  187. return send_tcp_packet(TCPFlags::ACK);
  188. }
  189. ErrorOr<void> TCPSocket::send_tcp_packet(u16 flags, UserOrKernelBuffer const* payload, size_t payload_size, RoutingDecision* user_routing_decision)
  190. {
  191. auto adapter = bound_interface().with([](auto& bound_device) -> RefPtr<NetworkAdapter> { return bound_device; });
  192. RoutingDecision routing_decision = user_routing_decision ? *user_routing_decision : route_to(peer_address(), local_address(), adapter);
  193. if (routing_decision.is_zero())
  194. return set_so_error(EHOSTUNREACH);
  195. auto ipv4_payload_offset = routing_decision.adapter->ipv4_payload_offset();
  196. bool const has_mss_option = flags == TCPFlags::SYN;
  197. const size_t options_size = has_mss_option ? sizeof(TCPOptionMSS) : 0;
  198. const size_t tcp_header_size = sizeof(TCPPacket) + options_size;
  199. const size_t buffer_size = ipv4_payload_offset + tcp_header_size + payload_size;
  200. auto packet = routing_decision.adapter->acquire_packet_buffer(buffer_size);
  201. if (!packet)
  202. return set_so_error(ENOMEM);
  203. routing_decision.adapter->fill_in_ipv4_header(*packet, local_address(),
  204. routing_decision.next_hop, peer_address(), IPv4Protocol::TCP,
  205. buffer_size - ipv4_payload_offset, type_of_service(), ttl());
  206. memset(packet->buffer->data() + ipv4_payload_offset, 0, sizeof(TCPPacket));
  207. auto& tcp_packet = *(TCPPacket*)(packet->buffer->data() + ipv4_payload_offset);
  208. VERIFY(local_port());
  209. tcp_packet.set_source_port(local_port());
  210. tcp_packet.set_destination_port(peer_port());
  211. tcp_packet.set_window_size(NumericLimits<u16>::max());
  212. tcp_packet.set_sequence_number(m_sequence_number);
  213. tcp_packet.set_data_offset(tcp_header_size / sizeof(u32));
  214. tcp_packet.set_flags(flags);
  215. if (payload) {
  216. if (auto result = payload->read(tcp_packet.payload(), payload_size); result.is_error()) {
  217. routing_decision.adapter->release_packet_buffer(*packet);
  218. return set_so_error(result.release_error());
  219. }
  220. }
  221. if (flags & TCPFlags::ACK) {
  222. m_last_ack_number_sent = m_ack_number;
  223. m_last_ack_sent_time = TimeManagement::the().monotonic_time();
  224. tcp_packet.set_ack_number(m_ack_number);
  225. }
  226. if (flags & TCPFlags::SYN) {
  227. ++m_sequence_number;
  228. } else {
  229. m_sequence_number += payload_size;
  230. }
  231. if (has_mss_option) {
  232. u16 mss = routing_decision.adapter->mtu() - sizeof(IPv4Packet) - sizeof(TCPPacket);
  233. TCPOptionMSS mss_option { mss };
  234. VERIFY(packet->buffer->size() >= ipv4_payload_offset + sizeof(TCPPacket) + sizeof(mss_option));
  235. memcpy(packet->buffer->data() + ipv4_payload_offset + sizeof(TCPPacket), &mss_option, sizeof(mss_option));
  236. }
  237. tcp_packet.set_checksum(compute_tcp_checksum(local_address(), peer_address(), tcp_packet, payload_size));
  238. bool expect_ack { tcp_packet.has_syn() || payload_size > 0 };
  239. if (expect_ack) {
  240. bool append_failed { false };
  241. m_unacked_packets.with_exclusive([&](auto& unacked_packets) {
  242. auto result = unacked_packets.packets.try_append({ m_sequence_number, packet, ipv4_payload_offset, *routing_decision.adapter });
  243. if (result.is_error()) {
  244. dbgln("TCPSocket: Dropped outbound packet because try_append() failed");
  245. append_failed = true;
  246. return;
  247. }
  248. unacked_packets.size += payload_size;
  249. enqueue_for_retransmit();
  250. });
  251. if (append_failed)
  252. return set_so_error(ENOMEM);
  253. }
  254. m_packets_out++;
  255. m_bytes_out += buffer_size;
  256. routing_decision.adapter->send_packet(packet->bytes());
  257. if (!expect_ack)
  258. routing_decision.adapter->release_packet_buffer(*packet);
  259. return {};
  260. }
  261. void TCPSocket::receive_tcp_packet(TCPPacket const& packet, u16 size)
  262. {
  263. if (packet.has_ack()) {
  264. u32 ack_number = packet.ack_number();
  265. dbgln_if(TCP_SOCKET_DEBUG, "TCPSocket: receive_tcp_packet: {}", ack_number);
  266. int removed = 0;
  267. m_unacked_packets.with_exclusive([&](auto& unacked_packets) {
  268. while (!unacked_packets.packets.is_empty()) {
  269. auto& packet = unacked_packets.packets.first();
  270. dbgln_if(TCP_SOCKET_DEBUG, "TCPSocket: iterate: {}", packet.ack_number);
  271. if (packet.ack_number <= ack_number) {
  272. auto old_adapter = packet.adapter.strong_ref();
  273. if (old_adapter)
  274. old_adapter->release_packet_buffer(*packet.buffer);
  275. TCPPacket& tcp_packet = *(TCPPacket*)(packet.buffer->buffer->data() + packet.ipv4_payload_offset);
  276. if (m_send_window_size != tcp_packet.window_size()) {
  277. m_send_window_size = tcp_packet.window_size();
  278. }
  279. auto payload_size = packet.buffer->buffer->data() + packet.buffer->buffer->size() - (u8*)tcp_packet.payload();
  280. unacked_packets.size -= payload_size;
  281. evaluate_block_conditions();
  282. unacked_packets.packets.take_first();
  283. removed++;
  284. } else {
  285. break;
  286. }
  287. }
  288. if (unacked_packets.packets.is_empty()) {
  289. m_retransmit_attempts = 0;
  290. dequeue_for_retransmit();
  291. }
  292. dbgln_if(TCP_SOCKET_DEBUG, "TCPSocket: receive_tcp_packet acknowledged {} packets", removed);
  293. });
  294. }
  295. m_packets_in++;
  296. m_bytes_in += packet.header_size() + size;
  297. }
  298. bool TCPSocket::should_delay_next_ack() const
  299. {
  300. // FIXME: We don't know the MSS here so make a reasonable guess.
  301. const size_t mss = 1500;
  302. // RFC 1122 says we should send an ACK for every two full-sized segments.
  303. if (m_ack_number >= m_last_ack_number_sent + 2 * mss)
  304. return false;
  305. // RFC 1122 says we should not delay ACKs for more than 500 milliseconds.
  306. if (TimeManagement::the().monotonic_time(TimePrecision::Precise) >= m_last_ack_sent_time + Duration::from_milliseconds(500))
  307. return false;
  308. return true;
  309. }
  310. NetworkOrdered<u16> TCPSocket::compute_tcp_checksum(IPv4Address const& source, IPv4Address const& destination, TCPPacket const& packet, u16 payload_size)
  311. {
  312. union PseudoHeader {
  313. struct [[gnu::packed]] {
  314. IPv4Address source;
  315. IPv4Address destination;
  316. u8 zero;
  317. u8 protocol;
  318. NetworkOrdered<u16> payload_size;
  319. } header;
  320. u16 raw[6];
  321. };
  322. static_assert(sizeof(PseudoHeader) == 12);
  323. Checked<u16> packet_size = packet.header_size();
  324. packet_size += payload_size;
  325. VERIFY(!packet_size.has_overflow());
  326. PseudoHeader pseudo_header { .header = { source, destination, 0, (u8)IPv4Protocol::TCP, packet_size.value() } };
  327. u32 checksum = 0;
  328. auto* raw_pseudo_header = pseudo_header.raw;
  329. for (size_t i = 0; i < sizeof(pseudo_header) / sizeof(u16); ++i) {
  330. checksum += AK::convert_between_host_and_network_endian(raw_pseudo_header[i]);
  331. if (checksum > 0xffff)
  332. checksum = (checksum >> 16) + (checksum & 0xffff);
  333. }
  334. auto* raw_packet = bit_cast<u16*>(&packet);
  335. for (size_t i = 0; i < packet.header_size() / sizeof(u16); ++i) {
  336. checksum += AK::convert_between_host_and_network_endian(raw_packet[i]);
  337. if (checksum > 0xffff)
  338. checksum = (checksum >> 16) + (checksum & 0xffff);
  339. }
  340. VERIFY(packet.data_offset() * 4 == packet.header_size());
  341. auto* raw_payload = bit_cast<u16*>(packet.payload());
  342. for (size_t i = 0; i < payload_size / sizeof(u16); ++i) {
  343. checksum += AK::convert_between_host_and_network_endian(raw_payload[i]);
  344. if (checksum > 0xffff)
  345. checksum = (checksum >> 16) + (checksum & 0xffff);
  346. }
  347. if (payload_size & 1) {
  348. u16 expanded_byte = ((u8 const*)packet.payload())[payload_size - 1] << 8;
  349. checksum += expanded_byte;
  350. if (checksum > 0xffff)
  351. checksum = (checksum >> 16) + (checksum & 0xffff);
  352. }
  353. return ~(checksum & 0xffff);
  354. }
  355. ErrorOr<void> TCPSocket::protocol_bind()
  356. {
  357. dbgln_if(TCP_SOCKET_DEBUG, "TCPSocket::protocol_bind(), local_port() is {}", local_port());
  358. // Check that we do have the address we're trying to bind to.
  359. TRY(m_adapter.with([this](auto& adapter) -> ErrorOr<void> {
  360. if (has_specific_local_address() && !adapter) {
  361. adapter = NetworkingManagement::the().from_ipv4_address(local_address());
  362. if (!adapter)
  363. return set_so_error(EADDRNOTAVAIL);
  364. }
  365. return {};
  366. }));
  367. if (local_port() == 0) {
  368. // Allocate an unused ephemeral port.
  369. constexpr u16 first_ephemeral_port = 32768;
  370. constexpr u16 last_ephemeral_port = 60999;
  371. constexpr u16 ephemeral_port_range_size = last_ephemeral_port - first_ephemeral_port;
  372. u16 first_scan_port = first_ephemeral_port + get_good_random<u16>() % ephemeral_port_range_size;
  373. return sockets_by_tuple().with_exclusive([&](auto& table) -> ErrorOr<void> {
  374. u16 port = first_scan_port;
  375. while (true) {
  376. IPv4SocketTuple proposed_tuple(local_address(), port, peer_address(), peer_port());
  377. auto it = table.find(proposed_tuple);
  378. if (it == table.end()) {
  379. set_local_port(port);
  380. table.set(proposed_tuple, this);
  381. dbgln_if(TCP_SOCKET_DEBUG, "...allocated port {}, tuple {}", port, proposed_tuple.to_string());
  382. return {};
  383. }
  384. ++port;
  385. if (port > last_ephemeral_port)
  386. port = first_ephemeral_port;
  387. if (port == first_scan_port)
  388. break;
  389. }
  390. return set_so_error(EADDRINUSE);
  391. });
  392. } else {
  393. // Verify that the user-supplied port is not already used by someone else.
  394. bool ok = sockets_by_tuple().with_exclusive([&](auto& table) -> bool {
  395. if (table.contains(tuple()))
  396. return false;
  397. table.set(tuple(), this);
  398. return true;
  399. });
  400. if (!ok)
  401. return set_so_error(EADDRINUSE);
  402. return {};
  403. }
  404. }
  405. ErrorOr<void> TCPSocket::protocol_listen()
  406. {
  407. set_direction(Direction::Passive);
  408. set_state(State::Listen);
  409. set_setup_state(SetupState::Completed);
  410. return {};
  411. }
  412. ErrorOr<void> TCPSocket::protocol_connect(OpenFileDescription& description)
  413. {
  414. MutexLocker locker(mutex());
  415. auto routing_decision = route_to(peer_address(), local_address());
  416. if (routing_decision.is_zero())
  417. return set_so_error(EHOSTUNREACH);
  418. if (!has_specific_local_address())
  419. set_local_address(routing_decision.adapter->ipv4_address());
  420. TRY(ensure_bound());
  421. m_sequence_number = get_good_random<u32>();
  422. m_ack_number = 0;
  423. set_setup_state(SetupState::InProgress);
  424. TRY(send_tcp_packet(TCPFlags::SYN));
  425. m_state = State::SynSent;
  426. set_role(Role::Connecting);
  427. m_direction = Direction::Outgoing;
  428. evaluate_block_conditions();
  429. if (description.is_blocking()) {
  430. locker.unlock();
  431. auto unblock_flags = Thread::FileBlocker::BlockFlags::None;
  432. if (Thread::current()->block<Thread::ConnectBlocker>({}, description, unblock_flags).was_interrupted())
  433. return set_so_error(EINTR);
  434. locker.lock();
  435. VERIFY(setup_state() == SetupState::Completed);
  436. if (has_error()) { // TODO: check unblock_flags
  437. set_role(Role::None);
  438. if (error() == TCPSocket::Error::RetransmitTimeout)
  439. return set_so_error(ETIMEDOUT);
  440. else
  441. return set_so_error(ECONNREFUSED);
  442. }
  443. return {};
  444. }
  445. return set_so_error(EINPROGRESS);
  446. }
  447. bool TCPSocket::protocol_is_disconnected() const
  448. {
  449. switch (m_state) {
  450. case State::Closed:
  451. case State::CloseWait:
  452. case State::LastAck:
  453. case State::FinWait1:
  454. case State::FinWait2:
  455. case State::Closing:
  456. case State::TimeWait:
  457. return true;
  458. default:
  459. return false;
  460. }
  461. }
  462. void TCPSocket::shut_down_for_writing()
  463. {
  464. if (state() == State::Established) {
  465. dbgln_if(TCP_SOCKET_DEBUG, " Sending FIN from Established and moving into FinWait1");
  466. (void)send_tcp_packet(TCPFlags::FIN | TCPFlags::ACK);
  467. set_state(State::FinWait1);
  468. } else {
  469. dbgln(" Shutting down TCPSocket for writing but not moving to FinWait1 since state is {}", to_string(state()));
  470. }
  471. }
  472. ErrorOr<void> TCPSocket::close()
  473. {
  474. MutexLocker locker(mutex());
  475. auto result = IPv4Socket::close();
  476. if (state() == State::CloseWait) {
  477. dbgln_if(TCP_SOCKET_DEBUG, " Sending FIN from CloseWait and moving into LastAck");
  478. [[maybe_unused]] auto rc = send_tcp_packet(TCPFlags::FIN | TCPFlags::ACK);
  479. set_state(State::LastAck);
  480. }
  481. if (state() != State::Closed && state() != State::Listen)
  482. closing_sockets().with_exclusive([&](auto& table) {
  483. table.set(tuple(), *this);
  484. });
  485. return result;
  486. }
  487. static Singleton<MutexProtected<TCPSocket::RetransmitList>> s_sockets_for_retransmit;
  488. MutexProtected<TCPSocket::RetransmitList>& TCPSocket::sockets_for_retransmit()
  489. {
  490. return *s_sockets_for_retransmit;
  491. }
  492. void TCPSocket::enqueue_for_retransmit()
  493. {
  494. sockets_for_retransmit().with_exclusive([&](auto& list) {
  495. list.append(*this);
  496. });
  497. }
  498. void TCPSocket::dequeue_for_retransmit()
  499. {
  500. sockets_for_retransmit().with_exclusive([&](auto& list) {
  501. list.remove(*this);
  502. });
  503. }
  504. void TCPSocket::retransmit_packets()
  505. {
  506. auto now = TimeManagement::the().monotonic_time();
  507. // RFC6298 says we should have at least one second between retransmits. According to
  508. // RFC1122 we must do exponential backoff - even for SYN packets.
  509. i64 retransmit_interval = 1;
  510. for (decltype(m_retransmit_attempts) i = 0; i < m_retransmit_attempts; i++)
  511. retransmit_interval *= 2;
  512. if (m_last_retransmit_time > now - Duration::from_seconds(retransmit_interval))
  513. return;
  514. dbgln_if(TCP_SOCKET_DEBUG, "TCPSocket({}) handling retransmit", this);
  515. m_last_retransmit_time = now;
  516. ++m_retransmit_attempts;
  517. if (m_retransmit_attempts > maximum_retransmits) {
  518. set_state(TCPSocket::State::Closed);
  519. set_error(TCPSocket::Error::RetransmitTimeout);
  520. set_setup_state(Socket::SetupState::Completed);
  521. return;
  522. }
  523. auto adapter = bound_interface().with([](auto& bound_device) -> RefPtr<NetworkAdapter> { return bound_device; });
  524. auto routing_decision = route_to(peer_address(), local_address(), adapter);
  525. if (routing_decision.is_zero())
  526. return;
  527. m_unacked_packets.with_exclusive([&](auto& unacked_packets) {
  528. for (auto& packet : unacked_packets.packets) {
  529. packet.tx_counter++;
  530. if constexpr (TCP_SOCKET_DEBUG) {
  531. auto& tcp_packet = *(const TCPPacket*)(packet.buffer->buffer->data() + packet.ipv4_payload_offset);
  532. dbgln("Sending TCP packet from {}:{} to {}:{} with ({}{}{}{}) seq_no={}, ack_no={}, tx_counter={}",
  533. local_address(), local_port(),
  534. peer_address(), peer_port(),
  535. (tcp_packet.has_syn() ? "SYN " : ""),
  536. (tcp_packet.has_ack() ? "ACK " : ""),
  537. (tcp_packet.has_fin() ? "FIN " : ""),
  538. (tcp_packet.has_rst() ? "RST " : ""),
  539. tcp_packet.sequence_number(),
  540. tcp_packet.ack_number(),
  541. packet.tx_counter);
  542. }
  543. size_t ipv4_payload_offset = routing_decision.adapter->ipv4_payload_offset();
  544. if (ipv4_payload_offset != packet.ipv4_payload_offset) {
  545. // FIXME: Add support for this. This can happen if after a route change
  546. // we ended up on another adapter which doesn't have the same layer 2 type
  547. // like the previous adapter.
  548. VERIFY_NOT_REACHED();
  549. }
  550. auto packet_buffer = packet.buffer->bytes();
  551. routing_decision.adapter->fill_in_ipv4_header(*packet.buffer,
  552. local_address(), routing_decision.next_hop, peer_address(),
  553. IPv4Protocol::TCP, packet_buffer.size() - ipv4_payload_offset, type_of_service(), ttl());
  554. routing_decision.adapter->send_packet(packet_buffer);
  555. m_packets_out++;
  556. m_bytes_out += packet_buffer.size();
  557. }
  558. });
  559. }
  560. bool TCPSocket::can_write(OpenFileDescription const& file_description, u64 size) const
  561. {
  562. if (!IPv4Socket::can_write(file_description, size))
  563. return false;
  564. if (m_state == State::SynSent || m_state == State::SynReceived)
  565. return false;
  566. if (!file_description.is_blocking())
  567. return true;
  568. return m_unacked_packets.with_shared([&](auto& unacked_packets) {
  569. return unacked_packets.size + size <= m_send_window_size;
  570. });
  571. }
  572. }