diff options
| author | 2022-07-30 05:58:23 +0200 | |
|---|---|---|
| committer | 2022-08-15 20:25:42 +0200 | |
| commit | f80c7c4cd5c090b9a31f89a0eb3d86cbe928c50b (patch) | |
| tree | ad359908ba2d3cd003082b39cc7217b61e5b18f6 /src/core | |
| parent | web_service: Correct jwt issuer string (diff) | |
| download | yuzu-f80c7c4cd5c090b9a31f89a0eb3d86cbe928c50b.tar.gz yuzu-f80c7c4cd5c090b9a31f89a0eb3d86cbe928c50b.tar.xz yuzu-f80c7c4cd5c090b9a31f89a0eb3d86cbe928c50b.zip | |
core, network: Add ability to proxy socket packets
Diffstat (limited to 'src/core')
| -rw-r--r-- | src/core/CMakeLists.txt | 2 | ||||
| -rw-r--r-- | src/core/hle/service/nifm/nifm.cpp | 341 | ||||
| -rw-r--r-- | src/core/hle/service/nifm/nifm.h | 27 | ||||
| -rw-r--r-- | src/core/hle/service/sockets/bsd.cpp | 40 | ||||
| -rw-r--r-- | src/core/hle/service/sockets/bsd.h | 13 | ||||
| -rw-r--r-- | src/core/hle/service/sockets/sockets.h | 6 | ||||
| -rw-r--r-- | src/core/hle/service/sockets/sockets_translate.cpp | 2 | ||||
| -rw-r--r-- | src/core/internal_network/network.cpp | 61 | ||||
| -rw-r--r-- | src/core/internal_network/network.h | 43 | ||||
| -rw-r--r-- | src/core/internal_network/socket_proxy.cpp | 282 | ||||
| -rw-r--r-- | src/core/internal_network/socket_proxy.h | 102 | ||||
| -rw-r--r-- | src/core/internal_network/sockets.h | 136 |
12 files changed, 783 insertions, 272 deletions
diff --git a/src/core/CMakeLists.txt b/src/core/CMakeLists.txt index 4e39649a8..3230d7199 100644 --- a/src/core/CMakeLists.txt +++ b/src/core/CMakeLists.txt | |||
| @@ -724,6 +724,8 @@ add_library(core STATIC | |||
| 724 | internal_network/network_interface.cpp | 724 | internal_network/network_interface.cpp |
| 725 | internal_network/network_interface.h | 725 | internal_network/network_interface.h |
| 726 | internal_network/sockets.h | 726 | internal_network/sockets.h |
| 727 | internal_network/socket_proxy.cpp | ||
| 728 | internal_network/socket_proxy.h | ||
| 727 | loader/deconstructed_rom_directory.cpp | 729 | loader/deconstructed_rom_directory.cpp |
| 728 | loader/deconstructed_rom_directory.h | 730 | loader/deconstructed_rom_directory.h |
| 729 | loader/kip.cpp | 731 | loader/kip.cpp |
diff --git a/src/core/hle/service/nifm/nifm.cpp b/src/core/hle/service/nifm/nifm.cpp index 2889973e4..42ed17187 100644 --- a/src/core/hle/service/nifm/nifm.cpp +++ b/src/core/hle/service/nifm/nifm.cpp | |||
| @@ -6,7 +6,6 @@ | |||
| 6 | #include "core/hle/kernel/k_event.h" | 6 | #include "core/hle/kernel/k_event.h" |
| 7 | #include "core/hle/service/kernel_helpers.h" | 7 | #include "core/hle/service/kernel_helpers.h" |
| 8 | #include "core/hle/service/nifm/nifm.h" | 8 | #include "core/hle/service/nifm/nifm.h" |
| 9 | #include "core/hle/service/service.h" | ||
| 10 | 9 | ||
| 11 | namespace { | 10 | namespace { |
| 12 | 11 | ||
| @@ -271,213 +270,227 @@ public: | |||
| 271 | } | 270 | } |
| 272 | }; | 271 | }; |
| 273 | 272 | ||
| 274 | class IGeneralService final : public ServiceFramework<IGeneralService> { | 273 | void IGeneralService::GetClientId(Kernel::HLERequestContext& ctx) { |
| 275 | public: | 274 | static constexpr u32 client_id = 1; |
| 276 | explicit IGeneralService(Core::System& system_); | 275 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 277 | 276 | ||
| 278 | private: | 277 | IPC::ResponseBuilder rb{ctx, 4}; |
| 279 | void GetClientId(Kernel::HLERequestContext& ctx) { | 278 | rb.Push(ResultSuccess); |
| 280 | static constexpr u32 client_id = 1; | 279 | rb.Push<u64>(client_id); // Client ID needs to be non zero otherwise it's considered invalid |
| 281 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 280 | } |
| 282 | 281 | ||
| 283 | IPC::ResponseBuilder rb{ctx, 4}; | 282 | void IGeneralService::CreateScanRequest(Kernel::HLERequestContext& ctx) { |
| 284 | rb.Push(ResultSuccess); | 283 | LOG_DEBUG(Service_NIFM, "called"); |
| 285 | rb.Push<u64>(client_id); // Client ID needs to be non zero otherwise it's considered invalid | ||
| 286 | } | ||
| 287 | 284 | ||
| 288 | void CreateScanRequest(Kernel::HLERequestContext& ctx) { | 285 | IPC::ResponseBuilder rb{ctx, 2, 0, 1}; |
| 289 | LOG_DEBUG(Service_NIFM, "called"); | ||
| 290 | 286 | ||
| 291 | IPC::ResponseBuilder rb{ctx, 2, 0, 1}; | 287 | rb.Push(ResultSuccess); |
| 288 | rb.PushIpcInterface<IScanRequest>(system); | ||
| 289 | } | ||
| 292 | 290 | ||
| 293 | rb.Push(ResultSuccess); | 291 | void IGeneralService::CreateRequest(Kernel::HLERequestContext& ctx) { |
| 294 | rb.PushIpcInterface<IScanRequest>(system); | 292 | LOG_DEBUG(Service_NIFM, "called"); |
| 295 | } | ||
| 296 | 293 | ||
| 297 | void CreateRequest(Kernel::HLERequestContext& ctx) { | 294 | IPC::ResponseBuilder rb{ctx, 2, 0, 1}; |
| 298 | LOG_DEBUG(Service_NIFM, "called"); | ||
| 299 | 295 | ||
| 300 | IPC::ResponseBuilder rb{ctx, 2, 0, 1}; | 296 | rb.Push(ResultSuccess); |
| 297 | rb.PushIpcInterface<IRequest>(system); | ||
| 298 | } | ||
| 301 | 299 | ||
| 302 | rb.Push(ResultSuccess); | 300 | void IGeneralService::GetCurrentNetworkProfile(Kernel::HLERequestContext& ctx) { |
| 303 | rb.PushIpcInterface<IRequest>(system); | 301 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 304 | } | ||
| 305 | 302 | ||
| 306 | void GetCurrentNetworkProfile(Kernel::HLERequestContext& ctx) { | 303 | const auto net_iface = Network::GetSelectedNetworkInterface(); |
| 307 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | ||
| 308 | 304 | ||
| 309 | const auto net_iface = Network::GetSelectedNetworkInterface(); | 305 | SfNetworkProfileData network_profile_data = [&net_iface] { |
| 310 | 306 | if (!net_iface) { | |
| 311 | const SfNetworkProfileData network_profile_data = [&net_iface] { | 307 | return SfNetworkProfileData{}; |
| 312 | if (!net_iface) { | 308 | } |
| 313 | return SfNetworkProfileData{}; | 309 | |
| 314 | } | 310 | return SfNetworkProfileData{ |
| 315 | 311 | .ip_setting_data{ | |
| 316 | return SfNetworkProfileData{ | 312 | .ip_address_setting{ |
| 317 | .ip_setting_data{ | 313 | .is_automatic{true}, |
| 318 | .ip_address_setting{ | 314 | .current_address{Network::TranslateIPv4(net_iface->ip_address)}, |
| 319 | .is_automatic{true}, | 315 | .subnet_mask{Network::TranslateIPv4(net_iface->subnet_mask)}, |
| 320 | .current_address{Network::TranslateIPv4(net_iface->ip_address)}, | 316 | .gateway{Network::TranslateIPv4(net_iface->gateway)}, |
| 321 | .subnet_mask{Network::TranslateIPv4(net_iface->subnet_mask)}, | ||
| 322 | .gateway{Network::TranslateIPv4(net_iface->gateway)}, | ||
| 323 | }, | ||
| 324 | .dns_setting{ | ||
| 325 | .is_automatic{true}, | ||
| 326 | .primary_dns{1, 1, 1, 1}, | ||
| 327 | .secondary_dns{1, 0, 0, 1}, | ||
| 328 | }, | ||
| 329 | .proxy_setting{ | ||
| 330 | .enabled{false}, | ||
| 331 | .port{}, | ||
| 332 | .proxy_server{}, | ||
| 333 | .automatic_auth_enabled{}, | ||
| 334 | .user{}, | ||
| 335 | .password{}, | ||
| 336 | }, | ||
| 337 | .mtu{1500}, | ||
| 338 | }, | 317 | }, |
| 339 | .uuid{0xdeadbeef, 0xdeadbeef}, | 318 | .dns_setting{ |
| 340 | .network_name{"yuzu Network"}, | 319 | .is_automatic{true}, |
| 341 | .wireless_setting_data{ | 320 | .primary_dns{1, 1, 1, 1}, |
| 342 | .ssid_length{12}, | 321 | .secondary_dns{1, 0, 0, 1}, |
| 343 | .ssid{"yuzu Network"}, | ||
| 344 | .passphrase{"yuzupassword"}, | ||
| 345 | }, | 322 | }, |
| 346 | }; | 323 | .proxy_setting{ |
| 347 | }(); | 324 | .enabled{false}, |
| 348 | 325 | .port{}, | |
| 349 | ctx.WriteBuffer(network_profile_data); | 326 | .proxy_server{}, |
| 327 | .automatic_auth_enabled{}, | ||
| 328 | .user{}, | ||
| 329 | .password{}, | ||
| 330 | }, | ||
| 331 | .mtu{1500}, | ||
| 332 | }, | ||
| 333 | .uuid{0xdeadbeef, 0xdeadbeef}, | ||
| 334 | .network_name{"yuzu Network"}, | ||
| 335 | .wireless_setting_data{ | ||
| 336 | .ssid_length{12}, | ||
| 337 | .ssid{"yuzu Network"}, | ||
| 338 | .passphrase{"yuzupassword"}, | ||
| 339 | }, | ||
| 340 | }; | ||
| 341 | }(); | ||
| 350 | 342 | ||
| 351 | IPC::ResponseBuilder rb{ctx, 2}; | 343 | // When we're connected to a room, spoof the hosts IP address |
| 352 | rb.Push(ResultSuccess); | 344 | if (auto room_member = network.GetRoomMember().lock()) { |
| 345 | if (room_member->IsConnected()) { | ||
| 346 | network_profile_data.ip_setting_data.ip_address_setting.current_address = | ||
| 347 | room_member->GetFakeIpAddress(); | ||
| 348 | } | ||
| 353 | } | 349 | } |
| 354 | 350 | ||
| 355 | void RemoveNetworkProfile(Kernel::HLERequestContext& ctx) { | 351 | ctx.WriteBuffer(network_profile_data); |
| 356 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | ||
| 357 | 352 | ||
| 358 | IPC::ResponseBuilder rb{ctx, 2}; | 353 | IPC::ResponseBuilder rb{ctx, 2}; |
| 359 | rb.Push(ResultSuccess); | 354 | rb.Push(ResultSuccess); |
| 360 | } | 355 | } |
| 361 | 356 | ||
| 362 | void GetCurrentIpAddress(Kernel::HLERequestContext& ctx) { | 357 | void IGeneralService::RemoveNetworkProfile(Kernel::HLERequestContext& ctx) { |
| 363 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 358 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 364 | 359 | ||
| 365 | auto ipv4 = Network::GetHostIPv4Address(); | 360 | IPC::ResponseBuilder rb{ctx, 2}; |
| 366 | if (!ipv4) { | 361 | rb.Push(ResultSuccess); |
| 367 | LOG_ERROR(Service_NIFM, "Couldn't get host IPv4 address, defaulting to 0.0.0.0"); | 362 | } |
| 368 | ipv4.emplace(Network::IPv4Address{0, 0, 0, 0}); | ||
| 369 | } | ||
| 370 | 363 | ||
| 371 | IPC::ResponseBuilder rb{ctx, 3}; | 364 | void IGeneralService::GetCurrentIpAddress(Kernel::HLERequestContext& ctx) { |
| 372 | rb.Push(ResultSuccess); | 365 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 373 | rb.PushRaw(*ipv4); | 366 | |
| 367 | auto ipv4 = Network::GetHostIPv4Address(); | ||
| 368 | if (!ipv4) { | ||
| 369 | LOG_ERROR(Service_NIFM, "Couldn't get host IPv4 address, defaulting to 0.0.0.0"); | ||
| 370 | ipv4.emplace(Network::IPv4Address{0, 0, 0, 0}); | ||
| 374 | } | 371 | } |
| 375 | 372 | ||
| 376 | void CreateTemporaryNetworkProfile(Kernel::HLERequestContext& ctx) { | 373 | // When we're connected to a room, spoof the hosts IP address |
| 377 | LOG_DEBUG(Service_NIFM, "called"); | 374 | if (auto room_member = network.GetRoomMember().lock()) { |
| 375 | if (room_member->IsConnected()) { | ||
| 376 | ipv4 = room_member->GetFakeIpAddress(); | ||
| 377 | } | ||
| 378 | } | ||
| 378 | 379 | ||
| 379 | ASSERT_MSG(ctx.GetReadBufferSize() == 0x17c, | 380 | IPC::ResponseBuilder rb{ctx, 3}; |
| 380 | "SfNetworkProfileData is not the correct size"); | 381 | rb.Push(ResultSuccess); |
| 381 | u128 uuid{}; | 382 | rb.PushRaw(*ipv4); |
| 382 | auto buffer = ctx.ReadBuffer(); | 383 | } |
| 383 | std::memcpy(&uuid, buffer.data() + 8, sizeof(u128)); | 384 | void IGeneralService::CreateTemporaryNetworkProfile(Kernel::HLERequestContext& ctx) { |
| 385 | LOG_DEBUG(Service_NIFM, "called"); | ||
| 384 | 386 | ||
| 385 | IPC::ResponseBuilder rb{ctx, 6, 0, 1}; | 387 | ASSERT_MSG(ctx.GetReadBufferSize() == 0x17c, "SfNetworkProfileData is not the correct size"); |
| 388 | u128 uuid{}; | ||
| 389 | auto buffer = ctx.ReadBuffer(); | ||
| 390 | std::memcpy(&uuid, buffer.data() + 8, sizeof(u128)); | ||
| 386 | 391 | ||
| 387 | rb.Push(ResultSuccess); | 392 | IPC::ResponseBuilder rb{ctx, 6, 0, 1}; |
| 388 | rb.PushIpcInterface<INetworkProfile>(system); | ||
| 389 | rb.PushRaw<u128>(uuid); | ||
| 390 | } | ||
| 391 | 393 | ||
| 392 | void GetCurrentIpConfigInfo(Kernel::HLERequestContext& ctx) { | 394 | rb.Push(ResultSuccess); |
| 393 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 395 | rb.PushIpcInterface<INetworkProfile>(system); |
| 396 | rb.PushRaw<u128>(uuid); | ||
| 397 | } | ||
| 394 | 398 | ||
| 395 | struct IpConfigInfo { | 399 | void IGeneralService::GetCurrentIpConfigInfo(Kernel::HLERequestContext& ctx) { |
| 396 | IpAddressSetting ip_address_setting{}; | 400 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 397 | DnsSetting dns_setting{}; | ||
| 398 | }; | ||
| 399 | static_assert(sizeof(IpConfigInfo) == sizeof(IpAddressSetting) + sizeof(DnsSetting), | ||
| 400 | "IpConfigInfo has incorrect size."); | ||
| 401 | 401 | ||
| 402 | const auto net_iface = Network::GetSelectedNetworkInterface(); | 402 | struct IpConfigInfo { |
| 403 | IpAddressSetting ip_address_setting{}; | ||
| 404 | DnsSetting dns_setting{}; | ||
| 405 | }; | ||
| 406 | static_assert(sizeof(IpConfigInfo) == sizeof(IpAddressSetting) + sizeof(DnsSetting), | ||
| 407 | "IpConfigInfo has incorrect size."); | ||
| 403 | 408 | ||
| 404 | const IpConfigInfo ip_config_info = [&net_iface] { | 409 | const auto net_iface = Network::GetSelectedNetworkInterface(); |
| 405 | if (!net_iface) { | ||
| 406 | return IpConfigInfo{}; | ||
| 407 | } | ||
| 408 | 410 | ||
| 409 | return IpConfigInfo{ | 411 | IpConfigInfo ip_config_info = [&net_iface] { |
| 410 | .ip_address_setting{ | 412 | if (!net_iface) { |
| 411 | .is_automatic{true}, | 413 | return IpConfigInfo{}; |
| 412 | .current_address{Network::TranslateIPv4(net_iface->ip_address)}, | 414 | } |
| 413 | .subnet_mask{Network::TranslateIPv4(net_iface->subnet_mask)}, | ||
| 414 | .gateway{Network::TranslateIPv4(net_iface->gateway)}, | ||
| 415 | }, | ||
| 416 | .dns_setting{ | ||
| 417 | .is_automatic{true}, | ||
| 418 | .primary_dns{1, 1, 1, 1}, | ||
| 419 | .secondary_dns{1, 0, 0, 1}, | ||
| 420 | }, | ||
| 421 | }; | ||
| 422 | }(); | ||
| 423 | 415 | ||
| 424 | IPC::ResponseBuilder rb{ctx, 2 + (sizeof(IpConfigInfo) + 3) / sizeof(u32)}; | 416 | return IpConfigInfo{ |
| 425 | rb.Push(ResultSuccess); | 417 | .ip_address_setting{ |
| 426 | rb.PushRaw<IpConfigInfo>(ip_config_info); | 418 | .is_automatic{true}, |
| 419 | .current_address{Network::TranslateIPv4(net_iface->ip_address)}, | ||
| 420 | .subnet_mask{Network::TranslateIPv4(net_iface->subnet_mask)}, | ||
| 421 | .gateway{Network::TranslateIPv4(net_iface->gateway)}, | ||
| 422 | }, | ||
| 423 | .dns_setting{ | ||
| 424 | .is_automatic{true}, | ||
| 425 | .primary_dns{1, 1, 1, 1}, | ||
| 426 | .secondary_dns{1, 0, 0, 1}, | ||
| 427 | }, | ||
| 428 | }; | ||
| 429 | }(); | ||
| 430 | |||
| 431 | // When we're connected to a room, spoof the hosts IP address | ||
| 432 | if (auto room_member = network.GetRoomMember().lock()) { | ||
| 433 | if (room_member->IsConnected()) { | ||
| 434 | ip_config_info.ip_address_setting.current_address = room_member->GetFakeIpAddress(); | ||
| 435 | } | ||
| 427 | } | 436 | } |
| 428 | 437 | ||
| 429 | void IsWirelessCommunicationEnabled(Kernel::HLERequestContext& ctx) { | 438 | IPC::ResponseBuilder rb{ctx, 2 + (sizeof(IpConfigInfo) + 3) / sizeof(u32)}; |
| 430 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 439 | rb.Push(ResultSuccess); |
| 440 | rb.PushRaw<IpConfigInfo>(ip_config_info); | ||
| 441 | } | ||
| 431 | 442 | ||
| 432 | IPC::ResponseBuilder rb{ctx, 3}; | 443 | void IGeneralService::IsWirelessCommunicationEnabled(Kernel::HLERequestContext& ctx) { |
| 433 | rb.Push(ResultSuccess); | 444 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 434 | rb.Push<u8>(0); | ||
| 435 | } | ||
| 436 | 445 | ||
| 437 | void GetInternetConnectionStatus(Kernel::HLERequestContext& ctx) { | 446 | IPC::ResponseBuilder rb{ctx, 3}; |
| 438 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 447 | rb.Push(ResultSuccess); |
| 448 | rb.Push<u8>(1); | ||
| 449 | } | ||
| 439 | 450 | ||
| 440 | struct Output { | 451 | void IGeneralService::GetInternetConnectionStatus(Kernel::HLERequestContext& ctx) { |
| 441 | InternetConnectionType type{InternetConnectionType::WiFi}; | 452 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 442 | u8 wifi_strength{3}; | ||
| 443 | InternetConnectionStatus state{InternetConnectionStatus::Connected}; | ||
| 444 | }; | ||
| 445 | static_assert(sizeof(Output) == 0x3, "Output has incorrect size."); | ||
| 446 | 453 | ||
| 447 | constexpr Output out{}; | 454 | struct Output { |
| 455 | InternetConnectionType type{InternetConnectionType::WiFi}; | ||
| 456 | u8 wifi_strength{3}; | ||
| 457 | InternetConnectionStatus state{InternetConnectionStatus::Connected}; | ||
| 458 | }; | ||
| 459 | static_assert(sizeof(Output) == 0x3, "Output has incorrect size."); | ||
| 448 | 460 | ||
| 449 | IPC::ResponseBuilder rb{ctx, 3}; | 461 | constexpr Output out{}; |
| 450 | rb.Push(ResultSuccess); | ||
| 451 | rb.PushRaw(out); | ||
| 452 | } | ||
| 453 | 462 | ||
| 454 | void IsEthernetCommunicationEnabled(Kernel::HLERequestContext& ctx) { | 463 | IPC::ResponseBuilder rb{ctx, 3}; |
| 455 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 464 | rb.Push(ResultSuccess); |
| 465 | rb.PushRaw(out); | ||
| 466 | } | ||
| 456 | 467 | ||
| 457 | IPC::ResponseBuilder rb{ctx, 3}; | 468 | void IGeneralService::IsEthernetCommunicationEnabled(Kernel::HLERequestContext& ctx) { |
| 458 | rb.Push(ResultSuccess); | 469 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); |
| 459 | if (Network::GetHostIPv4Address().has_value()) { | 470 | |
| 460 | rb.Push<u8>(1); | 471 | IPC::ResponseBuilder rb{ctx, 3}; |
| 461 | } else { | 472 | rb.Push(ResultSuccess); |
| 462 | rb.Push<u8>(0); | 473 | if (Network::GetHostIPv4Address().has_value()) { |
| 463 | } | 474 | rb.Push<u8>(1); |
| 475 | } else { | ||
| 476 | rb.Push<u8>(0); | ||
| 464 | } | 477 | } |
| 478 | } | ||
| 465 | 479 | ||
| 466 | void IsAnyInternetRequestAccepted(Kernel::HLERequestContext& ctx) { | 480 | void IGeneralService::IsAnyInternetRequestAccepted(Kernel::HLERequestContext& ctx) { |
| 467 | LOG_WARNING(Service_NIFM, "(STUBBED) called"); | 481 | LOG_ERROR(Service_NIFM, "(STUBBED) called"); |
| 468 | 482 | ||
| 469 | IPC::ResponseBuilder rb{ctx, 3}; | 483 | IPC::ResponseBuilder rb{ctx, 3}; |
| 470 | rb.Push(ResultSuccess); | 484 | rb.Push(ResultSuccess); |
| 471 | if (Network::GetHostIPv4Address().has_value()) { | 485 | if (Network::GetHostIPv4Address().has_value()) { |
| 472 | rb.Push<u8>(1); | 486 | rb.Push<u8>(1); |
| 473 | } else { | 487 | } else { |
| 474 | rb.Push<u8>(0); | 488 | rb.Push<u8>(0); |
| 475 | } | ||
| 476 | } | 489 | } |
| 477 | }; | 490 | } |
| 478 | 491 | ||
| 479 | IGeneralService::IGeneralService(Core::System& system_) | 492 | IGeneralService::IGeneralService(Core::System& system_) |
| 480 | : ServiceFramework{system_, "IGeneralService"} { | 493 | : ServiceFramework{system_, "IGeneralService"}, network{system_.GetRoomNetwork()} { |
| 481 | // clang-format off | 494 | // clang-format off |
| 482 | static const FunctionInfo functions[] = { | 495 | static const FunctionInfo functions[] = { |
| 483 | {1, &IGeneralService::GetClientId, "GetClientId"}, | 496 | {1, &IGeneralService::GetClientId, "GetClientId"}, |
| @@ -528,6 +541,8 @@ IGeneralService::IGeneralService(Core::System& system_) | |||
| 528 | RegisterHandlers(functions); | 541 | RegisterHandlers(functions); |
| 529 | } | 542 | } |
| 530 | 543 | ||
| 544 | IGeneralService::~IGeneralService() = default; | ||
| 545 | |||
| 531 | class NetworkInterface final : public ServiceFramework<NetworkInterface> { | 546 | class NetworkInterface final : public ServiceFramework<NetworkInterface> { |
| 532 | public: | 547 | public: |
| 533 | explicit NetworkInterface(const char* name, Core::System& system_) | 548 | explicit NetworkInterface(const char* name, Core::System& system_) |
diff --git a/src/core/hle/service/nifm/nifm.h b/src/core/hle/service/nifm/nifm.h index 5f62d0014..48161be28 100644 --- a/src/core/hle/service/nifm/nifm.h +++ b/src/core/hle/service/nifm/nifm.h | |||
| @@ -3,6 +3,11 @@ | |||
| 3 | 3 | ||
| 4 | #pragma once | 4 | #pragma once |
| 5 | 5 | ||
| 6 | #include "core/hle/service/service.h" | ||
| 7 | #include "network/network.h" | ||
| 8 | #include "network/room.h" | ||
| 9 | #include "network/room_member.h" | ||
| 10 | |||
| 6 | namespace Core { | 11 | namespace Core { |
| 7 | class System; | 12 | class System; |
| 8 | } | 13 | } |
| @@ -16,4 +21,26 @@ namespace Service::NIFM { | |||
| 16 | /// Registers all NIFM services with the specified service manager. | 21 | /// Registers all NIFM services with the specified service manager. |
| 17 | void InstallInterfaces(SM::ServiceManager& service_manager, Core::System& system); | 22 | void InstallInterfaces(SM::ServiceManager& service_manager, Core::System& system); |
| 18 | 23 | ||
| 24 | class IGeneralService final : public ServiceFramework<IGeneralService> { | ||
| 25 | public: | ||
| 26 | explicit IGeneralService(Core::System& system_); | ||
| 27 | ~IGeneralService() override; | ||
| 28 | |||
| 29 | private: | ||
| 30 | void GetClientId(Kernel::HLERequestContext& ctx); | ||
| 31 | void CreateScanRequest(Kernel::HLERequestContext& ctx); | ||
| 32 | void CreateRequest(Kernel::HLERequestContext& ctx); | ||
| 33 | void GetCurrentNetworkProfile(Kernel::HLERequestContext& ctx); | ||
| 34 | void RemoveNetworkProfile(Kernel::HLERequestContext& ctx); | ||
| 35 | void GetCurrentIpAddress(Kernel::HLERequestContext& ctx); | ||
| 36 | void CreateTemporaryNetworkProfile(Kernel::HLERequestContext& ctx); | ||
| 37 | void GetCurrentIpConfigInfo(Kernel::HLERequestContext& ctx); | ||
| 38 | void IsWirelessCommunicationEnabled(Kernel::HLERequestContext& ctx); | ||
| 39 | void GetInternetConnectionStatus(Kernel::HLERequestContext& ctx); | ||
| 40 | void IsEthernetCommunicationEnabled(Kernel::HLERequestContext& ctx); | ||
| 41 | void IsAnyInternetRequestAccepted(Kernel::HLERequestContext& ctx); | ||
| 42 | |||
| 43 | Network::RoomNetwork& network; | ||
| 44 | }; | ||
| 45 | |||
| 19 | } // namespace Service::NIFM | 46 | } // namespace Service::NIFM |
diff --git a/src/core/hle/service/sockets/bsd.cpp b/src/core/hle/service/sockets/bsd.cpp index c7194731e..e08c3cb67 100644 --- a/src/core/hle/service/sockets/bsd.cpp +++ b/src/core/hle/service/sockets/bsd.cpp | |||
| @@ -9,12 +9,16 @@ | |||
| 9 | #include <fmt/format.h> | 9 | #include <fmt/format.h> |
| 10 | 10 | ||
| 11 | #include "common/microprofile.h" | 11 | #include "common/microprofile.h" |
| 12 | #include "common/socket_types.h" | ||
| 13 | #include "core/core.h" | ||
| 12 | #include "core/hle/ipc_helpers.h" | 14 | #include "core/hle/ipc_helpers.h" |
| 13 | #include "core/hle/kernel/k_thread.h" | 15 | #include "core/hle/kernel/k_thread.h" |
| 14 | #include "core/hle/service/sockets/bsd.h" | 16 | #include "core/hle/service/sockets/bsd.h" |
| 15 | #include "core/hle/service/sockets/sockets_translate.h" | 17 | #include "core/hle/service/sockets/sockets_translate.h" |
| 16 | #include "core/internal_network/network.h" | 18 | #include "core/internal_network/network.h" |
| 19 | #include "core/internal_network/socket_proxy.h" | ||
| 17 | #include "core/internal_network/sockets.h" | 20 | #include "core/internal_network/sockets.h" |
| 21 | #include "network/network.h" | ||
| 18 | 22 | ||
| 19 | namespace Service::Sockets { | 23 | namespace Service::Sockets { |
| 20 | 24 | ||
| @@ -472,7 +476,13 @@ std::pair<s32, Errno> BSD::SocketImpl(Domain domain, Type type, Protocol protoco | |||
| 472 | 476 | ||
| 473 | LOG_INFO(Service, "New socket fd={}", fd); | 477 | LOG_INFO(Service, "New socket fd={}", fd); |
| 474 | 478 | ||
| 475 | descriptor.socket = std::make_unique<Network::Socket>(); | 479 | auto room_member = room_network.GetRoomMember().lock(); |
| 480 | if (room_member && room_member->IsConnected()) { | ||
| 481 | descriptor.socket = std::make_unique<Network::ProxySocket>(room_network); | ||
| 482 | } else { | ||
| 483 | descriptor.socket = std::make_unique<Network::Socket>(); | ||
| 484 | } | ||
| 485 | |||
| 476 | descriptor.socket->Initialize(Translate(domain), Translate(type), Translate(type, protocol)); | 486 | descriptor.socket->Initialize(Translate(domain), Translate(type), Translate(type, protocol)); |
| 477 | descriptor.is_connection_based = IsConnectionBased(type); | 487 | descriptor.is_connection_based = IsConnectionBased(type); |
| 478 | 488 | ||
| @@ -648,7 +658,7 @@ std::pair<s32, Errno> BSD::FcntlImpl(s32 fd, FcntlCmd cmd, s32 arg) { | |||
| 648 | ASSERT(arg == 0); | 658 | ASSERT(arg == 0); |
| 649 | return {descriptor.flags, Errno::SUCCESS}; | 659 | return {descriptor.flags, Errno::SUCCESS}; |
| 650 | case FcntlCmd::SETFL: { | 660 | case FcntlCmd::SETFL: { |
| 651 | const bool enable = (arg & FLAG_O_NONBLOCK) != 0; | 661 | const bool enable = (arg & Network::FLAG_O_NONBLOCK) != 0; |
| 652 | const Errno bsd_errno = Translate(descriptor.socket->SetNonBlock(enable)); | 662 | const Errno bsd_errno = Translate(descriptor.socket->SetNonBlock(enable)); |
| 653 | if (bsd_errno != Errno::SUCCESS) { | 663 | if (bsd_errno != Errno::SUCCESS) { |
| 654 | return {-1, bsd_errno}; | 664 | return {-1, bsd_errno}; |
| @@ -669,7 +679,7 @@ Errno BSD::SetSockOptImpl(s32 fd, u32 level, OptName optname, size_t optlen, con | |||
| 669 | return Errno::BADF; | 679 | return Errno::BADF; |
| 670 | } | 680 | } |
| 671 | 681 | ||
| 672 | Network::Socket* const socket = file_descriptors[fd]->socket.get(); | 682 | Network::SocketBase* const socket = file_descriptors[fd]->socket.get(); |
| 673 | 683 | ||
| 674 | if (optname == OptName::LINGER) { | 684 | if (optname == OptName::LINGER) { |
| 675 | ASSERT(optlen == sizeof(Linger)); | 685 | ASSERT(optlen == sizeof(Linger)); |
| @@ -724,6 +734,8 @@ std::pair<s32, Errno> BSD::RecvImpl(s32 fd, u32 flags, std::vector<u8>& message) | |||
| 724 | FileDescriptor& descriptor = *file_descriptors[fd]; | 734 | FileDescriptor& descriptor = *file_descriptors[fd]; |
| 725 | 735 | ||
| 726 | // Apply flags | 736 | // Apply flags |
| 737 | using Network::FLAG_MSG_DONTWAIT; | ||
| 738 | using Network::FLAG_O_NONBLOCK; | ||
| 727 | if ((flags & FLAG_MSG_DONTWAIT) != 0) { | 739 | if ((flags & FLAG_MSG_DONTWAIT) != 0) { |
| 728 | flags &= ~FLAG_MSG_DONTWAIT; | 740 | flags &= ~FLAG_MSG_DONTWAIT; |
| 729 | if ((descriptor.flags & FLAG_O_NONBLOCK) == 0) { | 741 | if ((descriptor.flags & FLAG_O_NONBLOCK) == 0) { |
| @@ -759,6 +771,8 @@ std::pair<s32, Errno> BSD::RecvFromImpl(s32 fd, u32 flags, std::vector<u8>& mess | |||
| 759 | } | 771 | } |
| 760 | 772 | ||
| 761 | // Apply flags | 773 | // Apply flags |
| 774 | using Network::FLAG_MSG_DONTWAIT; | ||
| 775 | using Network::FLAG_O_NONBLOCK; | ||
| 762 | if ((flags & FLAG_MSG_DONTWAIT) != 0) { | 776 | if ((flags & FLAG_MSG_DONTWAIT) != 0) { |
| 763 | flags &= ~FLAG_MSG_DONTWAIT; | 777 | flags &= ~FLAG_MSG_DONTWAIT; |
| 764 | if ((descriptor.flags & FLAG_O_NONBLOCK) == 0) { | 778 | if ((descriptor.flags & FLAG_O_NONBLOCK) == 0) { |
| @@ -857,8 +871,19 @@ void BSD::BuildErrnoResponse(Kernel::HLERequestContext& ctx, Errno bsd_errno) co | |||
| 857 | rb.PushEnum(bsd_errno); | 871 | rb.PushEnum(bsd_errno); |
| 858 | } | 872 | } |
| 859 | 873 | ||
| 874 | void BSD::OnProxyPacketReceived(const Network::ProxyPacket& packet) { | ||
| 875 | for (auto& optional_descriptor : file_descriptors) { | ||
| 876 | if (!optional_descriptor.has_value()) { | ||
| 877 | continue; | ||
| 878 | } | ||
| 879 | FileDescriptor& descriptor = *optional_descriptor; | ||
| 880 | descriptor.socket.get()->HandleProxyPacket(packet); | ||
| 881 | } | ||
| 882 | } | ||
| 883 | |||
| 860 | BSD::BSD(Core::System& system_, const char* name) | 884 | BSD::BSD(Core::System& system_, const char* name) |
| 861 | : ServiceFramework{system_, name, ServiceThreadType::CreateNew} { | 885 | : ServiceFramework{system_, name, ServiceThreadType::CreateNew}, room_network{ |
| 886 | system_.GetRoomNetwork()} { | ||
| 862 | // clang-format off | 887 | // clang-format off |
| 863 | static const FunctionInfo functions[] = { | 888 | static const FunctionInfo functions[] = { |
| 864 | {0, &BSD::RegisterClient, "RegisterClient"}, | 889 | {0, &BSD::RegisterClient, "RegisterClient"}, |
| @@ -899,6 +924,13 @@ BSD::BSD(Core::System& system_, const char* name) | |||
| 899 | // clang-format on | 924 | // clang-format on |
| 900 | 925 | ||
| 901 | RegisterHandlers(functions); | 926 | RegisterHandlers(functions); |
| 927 | |||
| 928 | if (auto room_member = room_network.GetRoomMember().lock()) { | ||
| 929 | proxy_packet_received = room_member->BindOnProxyPacketReceived( | ||
| 930 | [this](const Network::ProxyPacket& packet) { OnProxyPacketReceived(packet); }); | ||
| 931 | } else { | ||
| 932 | LOG_ERROR(Service, "Network isn't initalized"); | ||
| 933 | } | ||
| 902 | } | 934 | } |
| 903 | 935 | ||
| 904 | BSD::~BSD() = default; | 936 | BSD::~BSD() = default; |
diff --git a/src/core/hle/service/sockets/bsd.h b/src/core/hle/service/sockets/bsd.h index 9ea36428d..81e855e0f 100644 --- a/src/core/hle/service/sockets/bsd.h +++ b/src/core/hle/service/sockets/bsd.h | |||
| @@ -7,14 +7,17 @@ | |||
| 7 | #include <vector> | 7 | #include <vector> |
| 8 | 8 | ||
| 9 | #include "common/common_types.h" | 9 | #include "common/common_types.h" |
| 10 | #include "common/socket_types.h" | ||
| 10 | #include "core/hle/service/service.h" | 11 | #include "core/hle/service/service.h" |
| 11 | #include "core/hle/service/sockets/sockets.h" | 12 | #include "core/hle/service/sockets/sockets.h" |
| 13 | #include "network/network.h" | ||
| 12 | 14 | ||
| 13 | namespace Core { | 15 | namespace Core { |
| 14 | class System; | 16 | class System; |
| 15 | } | 17 | } |
| 16 | 18 | ||
| 17 | namespace Network { | 19 | namespace Network { |
| 20 | class SocketBase; | ||
| 18 | class Socket; | 21 | class Socket; |
| 19 | } // namespace Network | 22 | } // namespace Network |
| 20 | 23 | ||
| @@ -30,7 +33,7 @@ private: | |||
| 30 | static constexpr size_t MAX_FD = 128; | 33 | static constexpr size_t MAX_FD = 128; |
| 31 | 34 | ||
| 32 | struct FileDescriptor { | 35 | struct FileDescriptor { |
| 33 | std::unique_ptr<Network::Socket> socket; | 36 | std::unique_ptr<Network::SocketBase> socket; |
| 34 | s32 flags = 0; | 37 | s32 flags = 0; |
| 35 | bool is_connection_based = false; | 38 | bool is_connection_based = false; |
| 36 | }; | 39 | }; |
| @@ -165,6 +168,14 @@ private: | |||
| 165 | void BuildErrnoResponse(Kernel::HLERequestContext& ctx, Errno bsd_errno) const noexcept; | 168 | void BuildErrnoResponse(Kernel::HLERequestContext& ctx, Errno bsd_errno) const noexcept; |
| 166 | 169 | ||
| 167 | std::array<std::optional<FileDescriptor>, MAX_FD> file_descriptors; | 170 | std::array<std::optional<FileDescriptor>, MAX_FD> file_descriptors; |
| 171 | |||
| 172 | Network::RoomNetwork& room_network; | ||
| 173 | |||
| 174 | /// Callback to parse and handle a received wifi packet. | ||
| 175 | void OnProxyPacketReceived(const Network::ProxyPacket& packet); | ||
| 176 | |||
| 177 | // Callback identifier for the OnProxyPacketReceived event. | ||
| 178 | Network::RoomMember::CallbackHandle<Network::ProxyPacket> proxy_packet_received; | ||
| 168 | }; | 179 | }; |
| 169 | 180 | ||
| 170 | class BSDCFG final : public ServiceFramework<BSDCFG> { | 181 | class BSDCFG final : public ServiceFramework<BSDCFG> { |
diff --git a/src/core/hle/service/sockets/sockets.h b/src/core/hle/service/sockets/sockets.h index b735b00fc..31b7dad33 100644 --- a/src/core/hle/service/sockets/sockets.h +++ b/src/core/hle/service/sockets/sockets.h | |||
| @@ -22,7 +22,9 @@ enum class Errno : u32 { | |||
| 22 | AGAIN = 11, | 22 | AGAIN = 11, |
| 23 | INVAL = 22, | 23 | INVAL = 22, |
| 24 | MFILE = 24, | 24 | MFILE = 24, |
| 25 | MSGSIZE = 90, | ||
| 25 | NOTCONN = 107, | 26 | NOTCONN = 107, |
| 27 | TIMEDOUT = 110, | ||
| 26 | }; | 28 | }; |
| 27 | 29 | ||
| 28 | enum class Domain : u32 { | 30 | enum class Domain : u32 { |
| @@ -96,10 +98,6 @@ struct Linger { | |||
| 96 | u32 linger; | 98 | u32 linger; |
| 97 | }; | 99 | }; |
| 98 | 100 | ||
| 99 | constexpr u32 FLAG_MSG_DONTWAIT = 0x80; | ||
| 100 | |||
| 101 | constexpr u32 FLAG_O_NONBLOCK = 0x800; | ||
| 102 | |||
| 103 | /// Registers all Sockets services with the specified service manager. | 101 | /// Registers all Sockets services with the specified service manager. |
| 104 | void InstallInterfaces(SM::ServiceManager& service_manager, Core::System& system); | 102 | void InstallInterfaces(SM::ServiceManager& service_manager, Core::System& system); |
| 105 | 103 | ||
diff --git a/src/core/hle/service/sockets/sockets_translate.cpp b/src/core/hle/service/sockets/sockets_translate.cpp index 2db10ec81..023aa0486 100644 --- a/src/core/hle/service/sockets/sockets_translate.cpp +++ b/src/core/hle/service/sockets/sockets_translate.cpp | |||
| @@ -25,6 +25,8 @@ Errno Translate(Network::Errno value) { | |||
| 25 | return Errno::MFILE; | 25 | return Errno::MFILE; |
| 26 | case Network::Errno::NOTCONN: | 26 | case Network::Errno::NOTCONN: |
| 27 | return Errno::NOTCONN; | 27 | return Errno::NOTCONN; |
| 28 | case Network::Errno::TIMEDOUT: | ||
| 29 | return Errno::TIMEDOUT; | ||
| 28 | default: | 30 | default: |
| 29 | UNIMPLEMENTED_MSG("Unimplemented errno={}", value); | 31 | UNIMPLEMENTED_MSG("Unimplemented errno={}", value); |
| 30 | return Errno::SUCCESS; | 32 | return Errno::SUCCESS; |
diff --git a/src/core/internal_network/network.cpp b/src/core/internal_network/network.cpp index 36c43cc8f..160cc83e4 100644 --- a/src/core/internal_network/network.cpp +++ b/src/core/internal_network/network.cpp | |||
| @@ -32,6 +32,7 @@ | |||
| 32 | #include "core/internal_network/network.h" | 32 | #include "core/internal_network/network.h" |
| 33 | #include "core/internal_network/network_interface.h" | 33 | #include "core/internal_network/network_interface.h" |
| 34 | #include "core/internal_network/sockets.h" | 34 | #include "core/internal_network/sockets.h" |
| 35 | #include "network/network.h" | ||
| 35 | 36 | ||
| 36 | namespace Network { | 37 | namespace Network { |
| 37 | 38 | ||
| @@ -114,7 +115,10 @@ Errno TranslateNativeError(int e) { | |||
| 114 | return Errno::NETDOWN; | 115 | return Errno::NETDOWN; |
| 115 | case WSAENETUNREACH: | 116 | case WSAENETUNREACH: |
| 116 | return Errno::NETUNREACH; | 117 | return Errno::NETUNREACH; |
| 118 | case WSAEMSGSIZE: | ||
| 119 | return Errno::MSGSIZE; | ||
| 117 | default: | 120 | default: |
| 121 | UNIMPLEMENTED_MSG("Unimplemented errno={}", e); | ||
| 118 | return Errno::OTHER; | 122 | return Errno::OTHER; |
| 119 | } | 123 | } |
| 120 | } | 124 | } |
| @@ -125,7 +129,6 @@ using SOCKET = int; | |||
| 125 | using WSAPOLLFD = pollfd; | 129 | using WSAPOLLFD = pollfd; |
| 126 | using ULONG = u64; | 130 | using ULONG = u64; |
| 127 | 131 | ||
| 128 | constexpr SOCKET INVALID_SOCKET = -1; | ||
| 129 | constexpr SOCKET SOCKET_ERROR = -1; | 132 | constexpr SOCKET SOCKET_ERROR = -1; |
| 130 | 133 | ||
| 131 | constexpr int SD_RECEIVE = SHUT_RD; | 134 | constexpr int SD_RECEIVE = SHUT_RD; |
| @@ -206,7 +209,10 @@ Errno TranslateNativeError(int e) { | |||
| 206 | return Errno::NETDOWN; | 209 | return Errno::NETDOWN; |
| 207 | case ENETUNREACH: | 210 | case ENETUNREACH: |
| 208 | return Errno::NETUNREACH; | 211 | return Errno::NETUNREACH; |
| 212 | case EMSGSIZE: | ||
| 213 | return Errno::MSGSIZE; | ||
| 209 | default: | 214 | default: |
| 215 | UNIMPLEMENTED_MSG("Unimplemented errno={}", e); | ||
| 210 | return Errno::OTHER; | 216 | return Errno::OTHER; |
| 211 | } | 217 | } |
| 212 | } | 218 | } |
| @@ -329,16 +335,6 @@ PollEvents TranslatePollRevents(short revents) { | |||
| 329 | return result; | 335 | return result; |
| 330 | } | 336 | } |
| 331 | 337 | ||
| 332 | template <typename T> | ||
| 333 | Errno SetSockOpt(SOCKET fd, int option, T value) { | ||
| 334 | const int result = | ||
| 335 | setsockopt(fd, SOL_SOCKET, option, reinterpret_cast<const char*>(&value), sizeof(value)); | ||
| 336 | if (result != SOCKET_ERROR) { | ||
| 337 | return Errno::SUCCESS; | ||
| 338 | } | ||
| 339 | return GetAndLogLastError(); | ||
| 340 | } | ||
| 341 | |||
| 342 | } // Anonymous namespace | 338 | } // Anonymous namespace |
| 343 | 339 | ||
| 344 | NetworkInstance::NetworkInstance() { | 340 | NetworkInstance::NetworkInstance() { |
| @@ -350,26 +346,15 @@ NetworkInstance::~NetworkInstance() { | |||
| 350 | } | 346 | } |
| 351 | 347 | ||
| 352 | std::optional<IPv4Address> GetHostIPv4Address() { | 348 | std::optional<IPv4Address> GetHostIPv4Address() { |
| 353 | const std::string& selected_network_interface = Settings::values.network_interface.GetValue(); | 349 | const auto interface = Network::GetSelectedNetworkInterface(); |
| 354 | const auto network_interfaces = Network::GetAvailableNetworkInterfaces(); | 350 | if (!interface.has_value()) { |
| 355 | if (network_interfaces.size() == 0) { | 351 | LOG_ERROR(Network, "GetSelectedNetworkInterface returned no interface"); |
| 356 | LOG_ERROR(Network, "GetAvailableNetworkInterfaces returned no interfaces"); | ||
| 357 | return {}; | 352 | return {}; |
| 358 | } | 353 | } |
| 359 | 354 | ||
| 360 | const auto res = | 355 | char ip_addr[16] = {}; |
| 361 | std::ranges::find_if(network_interfaces, [&selected_network_interface](const auto& iface) { | 356 | ASSERT(inet_ntop(AF_INET, &interface->ip_address, ip_addr, sizeof(ip_addr)) != nullptr); |
| 362 | return iface.name == selected_network_interface; | 357 | return TranslateIPv4(interface->ip_address); |
| 363 | }); | ||
| 364 | |||
| 365 | if (res != network_interfaces.end()) { | ||
| 366 | char ip_addr[16] = {}; | ||
| 367 | ASSERT(inet_ntop(AF_INET, &res->ip_address, ip_addr, sizeof(ip_addr)) != nullptr); | ||
| 368 | return TranslateIPv4(res->ip_address); | ||
| 369 | } else { | ||
| 370 | LOG_ERROR(Network, "Couldn't find selected interface \"{}\"", selected_network_interface); | ||
| 371 | return {}; | ||
| 372 | } | ||
| 373 | } | 358 | } |
| 374 | 359 | ||
| 375 | std::pair<s32, Errno> Poll(std::vector<PollFD>& pollfds, s32 timeout) { | 360 | std::pair<s32, Errno> Poll(std::vector<PollFD>& pollfds, s32 timeout) { |
| @@ -412,7 +397,19 @@ Socket::~Socket() { | |||
| 412 | fd = INVALID_SOCKET; | 397 | fd = INVALID_SOCKET; |
| 413 | } | 398 | } |
| 414 | 399 | ||
| 415 | Socket::Socket(Socket&& rhs) noexcept : fd{std::exchange(rhs.fd, INVALID_SOCKET)} {} | 400 | Socket::Socket(Socket&& rhs) noexcept { |
| 401 | fd = std::exchange(rhs.fd, INVALID_SOCKET); | ||
| 402 | } | ||
| 403 | |||
| 404 | template <typename T> | ||
| 405 | Errno Socket::SetSockOpt(SOCKET _fd, int option, T value) { | ||
| 406 | const int result = | ||
| 407 | setsockopt(_fd, SOL_SOCKET, option, reinterpret_cast<const char*>(&value), sizeof(value)); | ||
| 408 | if (result != SOCKET_ERROR) { | ||
| 409 | return Errno::SUCCESS; | ||
| 410 | } | ||
| 411 | return GetAndLogLastError(); | ||
| 412 | } | ||
| 416 | 413 | ||
| 417 | Errno Socket::Initialize(Domain domain, Type type, Protocol protocol) { | 414 | Errno Socket::Initialize(Domain domain, Type type, Protocol protocol) { |
| 418 | fd = socket(TranslateDomain(domain), TranslateType(type), TranslateProtocol(protocol)); | 415 | fd = socket(TranslateDomain(domain), TranslateType(type), TranslateProtocol(protocol)); |
| @@ -423,7 +420,7 @@ Errno Socket::Initialize(Domain domain, Type type, Protocol protocol) { | |||
| 423 | return GetAndLogLastError(); | 420 | return GetAndLogLastError(); |
| 424 | } | 421 | } |
| 425 | 422 | ||
| 426 | std::pair<Socket::AcceptResult, Errno> Socket::Accept() { | 423 | std::pair<SocketBase::AcceptResult, Errno> Socket::Accept() { |
| 427 | sockaddr addr; | 424 | sockaddr addr; |
| 428 | socklen_t addrlen = sizeof(addr); | 425 | socklen_t addrlen = sizeof(addr); |
| 429 | const SOCKET new_socket = accept(fd, &addr, &addrlen); | 426 | const SOCKET new_socket = accept(fd, &addr, &addrlen); |
| @@ -634,4 +631,8 @@ bool Socket::IsOpened() const { | |||
| 634 | return fd != INVALID_SOCKET; | 631 | return fd != INVALID_SOCKET; |
| 635 | } | 632 | } |
| 636 | 633 | ||
| 634 | void Socket::HandleProxyPacket(const ProxyPacket& packet) { | ||
| 635 | LOG_WARNING(Network, "ProxyPacket received, but not in Proxy mode!"); | ||
| 636 | } | ||
| 637 | |||
| 637 | } // namespace Network | 638 | } // namespace Network |
diff --git a/src/core/internal_network/network.h b/src/core/internal_network/network.h index 10e5ef10d..36994c22e 100644 --- a/src/core/internal_network/network.h +++ b/src/core/internal_network/network.h | |||
| @@ -8,6 +8,7 @@ | |||
| 8 | 8 | ||
| 9 | #include "common/common_funcs.h" | 9 | #include "common/common_funcs.h" |
| 10 | #include "common/common_types.h" | 10 | #include "common/common_types.h" |
| 11 | #include "common/socket_types.h" | ||
| 11 | 12 | ||
| 12 | #ifdef _WIN32 | 13 | #ifdef _WIN32 |
| 13 | #include <winsock2.h> | 14 | #include <winsock2.h> |
| @@ -17,6 +18,7 @@ | |||
| 17 | 18 | ||
| 18 | namespace Network { | 19 | namespace Network { |
| 19 | 20 | ||
| 21 | class SocketBase; | ||
| 20 | class Socket; | 22 | class Socket; |
| 21 | 23 | ||
| 22 | /// Error code for network functions | 24 | /// Error code for network functions |
| @@ -31,46 +33,11 @@ enum class Errno { | |||
| 31 | HOSTUNREACH, | 33 | HOSTUNREACH, |
| 32 | NETDOWN, | 34 | NETDOWN, |
| 33 | NETUNREACH, | 35 | NETUNREACH, |
| 36 | TIMEDOUT, | ||
| 37 | MSGSIZE, | ||
| 34 | OTHER, | 38 | OTHER, |
| 35 | }; | 39 | }; |
| 36 | 40 | ||
| 37 | /// Address families | ||
| 38 | enum class Domain { | ||
| 39 | INET, ///< Address family for IPv4 | ||
| 40 | }; | ||
| 41 | |||
| 42 | /// Socket types | ||
| 43 | enum class Type { | ||
| 44 | STREAM, | ||
| 45 | DGRAM, | ||
| 46 | RAW, | ||
| 47 | SEQPACKET, | ||
| 48 | }; | ||
| 49 | |||
| 50 | /// Protocol values for sockets | ||
| 51 | enum class Protocol { | ||
| 52 | ICMP, | ||
| 53 | TCP, | ||
| 54 | UDP, | ||
| 55 | }; | ||
| 56 | |||
| 57 | /// Shutdown mode | ||
| 58 | enum class ShutdownHow { | ||
| 59 | RD, | ||
| 60 | WR, | ||
| 61 | RDWR, | ||
| 62 | }; | ||
| 63 | |||
| 64 | /// Array of IPv4 address | ||
| 65 | using IPv4Address = std::array<u8, 4>; | ||
| 66 | |||
| 67 | /// Cross-platform sockaddr structure | ||
| 68 | struct SockAddrIn { | ||
| 69 | Domain family; | ||
| 70 | IPv4Address ip; | ||
| 71 | u16 portno; | ||
| 72 | }; | ||
| 73 | |||
| 74 | /// Cross-platform poll fd structure | 41 | /// Cross-platform poll fd structure |
| 75 | 42 | ||
| 76 | enum class PollEvents : u16 { | 43 | enum class PollEvents : u16 { |
| @@ -86,7 +53,7 @@ enum class PollEvents : u16 { | |||
| 86 | DECLARE_ENUM_FLAG_OPERATORS(PollEvents); | 53 | DECLARE_ENUM_FLAG_OPERATORS(PollEvents); |
| 87 | 54 | ||
| 88 | struct PollFD { | 55 | struct PollFD { |
| 89 | Socket* socket; | 56 | SocketBase* socket; |
| 90 | PollEvents events; | 57 | PollEvents events; |
| 91 | PollEvents revents; | 58 | PollEvents revents; |
| 92 | }; | 59 | }; |
diff --git a/src/core/internal_network/socket_proxy.cpp b/src/core/internal_network/socket_proxy.cpp new file mode 100644 index 000000000..6e8924822 --- /dev/null +++ b/src/core/internal_network/socket_proxy.cpp | |||
| @@ -0,0 +1,282 @@ | |||
| 1 | // Copyright 2022 yuzu emulator team | ||
| 2 | // Licensed under GPLv2 or any later version | ||
| 3 | // Refer to the license.txt file included. | ||
| 4 | |||
| 5 | #include <chrono> | ||
| 6 | #include <thread> | ||
| 7 | |||
| 8 | #include "common/assert.h" | ||
| 9 | #include "common/logging/log.h" | ||
| 10 | #include "core/internal_network/network.h" | ||
| 11 | #include "core/internal_network/network_interface.h" | ||
| 12 | #include "core/internal_network/socket_proxy.h" | ||
| 13 | |||
| 14 | namespace Network { | ||
| 15 | |||
| 16 | ProxySocket::ProxySocket(RoomNetwork& room_network_) noexcept : room_network{room_network_} {} | ||
| 17 | |||
| 18 | ProxySocket::ProxySocket(ProxySocket&& rhs) noexcept : room_network{rhs.room_network} { | ||
| 19 | fd = std::exchange(rhs.fd, INVALID_SOCKET); | ||
| 20 | } | ||
| 21 | |||
| 22 | ProxySocket::~ProxySocket() { | ||
| 23 | if (fd == INVALID_SOCKET) { | ||
| 24 | return; | ||
| 25 | } | ||
| 26 | fd = INVALID_SOCKET; | ||
| 27 | } | ||
| 28 | |||
| 29 | void ProxySocket::HandleProxyPacket(const ProxyPacket& packet) { | ||
| 30 | if (protocol != packet.protocol || local_endpoint.portno != packet.remote_endpoint.portno || | ||
| 31 | closed) { | ||
| 32 | return; | ||
| 33 | } | ||
| 34 | std::lock_guard<std::mutex> guard(packets_mutex); | ||
| 35 | received_packets.push(packet); | ||
| 36 | } | ||
| 37 | |||
| 38 | template <typename T> | ||
| 39 | Errno ProxySocket::SetSockOpt(SOCKET _fd, int option, T value) { | ||
| 40 | socket_options[option] = reinterpret_cast<const char*>(&value); | ||
| 41 | return Errno::SUCCESS; | ||
| 42 | } | ||
| 43 | |||
| 44 | Errno ProxySocket::Initialize(Domain domain, Type type, Protocol socket_protocol) { | ||
| 45 | protocol = socket_protocol; | ||
| 46 | socket_options[0x1008] = reinterpret_cast<const char*>(&type); | ||
| 47 | |||
| 48 | return Errno::SUCCESS; | ||
| 49 | } | ||
| 50 | |||
| 51 | std::pair<ProxySocket::AcceptResult, Errno> ProxySocket::Accept() { | ||
| 52 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 53 | return {AcceptResult{}, Errno::SUCCESS}; | ||
| 54 | } | ||
| 55 | |||
| 56 | Errno ProxySocket::Connect(SockAddrIn addr_in) { | ||
| 57 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 58 | return Errno::SUCCESS; | ||
| 59 | } | ||
| 60 | |||
| 61 | std::pair<SockAddrIn, Errno> ProxySocket::GetPeerName() { | ||
| 62 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 63 | return {SockAddrIn{}, Errno::SUCCESS}; | ||
| 64 | } | ||
| 65 | |||
| 66 | std::pair<SockAddrIn, Errno> ProxySocket::GetSockName() { | ||
| 67 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 68 | return {SockAddrIn{}, Errno::SUCCESS}; | ||
| 69 | } | ||
| 70 | |||
| 71 | Errno ProxySocket::Bind(SockAddrIn addr) { | ||
| 72 | if (is_bound) { | ||
| 73 | LOG_WARNING(Network, "Rebinding Socket is unimplemented!"); | ||
| 74 | return Errno::SUCCESS; | ||
| 75 | } | ||
| 76 | local_endpoint = addr; | ||
| 77 | is_bound = true; | ||
| 78 | |||
| 79 | return Errno::SUCCESS; | ||
| 80 | } | ||
| 81 | |||
| 82 | Errno ProxySocket::Listen(s32 backlog) { | ||
| 83 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 84 | return Errno::SUCCESS; | ||
| 85 | } | ||
| 86 | |||
| 87 | Errno ProxySocket::Shutdown(ShutdownHow how) { | ||
| 88 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 89 | return Errno::SUCCESS; | ||
| 90 | } | ||
| 91 | |||
| 92 | std::pair<s32, Errno> ProxySocket::Recv(int flags, std::vector<u8>& message) { | ||
| 93 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 94 | ASSERT(flags == 0); | ||
| 95 | ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max())); | ||
| 96 | |||
| 97 | return {static_cast<s32>(0), Errno::SUCCESS}; | ||
| 98 | } | ||
| 99 | |||
| 100 | std::pair<s32, Errno> ProxySocket::RecvFrom(int flags, std::vector<u8>& message, SockAddrIn* addr) { | ||
| 101 | ASSERT(flags == 0); | ||
| 102 | ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max())); | ||
| 103 | |||
| 104 | { | ||
| 105 | std::lock_guard<std::mutex> guard(packets_mutex); | ||
| 106 | if (received_packets.size() > 0) { | ||
| 107 | return ReceivePacket(flags, message, addr, message.size()); | ||
| 108 | } | ||
| 109 | } | ||
| 110 | |||
| 111 | if (blocking) { | ||
| 112 | if (receive_timeout > 0) { | ||
| 113 | std::this_thread::sleep_for(std::chrono::milliseconds(receive_timeout)); | ||
| 114 | } | ||
| 115 | } else { | ||
| 116 | return {-1, Errno::AGAIN}; | ||
| 117 | } | ||
| 118 | |||
| 119 | std::lock_guard<std::mutex> guard(packets_mutex); | ||
| 120 | if (received_packets.size() > 0) { | ||
| 121 | return ReceivePacket(flags, message, addr, message.size()); | ||
| 122 | } | ||
| 123 | |||
| 124 | return {-1, Errno::TIMEDOUT}; | ||
| 125 | } | ||
| 126 | |||
| 127 | std::pair<s32, Errno> ProxySocket::ReceivePacket(int flags, std::vector<u8>& message, | ||
| 128 | SockAddrIn* addr, std::size_t max_length) { | ||
| 129 | ProxyPacket& packet = received_packets.front(); | ||
| 130 | if (addr) { | ||
| 131 | addr->family = Domain::INET; | ||
| 132 | addr->ip = packet.local_endpoint.ip; // The senders ip address | ||
| 133 | addr->portno = packet.local_endpoint.portno; // The senders port number | ||
| 134 | } | ||
| 135 | |||
| 136 | bool peek = (flags & FLAG_MSG_PEEK) != 0; | ||
| 137 | std::size_t read_bytes; | ||
| 138 | if (packet.data.size() > max_length) { | ||
| 139 | read_bytes = max_length; | ||
| 140 | message.clear(); | ||
| 141 | std::copy(packet.data.begin(), packet.data.begin() + read_bytes, | ||
| 142 | std::back_inserter(message)); | ||
| 143 | message.resize(max_length); | ||
| 144 | |||
| 145 | if (protocol == Protocol::UDP) { | ||
| 146 | if (!peek) { | ||
| 147 | received_packets.pop(); | ||
| 148 | } | ||
| 149 | return {-1, Errno::MSGSIZE}; | ||
| 150 | } else if (protocol == Protocol::TCP) { | ||
| 151 | std::vector<u8> numArray(packet.data.size() - max_length); | ||
| 152 | std::copy(packet.data.begin() + max_length, packet.data.end(), | ||
| 153 | std::back_inserter(numArray)); | ||
| 154 | packet.data = numArray; | ||
| 155 | } | ||
| 156 | } else { | ||
| 157 | read_bytes = packet.data.size(); | ||
| 158 | message.clear(); | ||
| 159 | std::copy(packet.data.begin(), packet.data.end(), std::back_inserter(message)); | ||
| 160 | message.resize(max_length); | ||
| 161 | if (!peek) { | ||
| 162 | received_packets.pop(); | ||
| 163 | } | ||
| 164 | } | ||
| 165 | |||
| 166 | return {static_cast<u32>(read_bytes), Errno::SUCCESS}; | ||
| 167 | } | ||
| 168 | |||
| 169 | std::pair<s32, Errno> ProxySocket::Send(const std::vector<u8>& message, int flags) { | ||
| 170 | LOG_WARNING(Network, "(STUBBED) called"); | ||
| 171 | ASSERT(message.size() < static_cast<size_t>(std::numeric_limits<int>::max())); | ||
| 172 | ASSERT(flags == 0); | ||
| 173 | |||
| 174 | return {static_cast<s32>(0), Errno::SUCCESS}; | ||
| 175 | } | ||
| 176 | |||
| 177 | void ProxySocket::SendPacket(ProxyPacket& packet) { | ||
| 178 | if (auto room_member = room_network.GetRoomMember().lock()) { | ||
| 179 | if (room_member->IsConnected()) { | ||
| 180 | room_member->SendProxyPacket(packet); | ||
| 181 | } | ||
| 182 | } | ||
| 183 | } | ||
| 184 | |||
| 185 | std::pair<s32, Errno> ProxySocket::SendTo(u32 flags, const std::vector<u8>& message, | ||
| 186 | const SockAddrIn* addr) { | ||
| 187 | ASSERT(flags == 0); | ||
| 188 | |||
| 189 | if (!is_bound) { | ||
| 190 | LOG_ERROR(Network, "ProxySocket is not bound!"); | ||
| 191 | return {static_cast<s32>(message.size()), Errno::SUCCESS}; | ||
| 192 | } | ||
| 193 | |||
| 194 | if (auto room_member = room_network.GetRoomMember().lock()) { | ||
| 195 | if (!room_member->IsConnected()) { | ||
| 196 | return {static_cast<s32>(message.size()), Errno::SUCCESS}; | ||
| 197 | } | ||
| 198 | } | ||
| 199 | |||
| 200 | ProxyPacket packet; | ||
| 201 | packet.local_endpoint = local_endpoint; | ||
| 202 | packet.remote_endpoint = *addr; | ||
| 203 | packet.protocol = protocol; | ||
| 204 | packet.broadcast = broadcast; | ||
| 205 | |||
| 206 | auto& ip = local_endpoint.ip; | ||
| 207 | auto ipv4 = Network::GetHostIPv4Address(); | ||
| 208 | // If the ip is all zeroes (INADDR_ANY) or if it matches the hosts ip address, | ||
| 209 | // replace it with a "fake" routing address | ||
| 210 | if (std::all_of(ip.begin(), ip.end(), [](u8 i) { return i == 0; }) || (ipv4 && ipv4 == ip)) { | ||
| 211 | if (auto room_member = room_network.GetRoomMember().lock()) { | ||
| 212 | packet.local_endpoint.ip = room_member->GetFakeIpAddress(); | ||
| 213 | } | ||
| 214 | } | ||
| 215 | |||
| 216 | packet.data.clear(); | ||
| 217 | std::copy(message.begin(), message.end(), std::back_inserter(packet.data)); | ||
| 218 | |||
| 219 | SendPacket(packet); | ||
| 220 | |||
| 221 | return {static_cast<s32>(message.size()), Errno::SUCCESS}; | ||
| 222 | } | ||
| 223 | |||
| 224 | Errno ProxySocket::Close() { | ||
| 225 | fd = INVALID_SOCKET; | ||
| 226 | closed = true; | ||
| 227 | |||
| 228 | return Errno::SUCCESS; | ||
| 229 | } | ||
| 230 | |||
| 231 | Errno ProxySocket::SetLinger(bool enable, u32 linger) { | ||
| 232 | struct Linger { | ||
| 233 | u16 linger_enable; | ||
| 234 | u16 linger_time; | ||
| 235 | } values; | ||
| 236 | values.linger_enable = enable ? 1 : 0; | ||
| 237 | values.linger_time = static_cast<u16>(linger); | ||
| 238 | |||
| 239 | return SetSockOpt(fd, SO_LINGER, values); | ||
| 240 | } | ||
| 241 | |||
| 242 | Errno ProxySocket::SetReuseAddr(bool enable) { | ||
| 243 | return SetSockOpt<u32>(fd, SO_REUSEADDR, enable ? 1 : 0); | ||
| 244 | } | ||
| 245 | |||
| 246 | Errno ProxySocket::SetBroadcast(bool enable) { | ||
| 247 | broadcast = enable; | ||
| 248 | return SetSockOpt<u32>(fd, SO_BROADCAST, enable ? 1 : 0); | ||
| 249 | } | ||
| 250 | |||
| 251 | Errno ProxySocket::SetSndBuf(u32 value) { | ||
| 252 | return SetSockOpt(fd, SO_SNDBUF, value); | ||
| 253 | } | ||
| 254 | |||
| 255 | Errno ProxySocket::SetKeepAlive(bool enable) { | ||
| 256 | return Errno::SUCCESS; | ||
| 257 | } | ||
| 258 | |||
| 259 | Errno ProxySocket::SetRcvBuf(u32 value) { | ||
| 260 | return SetSockOpt(fd, SO_RCVBUF, value); | ||
| 261 | } | ||
| 262 | |||
| 263 | Errno ProxySocket::SetSndTimeo(u32 value) { | ||
| 264 | send_timeout = value; | ||
| 265 | return SetSockOpt(fd, SO_SNDTIMEO, static_cast<int>(value)); | ||
| 266 | } | ||
| 267 | |||
| 268 | Errno ProxySocket::SetRcvTimeo(u32 value) { | ||
| 269 | receive_timeout = value; | ||
| 270 | return SetSockOpt(fd, SO_RCVTIMEO, static_cast<int>(value)); | ||
| 271 | } | ||
| 272 | |||
| 273 | Errno ProxySocket::SetNonBlock(bool enable) { | ||
| 274 | blocking = !enable; | ||
| 275 | return Errno::SUCCESS; | ||
| 276 | } | ||
| 277 | |||
| 278 | bool ProxySocket::IsOpened() const { | ||
| 279 | return fd != INVALID_SOCKET; | ||
| 280 | } | ||
| 281 | |||
| 282 | } // namespace Network | ||
diff --git a/src/core/internal_network/socket_proxy.h b/src/core/internal_network/socket_proxy.h new file mode 100644 index 000000000..ef7d5b554 --- /dev/null +++ b/src/core/internal_network/socket_proxy.h | |||
| @@ -0,0 +1,102 @@ | |||
| 1 | // Copyright 2022 yuzu emulator team | ||
| 2 | // Licensed under GPLv2 or any later version | ||
| 3 | // Refer to the license.txt file included. | ||
| 4 | |||
| 5 | #pragma once | ||
| 6 | |||
| 7 | #include <mutex> | ||
| 8 | #include <vector> | ||
| 9 | #include <queue> | ||
| 10 | |||
| 11 | #include "core/internal_network/sockets.h" | ||
| 12 | #include "network/network.h" | ||
| 13 | |||
| 14 | namespace Network { | ||
| 15 | |||
| 16 | class ProxySocket : public SocketBase { | ||
| 17 | public: | ||
| 18 | ProxySocket(RoomNetwork& room_network_) noexcept; | ||
| 19 | ~ProxySocket() override; | ||
| 20 | |||
| 21 | ProxySocket(const ProxySocket&) = delete; | ||
| 22 | ProxySocket& operator=(const ProxySocket&) = delete; | ||
| 23 | |||
| 24 | ProxySocket(ProxySocket&& rhs) noexcept; | ||
| 25 | |||
| 26 | // Avoid closing sockets implicitly | ||
| 27 | ProxySocket& operator=(ProxySocket&&) noexcept = delete; | ||
| 28 | |||
| 29 | void HandleProxyPacket(const ProxyPacket& packet); | ||
| 30 | |||
| 31 | Errno Initialize(Domain domain, Type type, Protocol socket_protocol) override; | ||
| 32 | |||
| 33 | Errno Close() override; | ||
| 34 | |||
| 35 | std::pair<AcceptResult, Errno> Accept() override; | ||
| 36 | |||
| 37 | Errno Connect(SockAddrIn addr_in) override; | ||
| 38 | |||
| 39 | std::pair<SockAddrIn, Errno> GetPeerName() override; | ||
| 40 | |||
| 41 | std::pair<SockAddrIn, Errno> GetSockName() override; | ||
| 42 | |||
| 43 | Errno Bind(SockAddrIn addr) override; | ||
| 44 | |||
| 45 | Errno Listen(s32 backlog) override; | ||
| 46 | |||
| 47 | Errno Shutdown(ShutdownHow how) override; | ||
| 48 | |||
| 49 | std::pair<s32, Errno> Recv(int flags, std::vector<u8>& message) override; | ||
| 50 | |||
| 51 | std::pair<s32, Errno> RecvFrom(int flags, std::vector<u8>& message, SockAddrIn* addr) override; | ||
| 52 | |||
| 53 | std::pair<s32, Errno> ReceivePacket(int flags, std::vector<u8>& message, SockAddrIn* addr, | ||
| 54 | std::size_t max_length); | ||
| 55 | |||
| 56 | std::pair<s32, Errno> Send(const std::vector<u8>& message, int flags) override; | ||
| 57 | |||
| 58 | void SendPacket(ProxyPacket& packet); | ||
| 59 | |||
| 60 | std::pair<s32, Errno> SendTo(u32 flags, const std::vector<u8>& message, | ||
| 61 | const SockAddrIn* addr) override; | ||
| 62 | |||
| 63 | Errno SetLinger(bool enable, u32 linger) override; | ||
| 64 | |||
| 65 | Errno SetReuseAddr(bool enable) override; | ||
| 66 | |||
| 67 | Errno SetBroadcast(bool enable) override; | ||
| 68 | |||
| 69 | Errno SetKeepAlive(bool enable) override; | ||
| 70 | |||
| 71 | Errno SetSndBuf(u32 value) override; | ||
| 72 | |||
| 73 | Errno SetRcvBuf(u32 value) override; | ||
| 74 | |||
| 75 | Errno SetSndTimeo(u32 value) override; | ||
| 76 | |||
| 77 | Errno SetRcvTimeo(u32 value) override; | ||
| 78 | |||
| 79 | Errno SetNonBlock(bool enable) override; | ||
| 80 | |||
| 81 | template <typename T> | ||
| 82 | Errno SetSockOpt(SOCKET fd, int option, T value); | ||
| 83 | |||
| 84 | bool IsOpened() const override; | ||
| 85 | |||
| 86 | bool broadcast = false; | ||
| 87 | bool closed = false; | ||
| 88 | u32 send_timeout = 0; | ||
| 89 | u32 receive_timeout = 0; | ||
| 90 | std::map<int, const char*> socket_options; | ||
| 91 | bool is_bound = false; | ||
| 92 | SockAddrIn local_endpoint{}; | ||
| 93 | bool blocking = true; | ||
| 94 | std::queue<ProxyPacket> received_packets; | ||
| 95 | Protocol protocol; | ||
| 96 | |||
| 97 | std::mutex packets_mutex; | ||
| 98 | |||
| 99 | RoomNetwork& room_network; | ||
| 100 | }; | ||
| 101 | |||
| 102 | } // namespace Network | ||
diff --git a/src/core/internal_network/sockets.h b/src/core/internal_network/sockets.h index 77e27e928..92dc49993 100644 --- a/src/core/internal_network/sockets.h +++ b/src/core/internal_network/sockets.h | |||
| @@ -14,20 +14,92 @@ | |||
| 14 | 14 | ||
| 15 | #include "common/common_types.h" | 15 | #include "common/common_types.h" |
| 16 | #include "core/internal_network/network.h" | 16 | #include "core/internal_network/network.h" |
| 17 | #include "network/network.h" | ||
| 17 | 18 | ||
| 18 | // TODO: C++20 Replace std::vector usages with std::span | 19 | // TODO: C++20 Replace std::vector usages with std::span |
| 19 | 20 | ||
| 20 | namespace Network { | 21 | namespace Network { |
| 21 | 22 | ||
| 22 | class Socket { | 23 | class SocketBase { |
| 23 | public: | 24 | public: |
| 25 | #ifdef YUZU_UNIX | ||
| 26 | using SOCKET = int; | ||
| 27 | static constexpr SOCKET INVALID_SOCKET = -1; | ||
| 28 | static constexpr SOCKET SOCKET_ERROR = -1; | ||
| 29 | #endif | ||
| 30 | |||
| 24 | struct AcceptResult { | 31 | struct AcceptResult { |
| 25 | std::unique_ptr<Socket> socket; | 32 | std::unique_ptr<SocketBase> socket; |
| 26 | SockAddrIn sockaddr_in; | 33 | SockAddrIn sockaddr_in; |
| 27 | }; | 34 | }; |
| 35 | virtual ~SocketBase() {} | ||
| 36 | |||
| 37 | virtual SocketBase& operator=(const SocketBase&) = delete; | ||
| 38 | |||
| 39 | // Avoid closing sockets implicitly | ||
| 40 | virtual SocketBase& operator=(SocketBase&&) noexcept = delete; | ||
| 41 | |||
| 42 | virtual Errno Initialize(Domain domain, Type type, Protocol protocol) = 0; | ||
| 43 | |||
| 44 | virtual Errno Close() = 0; | ||
| 45 | |||
| 46 | virtual std::pair<AcceptResult, Errno> Accept() = 0; | ||
| 47 | |||
| 48 | virtual Errno Connect(SockAddrIn addr_in) = 0; | ||
| 49 | |||
| 50 | virtual std::pair<SockAddrIn, Errno> GetPeerName() = 0; | ||
| 51 | |||
| 52 | virtual std::pair<SockAddrIn, Errno> GetSockName() = 0; | ||
| 53 | |||
| 54 | virtual Errno Bind(SockAddrIn addr) = 0; | ||
| 55 | |||
| 56 | virtual Errno Listen(s32 backlog) = 0; | ||
| 57 | |||
| 58 | virtual Errno Shutdown(ShutdownHow how) = 0; | ||
| 59 | |||
| 60 | virtual std::pair<s32, Errno> Recv(int flags, std::vector<u8>& message) = 0; | ||
| 61 | |||
| 62 | virtual std::pair<s32, Errno> RecvFrom(int flags, std::vector<u8>& message, | ||
| 63 | SockAddrIn* addr) = 0; | ||
| 64 | |||
| 65 | virtual std::pair<s32, Errno> Send(const std::vector<u8>& message, int flags) = 0; | ||
| 66 | |||
| 67 | virtual std::pair<s32, Errno> SendTo(u32 flags, const std::vector<u8>& message, | ||
| 68 | const SockAddrIn* addr) = 0; | ||
| 69 | |||
| 70 | virtual Errno SetLinger(bool enable, u32 linger) = 0; | ||
| 71 | |||
| 72 | virtual Errno SetReuseAddr(bool enable) = 0; | ||
| 73 | |||
| 74 | virtual Errno SetKeepAlive(bool enable) = 0; | ||
| 75 | |||
| 76 | virtual Errno SetBroadcast(bool enable) = 0; | ||
| 28 | 77 | ||
| 29 | explicit Socket() = default; | 78 | virtual Errno SetSndBuf(u32 value) = 0; |
| 30 | ~Socket(); | 79 | |
| 80 | virtual Errno SetRcvBuf(u32 value) = 0; | ||
| 81 | |||
| 82 | virtual Errno SetSndTimeo(u32 value) = 0; | ||
| 83 | |||
| 84 | virtual Errno SetRcvTimeo(u32 value) = 0; | ||
| 85 | |||
| 86 | virtual Errno SetNonBlock(bool enable) = 0; | ||
| 87 | |||
| 88 | virtual bool IsOpened() const = 0; | ||
| 89 | |||
| 90 | virtual void HandleProxyPacket(const ProxyPacket& packet) = 0; | ||
| 91 | |||
| 92 | #if defined(_WIN32) | ||
| 93 | SOCKET fd = INVALID_SOCKET; | ||
| 94 | #elif YUZU_UNIX | ||
| 95 | int fd = -1; | ||
| 96 | #endif | ||
| 97 | }; | ||
| 98 | |||
| 99 | class Socket : public SocketBase { | ||
| 100 | public: | ||
| 101 | Socket() = default; | ||
| 102 | ~Socket() override; | ||
| 31 | 103 | ||
| 32 | Socket(const Socket&) = delete; | 104 | Socket(const Socket&) = delete; |
| 33 | Socket& operator=(const Socket&) = delete; | 105 | Socket& operator=(const Socket&) = delete; |
| @@ -37,57 +109,57 @@ public: | |||
| 37 | // Avoid closing sockets implicitly | 109 | // Avoid closing sockets implicitly |
| 38 | Socket& operator=(Socket&&) noexcept = delete; | 110 | Socket& operator=(Socket&&) noexcept = delete; |
| 39 | 111 | ||
| 40 | Errno Initialize(Domain domain, Type type, Protocol protocol); | 112 | Errno Initialize(Domain domain, Type type, Protocol protocol) override; |
| 41 | 113 | ||
| 42 | Errno Close(); | 114 | Errno Close() override; |
| 43 | 115 | ||
| 44 | std::pair<AcceptResult, Errno> Accept(); | 116 | std::pair<AcceptResult, Errno> Accept() override; |
| 45 | 117 | ||
| 46 | Errno Connect(SockAddrIn addr_in); | 118 | Errno Connect(SockAddrIn addr_in) override; |
| 47 | 119 | ||
| 48 | std::pair<SockAddrIn, Errno> GetPeerName(); | 120 | std::pair<SockAddrIn, Errno> GetPeerName() override; |
| 49 | 121 | ||
| 50 | std::pair<SockAddrIn, Errno> GetSockName(); | 122 | std::pair<SockAddrIn, Errno> GetSockName() override; |
| 51 | 123 | ||
| 52 | Errno Bind(SockAddrIn addr); | 124 | Errno Bind(SockAddrIn addr) override; |
| 53 | 125 | ||
| 54 | Errno Listen(s32 backlog); | 126 | Errno Listen(s32 backlog) override; |
| 55 | 127 | ||
| 56 | Errno Shutdown(ShutdownHow how); | 128 | Errno Shutdown(ShutdownHow how) override; |
| 57 | 129 | ||
| 58 | std::pair<s32, Errno> Recv(int flags, std::vector<u8>& message); | 130 | std::pair<s32, Errno> Recv(int flags, std::vector<u8>& message) override; |
| 59 | 131 | ||
| 60 | std::pair<s32, Errno> RecvFrom(int flags, std::vector<u8>& message, SockAddrIn* addr); | 132 | std::pair<s32, Errno> RecvFrom(int flags, std::vector<u8>& message, SockAddrIn* addr) override; |
| 61 | 133 | ||
| 62 | std::pair<s32, Errno> Send(const std::vector<u8>& message, int flags); | 134 | std::pair<s32, Errno> Send(const std::vector<u8>& message, int flags) override; |
| 63 | 135 | ||
| 64 | std::pair<s32, Errno> SendTo(u32 flags, const std::vector<u8>& message, const SockAddrIn* addr); | 136 | std::pair<s32, Errno> SendTo(u32 flags, const std::vector<u8>& message, |
| 137 | const SockAddrIn* addr) override; | ||
| 65 | 138 | ||
| 66 | Errno SetLinger(bool enable, u32 linger); | 139 | Errno SetLinger(bool enable, u32 linger) override; |
| 67 | 140 | ||
| 68 | Errno SetReuseAddr(bool enable); | 141 | Errno SetReuseAddr(bool enable) override; |
| 69 | 142 | ||
| 70 | Errno SetKeepAlive(bool enable); | 143 | Errno SetKeepAlive(bool enable) override; |
| 71 | 144 | ||
| 72 | Errno SetBroadcast(bool enable); | 145 | Errno SetBroadcast(bool enable) override; |
| 73 | 146 | ||
| 74 | Errno SetSndBuf(u32 value); | 147 | Errno SetSndBuf(u32 value) override; |
| 75 | 148 | ||
| 76 | Errno SetRcvBuf(u32 value); | 149 | Errno SetRcvBuf(u32 value) override; |
| 77 | 150 | ||
| 78 | Errno SetSndTimeo(u32 value); | 151 | Errno SetSndTimeo(u32 value) override; |
| 79 | 152 | ||
| 80 | Errno SetRcvTimeo(u32 value); | 153 | Errno SetRcvTimeo(u32 value) override; |
| 81 | 154 | ||
| 82 | Errno SetNonBlock(bool enable); | 155 | Errno SetNonBlock(bool enable) override; |
| 83 | 156 | ||
| 84 | bool IsOpened() const; | 157 | template <typename T> |
| 158 | Errno SetSockOpt(SOCKET fd, int option, T value); | ||
| 85 | 159 | ||
| 86 | #if defined(_WIN32) | 160 | bool IsOpened() const override; |
| 87 | SOCKET fd = INVALID_SOCKET; | 161 | |
| 88 | #elif YUZU_UNIX | 162 | void HandleProxyPacket(const ProxyPacket& packet) override; |
| 89 | int fd = -1; | ||
| 90 | #endif | ||
| 91 | }; | 163 | }; |
| 92 | 164 | ||
| 93 | std::pair<s32, Errno> Poll(std::vector<PollFD>& poll_fds, s32 timeout); | 165 | std::pair<s32, Errno> Poll(std::vector<PollFD>& poll_fds, s32 timeout); |