Compare commits
5 Commits
ac951b2afa
...
net_prof
Author | SHA1 | Date | |
---|---|---|---|
037ae18d53 | |||
9ab3e84433 | |||
03633e41a2 | |||
ff3512a77e | |||
1679883328 |
@ -27,18 +27,9 @@ option(TOMATO_TOX_AV "Build tomato with ToxAV" OFF)
|
|||||||
if (TOMATO_ASAN)
|
if (TOMATO_ASAN)
|
||||||
if (${CMAKE_CXX_COMPILER_ID} STREQUAL "GNU" OR ${CMAKE_CXX_COMPILER_ID} STREQUAL "Clang")
|
if (${CMAKE_CXX_COMPILER_ID} STREQUAL "GNU" OR ${CMAKE_CXX_COMPILER_ID} STREQUAL "Clang")
|
||||||
if (NOT WIN32) # exclude mingw
|
if (NOT WIN32) # exclude mingw
|
||||||
add_compile_options(-fno-omit-frame-pointer)
|
#link_libraries(-fsanitize=address)
|
||||||
add_compile_options(-fsanitize=address,undefined)
|
link_libraries(-fsanitize=address,undefined)
|
||||||
#add_compile_options(-fsanitize=address,undefined,pointer-compare,pointer-subtract)
|
#link_libraries(-fsanitize=undefined)
|
||||||
#add_compile_options(-fhardened)
|
|
||||||
#add_compile_options(-D_FORTIFY_SOURCE=3 -D_GLIBCXX_ASSERTIONS -ftrivial-auto-var-init=zero -fPIE -pie -Wl,-z,relro,-z,now -fstack-protector-strong -fstack-clash-protection -fcf-protection=full)
|
|
||||||
|
|
||||||
add_link_options(-fno-omit-frame-pointer)
|
|
||||||
add_link_options(-fsanitize=address,undefined)
|
|
||||||
#add_link_options(-fsanitize=address,undefined,pointer-compare,pointer-subtract)
|
|
||||||
#add_link_options(-fhardened)
|
|
||||||
#add_link_options(-D_FORTIFY_SOURCE=3 -D_GLIBCXX_ASSERTIONS -ftrivial-auto-var-init=zero -fPIE -pie -Wl,-z,relro,-z,now -fstack-protector-strong -fstack-clash-protection -fcf-protection=full)
|
|
||||||
|
|
||||||
link_libraries(-static-libasan) # make it "work" on nix
|
link_libraries(-static-libasan) # make it "work" on nix
|
||||||
message("II enabled ASAN")
|
message("II enabled ASAN")
|
||||||
else()
|
else()
|
||||||
|
2
external/solanaceae_object_store
vendored
2
external/solanaceae_object_store
vendored
Submodule external/solanaceae_object_store updated: 18d2888e34...ed640ba08c
2
external/solanaceae_util
vendored
2
external/solanaceae_util
vendored
Submodule external/solanaceae_util updated: 85bbbb0e5a...717748e8fc
81
external/toxcore/c-toxcore/.circleci/config.yml
vendored
81
external/toxcore/c-toxcore/.circleci/config.yml
vendored
@ -3,15 +3,21 @@ version: 2
|
|||||||
|
|
||||||
workflows:
|
workflows:
|
||||||
version: 2
|
version: 2
|
||||||
circleci:
|
program-analysis:
|
||||||
|
# TODO(iphydf): Re-enable tsan when it's fixed.
|
||||||
jobs:
|
jobs:
|
||||||
|
# Dynamic analysis in the Bazel build
|
||||||
- bazel-asan
|
- bazel-asan
|
||||||
- bazel-msan
|
- bazel-msan
|
||||||
|
# - bazel-tsan
|
||||||
|
# Dynamic analysis with CMake
|
||||||
|
- asan
|
||||||
|
# - tsan
|
||||||
|
- ubsan
|
||||||
|
# Static analysis
|
||||||
- clang-analyze
|
- clang-analyze
|
||||||
- cpplint
|
- cpplint
|
||||||
- static-analysis
|
- static-analysis
|
||||||
- cimplefmt
|
|
||||||
- generate-events
|
|
||||||
|
|
||||||
jobs:
|
jobs:
|
||||||
bazel-asan:
|
bazel-asan:
|
||||||
@ -24,6 +30,20 @@ jobs:
|
|||||||
- run: .circleci/bazel-test
|
- run: .circleci/bazel-test
|
||||||
//c-toxcore/...
|
//c-toxcore/...
|
||||||
|
|
||||||
|
bazel-tsan:
|
||||||
|
working_directory: /tmp/cirrus-ci-build
|
||||||
|
docker:
|
||||||
|
- image: toxchat/toktok-stack:latest-tsan
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- checkout
|
||||||
|
- run: .circleci/bazel-test
|
||||||
|
//c-toxcore/...
|
||||||
|
-//c-toxcore/auto_tests:conference_av_test
|
||||||
|
-//c-toxcore/auto_tests:conference_test
|
||||||
|
-//c-toxcore/auto_tests:onion_test
|
||||||
|
-//c-toxcore/auto_tests:tox_many_test
|
||||||
|
|
||||||
bazel-msan:
|
bazel-msan:
|
||||||
working_directory: /tmp/cirrus-ci-build
|
working_directory: /tmp/cirrus-ci-build
|
||||||
docker:
|
docker:
|
||||||
@ -34,7 +54,7 @@ jobs:
|
|||||||
- run: .circleci/bazel-test
|
- run: .circleci/bazel-test
|
||||||
//c-toxcore/auto_tests:lossless_packet_test
|
//c-toxcore/auto_tests:lossless_packet_test
|
||||||
|
|
||||||
static-analysis:
|
asan:
|
||||||
working_directory: ~/work
|
working_directory: ~/work
|
||||||
docker:
|
docker:
|
||||||
- image: ubuntu
|
- image: ubuntu
|
||||||
@ -48,7 +68,6 @@ jobs:
|
|||||||
clang
|
clang
|
||||||
cmake
|
cmake
|
||||||
git
|
git
|
||||||
libbenchmark-dev
|
|
||||||
libconfig-dev
|
libconfig-dev
|
||||||
libgmock-dev
|
libgmock-dev
|
||||||
libgtest-dev
|
libgtest-dev
|
||||||
@ -58,6 +77,39 @@ jobs:
|
|||||||
llvm-dev
|
llvm-dev
|
||||||
ninja-build
|
ninja-build
|
||||||
pkg-config
|
pkg-config
|
||||||
|
- checkout
|
||||||
|
- run: git submodule update --init --recursive
|
||||||
|
- run: CC=clang .circleci/cmake-asan
|
||||||
|
|
||||||
|
tsan:
|
||||||
|
working_directory: ~/work
|
||||||
|
docker:
|
||||||
|
- image: ubuntu
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- run: *apt_install
|
||||||
|
- checkout
|
||||||
|
- run: git submodule update --init --recursive
|
||||||
|
- run: CC=clang .circleci/cmake-tsan
|
||||||
|
|
||||||
|
ubsan:
|
||||||
|
working_directory: ~/work
|
||||||
|
docker:
|
||||||
|
- image: ubuntu
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- run: *apt_install
|
||||||
|
- checkout
|
||||||
|
- run: git submodule update --init --recursive
|
||||||
|
- run: CC=clang .circleci/cmake-ubsan
|
||||||
|
|
||||||
|
static-analysis:
|
||||||
|
working_directory: ~/work
|
||||||
|
docker:
|
||||||
|
- image: ubuntu
|
||||||
|
|
||||||
|
steps:
|
||||||
|
- run: *apt_install
|
||||||
- run:
|
- run:
|
||||||
apt-get install -y --no-install-recommends
|
apt-get install -y --no-install-recommends
|
||||||
ca-certificates
|
ca-certificates
|
||||||
@ -94,22 +146,3 @@ jobs:
|
|||||||
- checkout
|
- checkout
|
||||||
- run: git submodule update --init --recursive
|
- run: git submodule update --init --recursive
|
||||||
- run: other/analysis/run-cpplint
|
- run: other/analysis/run-cpplint
|
||||||
|
|
||||||
cimplefmt:
|
|
||||||
working_directory: ~/work
|
|
||||||
machine: { image: ubuntu-2204:current }
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- checkout
|
|
||||||
- run: git submodule update --init --recursive
|
|
||||||
- run: other/docker/cimplefmt/run -u $(find tox* -name "*.[ch]")
|
|
||||||
|
|
||||||
generate-events:
|
|
||||||
working_directory: ~/work
|
|
||||||
machine: { image: ubuntu-2204:current }
|
|
||||||
|
|
||||||
steps:
|
|
||||||
- checkout
|
|
||||||
- run: git submodule update --init --recursive
|
|
||||||
- run: other/event_tooling/run
|
|
||||||
- run: git diff --exit-code
|
|
||||||
|
@ -18,7 +18,7 @@ jobs:
|
|||||||
fail-fast: false
|
fail-fast: false
|
||||||
matrix:
|
matrix:
|
||||||
tool: [autotools, clang-tidy, compcert, cppcheck, doxygen, goblint, infer, freebsd, misra, modules, pkgsrc, rpm, slimcc, sparse, tcc, tokstyle]
|
tool: [autotools, clang-tidy, compcert, cppcheck, doxygen, goblint, infer, freebsd, misra, modules, pkgsrc, rpm, slimcc, sparse, tcc, tokstyle]
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- name: Set up Docker Buildx
|
- name: Set up Docker Buildx
|
||||||
uses: docker/setup-buildx-action@v3
|
uses: docker/setup-buildx-action@v3
|
||||||
@ -34,25 +34,8 @@ jobs:
|
|||||||
with:
|
with:
|
||||||
file: other/docker/${{ matrix.tool }}/${{ matrix.tool }}.Dockerfile
|
file: other/docker/${{ matrix.tool }}/${{ matrix.tool }}.Dockerfile
|
||||||
|
|
||||||
sanitizer:
|
|
||||||
strategy:
|
|
||||||
fail-fast: false
|
|
||||||
matrix:
|
|
||||||
sanitizer: [asan, tsan, ubsan]
|
|
||||||
runs-on: ubuntu-22.04
|
|
||||||
steps:
|
|
||||||
- name: Set up Docker Buildx
|
|
||||||
uses: docker/setup-buildx-action@v3
|
|
||||||
with:
|
|
||||||
driver: docker
|
|
||||||
- uses: actions/checkout@v4
|
|
||||||
with:
|
|
||||||
submodules: recursive
|
|
||||||
- name: Run sanitizer
|
|
||||||
run: other/docker/circleci/run "${{ matrix.sanitizer }}"
|
|
||||||
|
|
||||||
coverage-linux:
|
coverage-linux:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
@ -60,8 +43,28 @@ jobs:
|
|||||||
- name: Build, test, and upload coverage
|
- name: Build, test, and upload coverage
|
||||||
run: other/docker/coverage/run
|
run: other/docker/coverage/run
|
||||||
|
|
||||||
|
generate-events:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
submodules: recursive
|
||||||
|
- name: Run generate_event_c
|
||||||
|
run: |
|
||||||
|
other/event_tooling/run
|
||||||
|
git diff --exit-code
|
||||||
|
|
||||||
|
cimplefmt:
|
||||||
|
runs-on: ubuntu-latest
|
||||||
|
steps:
|
||||||
|
- uses: actions/checkout@v4
|
||||||
|
with:
|
||||||
|
submodules: recursive
|
||||||
|
- name: Run cimplefmt
|
||||||
|
run: other/docker/cimplefmt/run -u $(find tox* -name "*.[ch]")
|
||||||
|
|
||||||
build-android:
|
build-android:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
@ -108,7 +111,7 @@ jobs:
|
|||||||
ctest -j50 --output-on-failure --rerun-failed --repeat until-pass:6 --build-config Debug
|
ctest -j50 --output-on-failure --rerun-failed --repeat until-pass:6 --build-config Debug
|
||||||
|
|
||||||
build-netbsd:
|
build-netbsd:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
@ -143,7 +146,7 @@ jobs:
|
|||||||
ctest -j50 --output-on-failure --rerun-failed --repeat until-pass:6
|
ctest -j50 --output-on-failure --rerun-failed --repeat until-pass:6
|
||||||
|
|
||||||
build-freebsd:
|
build-freebsd:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
@ -180,7 +183,7 @@ jobs:
|
|||||||
ctest -j50 --output-on-failure --rerun-failed --repeat until-pass:6
|
ctest -j50 --output-on-failure --rerun-failed --repeat until-pass:6
|
||||||
|
|
||||||
mypy:
|
mypy:
|
||||||
runs-on: ubuntu-22.04
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@v4
|
||||||
with:
|
with:
|
||||||
|
1
external/toxcore/c-toxcore/.gitignore
vendored
1
external/toxcore/c-toxcore/.gitignore
vendored
@ -36,6 +36,7 @@ testing/data
|
|||||||
|
|
||||||
# Vim
|
# Vim
|
||||||
*.swp
|
*.swp
|
||||||
|
*.nvimlog
|
||||||
|
|
||||||
# Object files
|
# Object files
|
||||||
*.o
|
*.o
|
||||||
|
4
external/toxcore/c-toxcore/CMakeLists.txt
vendored
4
external/toxcore/c-toxcore/CMakeLists.txt
vendored
@ -309,6 +309,8 @@ set(toxcore_SOURCES
|
|||||||
toxcore/mono_time.h
|
toxcore/mono_time.h
|
||||||
toxcore/net_crypto.c
|
toxcore/net_crypto.c
|
||||||
toxcore/net_crypto.h
|
toxcore/net_crypto.h
|
||||||
|
toxcore/net_profile.c
|
||||||
|
toxcore/net_profile.h
|
||||||
toxcore/network.c
|
toxcore/network.c
|
||||||
toxcore/network.h
|
toxcore/network.h
|
||||||
toxcore/onion_announce.c
|
toxcore/onion_announce.c
|
||||||
@ -323,8 +325,6 @@ set(toxcore_SOURCES
|
|||||||
toxcore/ping.h
|
toxcore/ping.h
|
||||||
toxcore/shared_key_cache.c
|
toxcore/shared_key_cache.c
|
||||||
toxcore/shared_key_cache.h
|
toxcore/shared_key_cache.h
|
||||||
toxcore/sort.c
|
|
||||||
toxcore/sort.h
|
|
||||||
toxcore/state.c
|
toxcore/state.c
|
||||||
toxcore/state.h
|
toxcore/state.h
|
||||||
toxcore/TCP_client.c
|
toxcore/TCP_client.c
|
||||||
|
@ -78,7 +78,6 @@ extra_data = {
|
|||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
"//c-toxcore/toxcore:tox_dispatch",
|
"//c-toxcore/toxcore:tox_dispatch",
|
||||||
"//c-toxcore/toxcore:tox_events",
|
"//c-toxcore/toxcore:tox_events",
|
||||||
"//c-toxcore/toxcore:tox_unpack",
|
|
||||||
"//c-toxcore/toxcore:util",
|
"//c-toxcore/toxcore:util",
|
||||||
"//c-toxcore/toxencryptsave",
|
"//c-toxcore/toxencryptsave",
|
||||||
"@libsodium",
|
"@libsodium",
|
||||||
|
@ -71,6 +71,7 @@ auto_test(invalid_udp_proxy)
|
|||||||
auto_test(lan_discovery)
|
auto_test(lan_discovery)
|
||||||
auto_test(lossless_packet)
|
auto_test(lossless_packet)
|
||||||
auto_test(lossy_packet)
|
auto_test(lossy_packet)
|
||||||
|
auto_test(netprof)
|
||||||
auto_test(network)
|
auto_test(network)
|
||||||
auto_test(onion)
|
auto_test(onion)
|
||||||
auto_test(overflow_recvq)
|
auto_test(overflow_recvq)
|
||||||
|
34
external/toxcore/c-toxcore/auto_tests/TCP_test.c
vendored
34
external/toxcore/c-toxcore/auto_tests/TCP_test.c
vendored
@ -102,7 +102,7 @@ static void test_basic(void)
|
|||||||
random_nonce(rng, handshake + CRYPTO_PUBLIC_KEY_SIZE);
|
random_nonce(rng, handshake + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
// Encrypting handshake
|
// Encrypting handshake
|
||||||
int ret = encrypt_data(mem, self_public_key, f_secret_key, handshake + CRYPTO_PUBLIC_KEY_SIZE, handshake_plain,
|
int ret = encrypt_data(self_public_key, f_secret_key, handshake + CRYPTO_PUBLIC_KEY_SIZE, handshake_plain,
|
||||||
TCP_HANDSHAKE_PLAIN_SIZE, handshake + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
TCP_HANDSHAKE_PLAIN_SIZE, handshake + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
ck_assert_msg(ret == TCP_CLIENT_HANDSHAKE_SIZE - (CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE),
|
ck_assert_msg(ret == TCP_CLIENT_HANDSHAKE_SIZE - (CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE),
|
||||||
"encrypt_data() call failed.");
|
"encrypt_data() call failed.");
|
||||||
@ -111,12 +111,12 @@ static void test_basic(void)
|
|||||||
|
|
||||||
// Sending the handshake
|
// Sending the handshake
|
||||||
ck_assert_msg(net_send(ns, logger, sock, handshake, TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
ck_assert_msg(net_send(ns, logger, sock, handshake, TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
||||||
&localhost) == TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
&localhost, nullptr) == TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
||||||
"An attempt to send the initial handshake minus last byte failed.");
|
"An attempt to send the initial handshake minus last byte failed.");
|
||||||
|
|
||||||
do_tcp_server_delay(tcp_s, mono_time, 50);
|
do_tcp_server_delay(tcp_s, mono_time, 50);
|
||||||
|
|
||||||
ck_assert_msg(net_send(ns, logger, sock, handshake + (TCP_CLIENT_HANDSHAKE_SIZE - 1), 1, &localhost) == 1,
|
ck_assert_msg(net_send(ns, logger, sock, handshake + (TCP_CLIENT_HANDSHAKE_SIZE - 1), 1, &localhost, nullptr) == 1,
|
||||||
"The attempt to send the last byte of handshake failed.");
|
"The attempt to send the last byte of handshake failed.");
|
||||||
|
|
||||||
free(handshake);
|
free(handshake);
|
||||||
@ -128,7 +128,7 @@ static void test_basic(void)
|
|||||||
uint8_t response_plain[TCP_HANDSHAKE_PLAIN_SIZE];
|
uint8_t response_plain[TCP_HANDSHAKE_PLAIN_SIZE];
|
||||||
ck_assert_msg(net_recv(ns, logger, sock, response, TCP_SERVER_HANDSHAKE_SIZE, &localhost) == TCP_SERVER_HANDSHAKE_SIZE,
|
ck_assert_msg(net_recv(ns, logger, sock, response, TCP_SERVER_HANDSHAKE_SIZE, &localhost) == TCP_SERVER_HANDSHAKE_SIZE,
|
||||||
"Could/did not receive a server response to the initial handshake.");
|
"Could/did not receive a server response to the initial handshake.");
|
||||||
ret = decrypt_data(mem, self_public_key, f_secret_key, response, response + CRYPTO_NONCE_SIZE,
|
ret = decrypt_data(self_public_key, f_secret_key, response, response + CRYPTO_NONCE_SIZE,
|
||||||
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, response_plain);
|
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, response_plain);
|
||||||
ck_assert_msg(ret == TCP_HANDSHAKE_PLAIN_SIZE, "Failed to decrypt handshake response.");
|
ck_assert_msg(ret == TCP_HANDSHAKE_PLAIN_SIZE, "Failed to decrypt handshake response.");
|
||||||
uint8_t f_nonce_r[CRYPTO_NONCE_SIZE];
|
uint8_t f_nonce_r[CRYPTO_NONCE_SIZE];
|
||||||
@ -143,7 +143,7 @@ static void test_basic(void)
|
|||||||
uint8_t r_req[2 + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_MAC_SIZE];
|
uint8_t r_req[2 + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_MAC_SIZE];
|
||||||
uint16_t size = 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_MAC_SIZE;
|
uint16_t size = 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_MAC_SIZE;
|
||||||
size = net_htons(size);
|
size = net_htons(size);
|
||||||
encrypt_data_symmetric(mem, f_shared_key, f_nonce, r_req_p, 1 + CRYPTO_PUBLIC_KEY_SIZE, r_req + 2);
|
encrypt_data_symmetric(f_shared_key, f_nonce, r_req_p, 1 + CRYPTO_PUBLIC_KEY_SIZE, r_req + 2);
|
||||||
increment_nonce(f_nonce);
|
increment_nonce(f_nonce);
|
||||||
memcpy(r_req, &size, 2);
|
memcpy(r_req, &size, 2);
|
||||||
|
|
||||||
@ -155,7 +155,7 @@ static void test_basic(void)
|
|||||||
msg_length = sizeof(r_req) - i;
|
msg_length = sizeof(r_req) - i;
|
||||||
}
|
}
|
||||||
|
|
||||||
ck_assert_msg(net_send(ns, logger, sock, r_req + i, msg_length, &localhost) == msg_length,
|
ck_assert_msg(net_send(ns, logger, sock, r_req + i, msg_length, &localhost, nullptr) == msg_length,
|
||||||
"Failed to send request after completing the handshake.");
|
"Failed to send request after completing the handshake.");
|
||||||
i += msg_length;
|
i += msg_length;
|
||||||
|
|
||||||
@ -174,7 +174,7 @@ static void test_basic(void)
|
|||||||
"Wrong packet size for request response.");
|
"Wrong packet size for request response.");
|
||||||
|
|
||||||
uint8_t packet_resp_plain[4096];
|
uint8_t packet_resp_plain[4096];
|
||||||
ret = decrypt_data_symmetric(mem, f_shared_key, f_nonce_r, packet_resp + 2, recv_data_len - 2, packet_resp_plain);
|
ret = decrypt_data_symmetric(f_shared_key, f_nonce_r, packet_resp + 2, recv_data_len - 2, packet_resp_plain);
|
||||||
ck_assert_msg(ret != -1, "Failed to decrypt the TCP server's response.");
|
ck_assert_msg(ret != -1, "Failed to decrypt the TCP server's response.");
|
||||||
increment_nonce(f_nonce_r);
|
increment_nonce(f_nonce_r);
|
||||||
|
|
||||||
@ -228,18 +228,18 @@ static struct sec_TCP_con *new_tcp_con(const Logger *logger, const Memory *mem,
|
|||||||
memcpy(handshake, sec_c->public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(handshake, sec_c->public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
random_nonce(rng, handshake + CRYPTO_PUBLIC_KEY_SIZE);
|
random_nonce(rng, handshake + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
int ret = encrypt_data(mem, tcp_server_public_key(tcp_s), f_secret_key, handshake + CRYPTO_PUBLIC_KEY_SIZE, handshake_plain,
|
int ret = encrypt_data(tcp_server_public_key(tcp_s), f_secret_key, handshake + CRYPTO_PUBLIC_KEY_SIZE, handshake_plain,
|
||||||
TCP_HANDSHAKE_PLAIN_SIZE, handshake + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
TCP_HANDSHAKE_PLAIN_SIZE, handshake + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
ck_assert_msg(ret == TCP_CLIENT_HANDSHAKE_SIZE - (CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE),
|
ck_assert_msg(ret == TCP_CLIENT_HANDSHAKE_SIZE - (CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE),
|
||||||
"Failed to encrypt the outgoing handshake.");
|
"Failed to encrypt the outgoing handshake.");
|
||||||
|
|
||||||
ck_assert_msg(net_send(ns, logger, sock, handshake, TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
ck_assert_msg(net_send(ns, logger, sock, handshake, TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
||||||
&localhost) == TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
&localhost, nullptr) == TCP_CLIENT_HANDSHAKE_SIZE - 1,
|
||||||
"Failed to send the first portion of the handshake to the TCP relay server.");
|
"Failed to send the first portion of the handshake to the TCP relay server.");
|
||||||
|
|
||||||
do_tcp_server_delay(tcp_s, mono_time, 50);
|
do_tcp_server_delay(tcp_s, mono_time, 50);
|
||||||
|
|
||||||
ck_assert_msg(net_send(ns, logger, sock, handshake + (TCP_CLIENT_HANDSHAKE_SIZE - 1), 1, &localhost) == 1,
|
ck_assert_msg(net_send(ns, logger, sock, handshake + (TCP_CLIENT_HANDSHAKE_SIZE - 1), 1, &localhost, nullptr) == 1,
|
||||||
"Failed to send last byte of handshake.");
|
"Failed to send last byte of handshake.");
|
||||||
|
|
||||||
do_tcp_server_delay(tcp_s, mono_time, 50);
|
do_tcp_server_delay(tcp_s, mono_time, 50);
|
||||||
@ -248,7 +248,7 @@ static struct sec_TCP_con *new_tcp_con(const Logger *logger, const Memory *mem,
|
|||||||
uint8_t response_plain[TCP_HANDSHAKE_PLAIN_SIZE];
|
uint8_t response_plain[TCP_HANDSHAKE_PLAIN_SIZE];
|
||||||
ck_assert_msg(net_recv(sec_c->ns, logger, sock, response, TCP_SERVER_HANDSHAKE_SIZE, &localhost) == TCP_SERVER_HANDSHAKE_SIZE,
|
ck_assert_msg(net_recv(sec_c->ns, logger, sock, response, TCP_SERVER_HANDSHAKE_SIZE, &localhost) == TCP_SERVER_HANDSHAKE_SIZE,
|
||||||
"Failed to receive server handshake response.");
|
"Failed to receive server handshake response.");
|
||||||
ret = decrypt_data(mem, tcp_server_public_key(tcp_s), f_secret_key, response, response + CRYPTO_NONCE_SIZE,
|
ret = decrypt_data(tcp_server_public_key(tcp_s), f_secret_key, response, response + CRYPTO_NONCE_SIZE,
|
||||||
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, response_plain);
|
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, response_plain);
|
||||||
ck_assert_msg(ret == TCP_HANDSHAKE_PLAIN_SIZE, "Failed to decrypt server handshake response.");
|
ck_assert_msg(ret == TCP_HANDSHAKE_PLAIN_SIZE, "Failed to decrypt server handshake response.");
|
||||||
encrypt_precompute(response_plain, t_secret_key, sec_c->shared_key);
|
encrypt_precompute(response_plain, t_secret_key, sec_c->shared_key);
|
||||||
@ -271,7 +271,7 @@ static int write_packet_tcp_test_connection(const Logger *logger, struct sec_TCP
|
|||||||
|
|
||||||
uint16_t c_length = net_htons(length + CRYPTO_MAC_SIZE);
|
uint16_t c_length = net_htons(length + CRYPTO_MAC_SIZE);
|
||||||
memcpy(packet, &c_length, sizeof(uint16_t));
|
memcpy(packet, &c_length, sizeof(uint16_t));
|
||||||
int len = encrypt_data_symmetric(con->mem, con->shared_key, con->sent_nonce, data, length, packet + sizeof(uint16_t));
|
int len = encrypt_data_symmetric(con->shared_key, con->sent_nonce, data, length, packet + sizeof(uint16_t));
|
||||||
|
|
||||||
if ((unsigned int)len != (packet_size - sizeof(uint16_t))) {
|
if ((unsigned int)len != (packet_size - sizeof(uint16_t))) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -283,7 +283,7 @@ static int write_packet_tcp_test_connection(const Logger *logger, struct sec_TCP
|
|||||||
localhost.ip = get_loopback();
|
localhost.ip = get_loopback();
|
||||||
localhost.port = 0;
|
localhost.port = 0;
|
||||||
|
|
||||||
ck_assert_msg(net_send(con->ns, logger, con->sock, packet, packet_size, &localhost) == packet_size,
|
ck_assert_msg(net_send(con->ns, logger, con->sock, packet, packet_size, &localhost, nullptr) == packet_size,
|
||||||
"Failed to send a packet.");
|
"Failed to send a packet.");
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@ -296,7 +296,7 @@ static int read_packet_sec_tcp(const Logger *logger, struct sec_TCP_con *con, ui
|
|||||||
|
|
||||||
int rlen = net_recv(con->ns, logger, con->sock, data, length, &localhost);
|
int rlen = net_recv(con->ns, logger, con->sock, data, length, &localhost);
|
||||||
ck_assert_msg(rlen == length, "Did not receive packet of correct length. Wanted %i, instead got %i", length, rlen);
|
ck_assert_msg(rlen == length, "Did not receive packet of correct length. Wanted %i, instead got %i", length, rlen);
|
||||||
rlen = decrypt_data_symmetric(con->mem, con->shared_key, con->recv_nonce, data + 2, length - 2, data);
|
rlen = decrypt_data_symmetric(con->shared_key, con->recv_nonce, data + 2, length - 2, data);
|
||||||
ck_assert_msg(rlen != -1, "Failed to decrypt a received packet from the Relay server.");
|
ck_assert_msg(rlen != -1, "Failed to decrypt a received packet from the Relay server.");
|
||||||
increment_nonce(con->recv_nonce);
|
increment_nonce(con->recv_nonce);
|
||||||
return rlen;
|
return rlen;
|
||||||
@ -524,7 +524,7 @@ static void test_client(void)
|
|||||||
ip_port_tcp_s.port = net_htons(ports[random_u32(rng) % NUM_PORTS]);
|
ip_port_tcp_s.port = net_htons(ports[random_u32(rng) % NUM_PORTS]);
|
||||||
ip_port_tcp_s.ip = get_loopback();
|
ip_port_tcp_s.ip = get_loopback();
|
||||||
|
|
||||||
TCP_Client_Connection *conn = new_tcp_connection(logger, mem, mono_time, rng, ns, &ip_port_tcp_s, self_public_key, f_public_key, f_secret_key, nullptr);
|
TCP_Client_Connection *conn = new_tcp_connection(logger, mem, mono_time, rng, ns, &ip_port_tcp_s, self_public_key, f_public_key, f_secret_key, nullptr, nullptr);
|
||||||
// TCP sockets might need a moment before they can be written to.
|
// TCP sockets might need a moment before they can be written to.
|
||||||
c_sleep(50);
|
c_sleep(50);
|
||||||
do_tcp_connection(logger, mono_time, conn, nullptr);
|
do_tcp_connection(logger, mono_time, conn, nullptr);
|
||||||
@ -560,7 +560,7 @@ static void test_client(void)
|
|||||||
crypto_new_keypair(rng, f2_public_key, f2_secret_key);
|
crypto_new_keypair(rng, f2_public_key, f2_secret_key);
|
||||||
ip_port_tcp_s.port = net_htons(ports[random_u32(rng) % NUM_PORTS]);
|
ip_port_tcp_s.port = net_htons(ports[random_u32(rng) % NUM_PORTS]);
|
||||||
TCP_Client_Connection *conn2 = new_tcp_connection(logger, mem, mono_time, rng, ns, &ip_port_tcp_s, self_public_key, f2_public_key,
|
TCP_Client_Connection *conn2 = new_tcp_connection(logger, mem, mono_time, rng, ns, &ip_port_tcp_s, self_public_key, f2_public_key,
|
||||||
f2_secret_key, nullptr);
|
f2_secret_key, nullptr, nullptr);
|
||||||
c_sleep(50);
|
c_sleep(50);
|
||||||
|
|
||||||
// The client should call this function (defined earlier) during the routing process.
|
// The client should call this function (defined earlier) during the routing process.
|
||||||
@ -657,7 +657,7 @@ static void test_client_invalid(void)
|
|||||||
ip_port_tcp_s.port = net_htons(ports[random_u32(rng) % NUM_PORTS]);
|
ip_port_tcp_s.port = net_htons(ports[random_u32(rng) % NUM_PORTS]);
|
||||||
ip_port_tcp_s.ip = get_loopback();
|
ip_port_tcp_s.ip = get_loopback();
|
||||||
TCP_Client_Connection *conn = new_tcp_connection(logger, mem, mono_time, rng, ns, &ip_port_tcp_s,
|
TCP_Client_Connection *conn = new_tcp_connection(logger, mem, mono_time, rng, ns, &ip_port_tcp_s,
|
||||||
self_public_key, f_public_key, f_secret_key, nullptr);
|
self_public_key, f_public_key, f_secret_key, nullptr, nullptr);
|
||||||
|
|
||||||
// Run the client's main loop but not the server.
|
// Run the client's main loop but not the server.
|
||||||
mono_time_update(mono_time);
|
mono_time_update(mono_time);
|
||||||
|
@ -80,9 +80,6 @@ static const uint8_t test_c[147] = {
|
|||||||
|
|
||||||
static void test_known(void)
|
static void test_known(void)
|
||||||
{
|
{
|
||||||
const Memory *mem = os_memory();
|
|
||||||
ck_assert(mem != nullptr);
|
|
||||||
|
|
||||||
uint8_t c[147];
|
uint8_t c[147];
|
||||||
uint8_t m[131];
|
uint8_t m[131];
|
||||||
|
|
||||||
@ -91,12 +88,12 @@ static void test_known(void)
|
|||||||
ck_assert_msg(sizeof(test_c) == sizeof(c), "sanity check failed");
|
ck_assert_msg(sizeof(test_c) == sizeof(c), "sanity check failed");
|
||||||
ck_assert_msg(sizeof(test_m) == sizeof(m), "sanity check failed");
|
ck_assert_msg(sizeof(test_m) == sizeof(m), "sanity check failed");
|
||||||
|
|
||||||
const uint16_t clen = encrypt_data(mem, bobpk, alicesk, test_nonce, test_m, sizeof(test_m) / sizeof(uint8_t), c);
|
const uint16_t clen = encrypt_data(bobpk, alicesk, test_nonce, test_m, sizeof(test_m) / sizeof(uint8_t), c);
|
||||||
|
|
||||||
ck_assert_msg(memcmp(test_c, c, sizeof(c)) == 0, "cyphertext doesn't match test vector");
|
ck_assert_msg(memcmp(test_c, c, sizeof(c)) == 0, "cyphertext doesn't match test vector");
|
||||||
ck_assert_msg(clen == sizeof(c) / sizeof(uint8_t), "wrong ciphertext length");
|
ck_assert_msg(clen == sizeof(c) / sizeof(uint8_t), "wrong ciphertext length");
|
||||||
|
|
||||||
const uint16_t mlen = decrypt_data(mem, bobpk, alicesk, test_nonce, test_c, sizeof(test_c) / sizeof(uint8_t), m);
|
const uint16_t mlen = decrypt_data(bobpk, alicesk, test_nonce, test_c, sizeof(test_c) / sizeof(uint8_t), m);
|
||||||
|
|
||||||
ck_assert_msg(memcmp(test_m, m, sizeof(m)) == 0, "decrypted text doesn't match test vector");
|
ck_assert_msg(memcmp(test_m, m, sizeof(m)) == 0, "decrypted text doesn't match test vector");
|
||||||
ck_assert_msg(mlen == sizeof(m) / sizeof(uint8_t), "wrong plaintext length");
|
ck_assert_msg(mlen == sizeof(m) / sizeof(uint8_t), "wrong plaintext length");
|
||||||
@ -104,9 +101,6 @@ static void test_known(void)
|
|||||||
|
|
||||||
static void test_fast_known(void)
|
static void test_fast_known(void)
|
||||||
{
|
{
|
||||||
const Memory *mem = os_memory();
|
|
||||||
ck_assert(mem != nullptr);
|
|
||||||
|
|
||||||
uint8_t k[CRYPTO_SHARED_KEY_SIZE];
|
uint8_t k[CRYPTO_SHARED_KEY_SIZE];
|
||||||
uint8_t c[147];
|
uint8_t c[147];
|
||||||
uint8_t m[131];
|
uint8_t m[131];
|
||||||
@ -118,12 +112,12 @@ static void test_fast_known(void)
|
|||||||
ck_assert_msg(sizeof(test_c) == sizeof(c), "sanity check failed");
|
ck_assert_msg(sizeof(test_c) == sizeof(c), "sanity check failed");
|
||||||
ck_assert_msg(sizeof(test_m) == sizeof(m), "sanity check failed");
|
ck_assert_msg(sizeof(test_m) == sizeof(m), "sanity check failed");
|
||||||
|
|
||||||
const uint16_t clen = encrypt_data_symmetric(mem, k, test_nonce, test_m, sizeof(test_m) / sizeof(uint8_t), c);
|
const uint16_t clen = encrypt_data_symmetric(k, test_nonce, test_m, sizeof(test_m) / sizeof(uint8_t), c);
|
||||||
|
|
||||||
ck_assert_msg(memcmp(test_c, c, sizeof(c)) == 0, "cyphertext doesn't match test vector");
|
ck_assert_msg(memcmp(test_c, c, sizeof(c)) == 0, "cyphertext doesn't match test vector");
|
||||||
ck_assert_msg(clen == sizeof(c) / sizeof(uint8_t), "wrong ciphertext length");
|
ck_assert_msg(clen == sizeof(c) / sizeof(uint8_t), "wrong ciphertext length");
|
||||||
|
|
||||||
const uint16_t mlen = decrypt_data_symmetric(mem, k, test_nonce, test_c, sizeof(test_c) / sizeof(uint8_t), m);
|
const uint16_t mlen = decrypt_data_symmetric(k, test_nonce, test_c, sizeof(test_c) / sizeof(uint8_t), m);
|
||||||
|
|
||||||
ck_assert_msg(memcmp(test_m, m, sizeof(m)) == 0, "decrypted text doesn't match test vector");
|
ck_assert_msg(memcmp(test_m, m, sizeof(m)) == 0, "decrypted text doesn't match test vector");
|
||||||
ck_assert_msg(mlen == sizeof(m) / sizeof(uint8_t), "wrong plaintext length");
|
ck_assert_msg(mlen == sizeof(m) / sizeof(uint8_t), "wrong plaintext length");
|
||||||
@ -131,8 +125,6 @@ static void test_fast_known(void)
|
|||||||
|
|
||||||
static void test_endtoend(void)
|
static void test_endtoend(void)
|
||||||
{
|
{
|
||||||
const Memory *mem = os_memory();
|
|
||||||
ck_assert(mem != nullptr);
|
|
||||||
const Random *rng = os_random();
|
const Random *rng = os_random();
|
||||||
ck_assert(rng != nullptr);
|
ck_assert(rng != nullptr);
|
||||||
|
|
||||||
@ -174,10 +166,10 @@ static void test_endtoend(void)
|
|||||||
ck_assert_msg(memcmp(k1, k2, CRYPTO_SHARED_KEY_SIZE) == 0, "encrypt_precompute: bad");
|
ck_assert_msg(memcmp(k1, k2, CRYPTO_SHARED_KEY_SIZE) == 0, "encrypt_precompute: bad");
|
||||||
|
|
||||||
//Encrypt all four ways
|
//Encrypt all four ways
|
||||||
const uint16_t c1len = encrypt_data(mem, pk2, sk1, n, m, mlen, c1);
|
const uint16_t c1len = encrypt_data(pk2, sk1, n, m, mlen, c1);
|
||||||
const uint16_t c2len = encrypt_data(mem, pk1, sk2, n, m, mlen, c2);
|
const uint16_t c2len = encrypt_data(pk1, sk2, n, m, mlen, c2);
|
||||||
const uint16_t c3len = encrypt_data_symmetric(mem, k1, n, m, mlen, c3);
|
const uint16_t c3len = encrypt_data_symmetric(k1, n, m, mlen, c3);
|
||||||
const uint16_t c4len = encrypt_data_symmetric(mem, k2, n, m, mlen, c4);
|
const uint16_t c4len = encrypt_data_symmetric(k2, n, m, mlen, c4);
|
||||||
|
|
||||||
ck_assert_msg(c1len == c2len && c1len == c3len && c1len == c4len, "cyphertext lengths differ");
|
ck_assert_msg(c1len == c2len && c1len == c3len && c1len == c4len, "cyphertext lengths differ");
|
||||||
ck_assert_msg(c1len == mlen + (uint16_t)CRYPTO_MAC_SIZE, "wrong cyphertext length");
|
ck_assert_msg(c1len == mlen + (uint16_t)CRYPTO_MAC_SIZE, "wrong cyphertext length");
|
||||||
@ -185,10 +177,10 @@ static void test_endtoend(void)
|
|||||||
&& memcmp(c1, c4, c1len) == 0, "crypertexts differ");
|
&& memcmp(c1, c4, c1len) == 0, "crypertexts differ");
|
||||||
|
|
||||||
//Decrypt all four ways
|
//Decrypt all four ways
|
||||||
const uint16_t m1len = decrypt_data(mem, pk2, sk1, n, c1, c1len, m1);
|
const uint16_t m1len = decrypt_data(pk2, sk1, n, c1, c1len, m1);
|
||||||
const uint16_t m2len = decrypt_data(mem, pk1, sk2, n, c1, c1len, m2);
|
const uint16_t m2len = decrypt_data(pk1, sk2, n, c1, c1len, m2);
|
||||||
const uint16_t m3len = decrypt_data_symmetric(mem, k1, n, c1, c1len, m3);
|
const uint16_t m3len = decrypt_data_symmetric(k1, n, c1, c1len, m3);
|
||||||
const uint16_t m4len = decrypt_data_symmetric(mem, k2, n, c1, c1len, m4);
|
const uint16_t m4len = decrypt_data_symmetric(k2, n, c1, c1len, m4);
|
||||||
|
|
||||||
ck_assert_msg(m1len == m2len && m1len == m3len && m1len == m4len, "decrypted text lengths differ");
|
ck_assert_msg(m1len == m2len && m1len == m3len && m1len == m4len, "decrypted text lengths differ");
|
||||||
ck_assert_msg(m1len == mlen, "wrong decrypted text length");
|
ck_assert_msg(m1len == mlen, "wrong decrypted text length");
|
||||||
@ -200,8 +192,6 @@ static void test_endtoend(void)
|
|||||||
|
|
||||||
static void test_large_data(void)
|
static void test_large_data(void)
|
||||||
{
|
{
|
||||||
const Memory *mem = os_memory();
|
|
||||||
ck_assert(mem != nullptr);
|
|
||||||
const Random *rng = os_random();
|
const Random *rng = os_random();
|
||||||
ck_assert(rng != nullptr);
|
ck_assert(rng != nullptr);
|
||||||
uint8_t k[CRYPTO_SHARED_KEY_SIZE];
|
uint8_t k[CRYPTO_SHARED_KEY_SIZE];
|
||||||
@ -226,13 +216,13 @@ static void test_large_data(void)
|
|||||||
//Generate key
|
//Generate key
|
||||||
rand_bytes(rng, k, CRYPTO_SHARED_KEY_SIZE);
|
rand_bytes(rng, k, CRYPTO_SHARED_KEY_SIZE);
|
||||||
|
|
||||||
const uint16_t c1len = encrypt_data_symmetric(mem, k, n, m1, m1_size, c1);
|
const uint16_t c1len = encrypt_data_symmetric(k, n, m1, m1_size, c1);
|
||||||
const uint16_t c2len = encrypt_data_symmetric(mem, k, n, m2, m2_size, c2);
|
const uint16_t c2len = encrypt_data_symmetric(k, n, m2, m2_size, c2);
|
||||||
|
|
||||||
ck_assert_msg(c1len == m1_size + CRYPTO_MAC_SIZE, "could not encrypt");
|
ck_assert_msg(c1len == m1_size + CRYPTO_MAC_SIZE, "could not encrypt");
|
||||||
ck_assert_msg(c2len == m2_size + CRYPTO_MAC_SIZE, "could not encrypt");
|
ck_assert_msg(c2len == m2_size + CRYPTO_MAC_SIZE, "could not encrypt");
|
||||||
|
|
||||||
const uint16_t m1plen = decrypt_data_symmetric(mem, k, n, c1, c1len, m1prime);
|
const uint16_t m1plen = decrypt_data_symmetric(k, n, c1, c1len, m1prime);
|
||||||
|
|
||||||
ck_assert_msg(m1plen == m1_size, "decrypted text lengths differ");
|
ck_assert_msg(m1plen == m1_size, "decrypted text lengths differ");
|
||||||
ck_assert_msg(memcmp(m1prime, m1, m1_size) == 0, "decrypted texts differ");
|
ck_assert_msg(memcmp(m1prime, m1, m1_size) == 0, "decrypted texts differ");
|
||||||
@ -246,8 +236,6 @@ static void test_large_data(void)
|
|||||||
|
|
||||||
static void test_large_data_symmetric(void)
|
static void test_large_data_symmetric(void)
|
||||||
{
|
{
|
||||||
const Memory *mem = os_memory();
|
|
||||||
ck_assert(mem != nullptr);
|
|
||||||
const Random *rng = os_random();
|
const Random *rng = os_random();
|
||||||
ck_assert(rng != nullptr);
|
ck_assert(rng != nullptr);
|
||||||
uint8_t k[CRYPTO_SYMMETRIC_KEY_SIZE];
|
uint8_t k[CRYPTO_SYMMETRIC_KEY_SIZE];
|
||||||
@ -268,10 +256,10 @@ static void test_large_data_symmetric(void)
|
|||||||
//Generate key
|
//Generate key
|
||||||
new_symmetric_key(rng, k);
|
new_symmetric_key(rng, k);
|
||||||
|
|
||||||
const uint16_t c1len = encrypt_data_symmetric(mem, k, n, m1, m1_size, c1);
|
const uint16_t c1len = encrypt_data_symmetric(k, n, m1, m1_size, c1);
|
||||||
ck_assert_msg(c1len == m1_size + CRYPTO_MAC_SIZE, "could not encrypt data");
|
ck_assert_msg(c1len == m1_size + CRYPTO_MAC_SIZE, "could not encrypt data");
|
||||||
|
|
||||||
const uint16_t m1plen = decrypt_data_symmetric(mem, k, n, c1, c1len, m1prime);
|
const uint16_t m1plen = decrypt_data_symmetric(k, n, c1, c1len, m1prime);
|
||||||
|
|
||||||
ck_assert_msg(m1plen == m1_size, "decrypted text lengths differ");
|
ck_assert_msg(m1plen == m1_size, "decrypted text lengths differ");
|
||||||
ck_assert_msg(memcmp(m1prime, m1, m1_size) == 0, "decrypted texts differ");
|
ck_assert_msg(memcmp(m1prime, m1, m1_size) == 0, "decrypted texts differ");
|
||||||
@ -283,8 +271,6 @@ static void test_large_data_symmetric(void)
|
|||||||
|
|
||||||
static void test_very_large_data(void)
|
static void test_very_large_data(void)
|
||||||
{
|
{
|
||||||
const Memory *mem = os_memory();
|
|
||||||
ck_assert(mem != nullptr);
|
|
||||||
const Random *rng = os_random();
|
const Random *rng = os_random();
|
||||||
ck_assert(rng != nullptr);
|
ck_assert(rng != nullptr);
|
||||||
|
|
||||||
@ -301,9 +287,7 @@ static void test_very_large_data(void)
|
|||||||
ck_assert(plain != nullptr);
|
ck_assert(plain != nullptr);
|
||||||
ck_assert(encrypted != nullptr);
|
ck_assert(encrypted != nullptr);
|
||||||
|
|
||||||
memset(plain, 0, plain_size);
|
encrypt_data(pk, sk, nonce, plain, plain_size, encrypted);
|
||||||
|
|
||||||
encrypt_data(mem, pk, sk, nonce, plain, plain_size, encrypted);
|
|
||||||
|
|
||||||
free(encrypted);
|
free(encrypted);
|
||||||
free(plain);
|
free(plain);
|
||||||
|
121
external/toxcore/c-toxcore/auto_tests/netprof_test.c
vendored
Normal file
121
external/toxcore/c-toxcore/auto_tests/netprof_test.c
vendored
Normal file
@ -0,0 +1,121 @@
|
|||||||
|
/** Auto Tests: basic network profile functionality test (UDP only)
|
||||||
|
* TODO(JFreegman): test TCP packets as well
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include <inttypes.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
|
||||||
|
#include "../toxcore/tox_private.h"
|
||||||
|
#include "../toxcore/util.h"
|
||||||
|
|
||||||
|
#include "auto_test_support.h"
|
||||||
|
#include "check_compat.h"
|
||||||
|
|
||||||
|
#define NUM_TOXES 2
|
||||||
|
|
||||||
|
static void test_netprof(AutoTox *autotoxes)
|
||||||
|
{
|
||||||
|
// Send some messages to create fake traffic
|
||||||
|
for (size_t i = 0; i < 256; ++i) {
|
||||||
|
for (uint32_t j = 0; j < NUM_TOXES; ++j) {
|
||||||
|
tox_friend_send_message(autotoxes[j].tox, 0, TOX_MESSAGE_TYPE_NORMAL, (const uint8_t *)"test", 4, nullptr);
|
||||||
|
}
|
||||||
|
|
||||||
|
iterate_all_wait(autotoxes, NUM_TOXES, ITERATION_INTERVAL);
|
||||||
|
}
|
||||||
|
|
||||||
|
// idle traffic for a while
|
||||||
|
for (size_t i = 0; i < 100; ++i) {
|
||||||
|
iterate_all_wait(autotoxes, NUM_TOXES, ITERATION_INTERVAL);
|
||||||
|
}
|
||||||
|
|
||||||
|
const Tox *tox1 = autotoxes[0].tox;
|
||||||
|
|
||||||
|
const uint64_t UDP_count_sent1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
const uint64_t UDP_count_recv1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
const uint64_t TCP_count_sent1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
const uint64_t TCP_count_recv1 = tox_netprof_get_packet_total_count(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
|
||||||
|
const uint64_t UDP_bytes_sent1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
const uint64_t UDP_bytes_recv1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
const uint64_t TCP_bytes_sent1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
const uint64_t TCP_bytes_recv1 = tox_netprof_get_packet_total_bytes(tox1, TOX_NETPROF_PACKET_TYPE_TCP,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
|
||||||
|
ck_assert(UDP_count_recv1 > 0 && UDP_count_sent1 > 0);
|
||||||
|
ck_assert(UDP_bytes_recv1 > 0 && UDP_bytes_sent1 > 0);
|
||||||
|
|
||||||
|
(void)TCP_count_sent1;
|
||||||
|
(void)TCP_bytes_sent1;
|
||||||
|
(void)TCP_bytes_recv1;
|
||||||
|
(void)TCP_count_recv1;
|
||||||
|
|
||||||
|
uint64_t total_sent_count = 0;
|
||||||
|
uint64_t total_recv_count = 0;
|
||||||
|
uint64_t total_sent_bytes = 0;
|
||||||
|
uint64_t total_recv_bytes = 0;
|
||||||
|
|
||||||
|
// tox1 makes sure the sum value of all packet ID's is equal to the totals
|
||||||
|
for (size_t i = 0; i < 256; ++i) {
|
||||||
|
// this id isn't valid for UDP packets but we still want to call the
|
||||||
|
// functions and make sure they return some non-zero value
|
||||||
|
if (i == TOX_NETPROF_PACKET_ID_TCP_DATA) {
|
||||||
|
ck_assert(tox_netprof_get_packet_id_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT) > 0);
|
||||||
|
ck_assert(tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT) > 0);
|
||||||
|
ck_assert(tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT) > 0);
|
||||||
|
ck_assert(tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV) > 0);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
total_sent_count += tox_netprof_get_packet_id_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
total_recv_count += tox_netprof_get_packet_id_count(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
|
||||||
|
total_sent_bytes += tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
total_recv_bytes += tox_netprof_get_packet_id_bytes(tox1, TOX_NETPROF_PACKET_TYPE_UDP, i,
|
||||||
|
TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
}
|
||||||
|
|
||||||
|
const uint64_t total_packets = total_sent_count + total_recv_count;
|
||||||
|
ck_assert_msg(total_packets == UDP_count_sent1 + UDP_count_recv1,
|
||||||
|
"%" PRIu64 "does not match %" PRIu64 "\n", total_packets, UDP_count_sent1 + UDP_count_recv1);
|
||||||
|
|
||||||
|
ck_assert_msg(total_sent_count == UDP_count_sent1, "%" PRIu64 " does not match %" PRIu64 "\n", total_sent_count, UDP_count_sent1);
|
||||||
|
ck_assert_msg(total_recv_count == UDP_count_recv1, "%" PRIu64 " does not match %" PRIu64"\n", total_recv_count, UDP_count_recv1);
|
||||||
|
|
||||||
|
|
||||||
|
const uint64_t total_bytes = total_sent_bytes + total_recv_bytes;
|
||||||
|
ck_assert_msg(total_bytes == UDP_bytes_sent1 + UDP_bytes_recv1,
|
||||||
|
"%" PRIu64 "does not match %" PRIu64 "\n", total_bytes, UDP_bytes_sent1 + UDP_bytes_recv1);
|
||||||
|
|
||||||
|
ck_assert_msg(total_sent_bytes == UDP_bytes_sent1, "%" PRIu64 " does not match %" PRIu64 "\n", total_sent_bytes, UDP_bytes_sent1);
|
||||||
|
ck_assert_msg(total_recv_bytes == UDP_bytes_recv1, "%" PRIu64 " does not match %" PRIu64 "\n", total_recv_bytes, UDP_bytes_recv1);
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(void)
|
||||||
|
{
|
||||||
|
setvbuf(stdout, nullptr, _IONBF, 0);
|
||||||
|
|
||||||
|
Run_Auto_Options autotox_opts = default_run_auto_options();
|
||||||
|
autotox_opts.graph = GRAPH_COMPLETE;
|
||||||
|
|
||||||
|
run_auto_test(nullptr, NUM_TOXES, test_netprof, 0, &autotox_opts);
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
#undef NUM_TOXES
|
@ -109,7 +109,7 @@ static int handle_test_3(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
#if 0
|
#if 0
|
||||||
print_client_id(packet, length);
|
print_client_id(packet, length);
|
||||||
#endif
|
#endif
|
||||||
int len = decrypt_data(onion->mem, test_3_pub_key, dht_get_self_secret_key(onion->dht),
|
int len = decrypt_data(test_3_pub_key, dht_get_self_secret_key(onion->dht),
|
||||||
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH,
|
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH,
|
||||||
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH + CRYPTO_NONCE_SIZE,
|
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH + CRYPTO_NONCE_SIZE,
|
||||||
2 + CRYPTO_SHA256_SIZE + CRYPTO_MAC_SIZE, plain);
|
2 + CRYPTO_SHA256_SIZE + CRYPTO_MAC_SIZE, plain);
|
||||||
@ -144,7 +144,7 @@ static int handle_test_3_old(void *object, const IP_Port *source, const uint8_t
|
|||||||
#if 0
|
#if 0
|
||||||
print_client_id(packet, length);
|
print_client_id(packet, length);
|
||||||
#endif
|
#endif
|
||||||
int len = decrypt_data(onion->mem, test_3_pub_key, dht_get_self_secret_key(onion->dht),
|
int len = decrypt_data(test_3_pub_key, dht_get_self_secret_key(onion->dht),
|
||||||
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH,
|
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH,
|
||||||
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH + CRYPTO_NONCE_SIZE,
|
packet + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH + CRYPTO_NONCE_SIZE,
|
||||||
1 + CRYPTO_SHA256_SIZE + CRYPTO_MAC_SIZE, plain);
|
1 + CRYPTO_SHA256_SIZE + CRYPTO_MAC_SIZE, plain);
|
||||||
@ -182,7 +182,7 @@ static int handle_test_4(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int len = decrypt_data(onion->mem, packet + 1 + CRYPTO_NONCE_SIZE, dht_get_self_secret_key(onion->dht), packet + 1,
|
int len = decrypt_data(packet + 1 + CRYPTO_NONCE_SIZE, dht_get_self_secret_key(onion->dht), packet + 1,
|
||||||
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE, sizeof("Install gentoo") + CRYPTO_MAC_SIZE, plain);
|
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE, sizeof("Install gentoo") + CRYPTO_MAC_SIZE, plain);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
@ -202,10 +202,10 @@ static int handle_test_4(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
* Use Onion_Path path to send data of length to dest.
|
* Use Onion_Path path to send data of length to dest.
|
||||||
* Maximum length of data is ONION_MAX_DATA_SIZE.
|
* Maximum length of data is ONION_MAX_DATA_SIZE.
|
||||||
*/
|
*/
|
||||||
static void send_onion_packet(const Networking_Core *net, const Memory *mem, const Random *rng, const Onion_Path *path, const IP_Port *dest, const uint8_t *data, uint16_t length)
|
static void send_onion_packet(const Networking_Core *net, const Random *rng, const Onion_Path *path, const IP_Port *dest, const uint8_t *data, uint16_t length)
|
||||||
{
|
{
|
||||||
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
||||||
const int len = create_onion_packet(mem, rng, packet, sizeof(packet), path, dest, data, length);
|
const int len = create_onion_packet(rng, packet, sizeof(packet), path, dest, data, length);
|
||||||
ck_assert_msg(len != -1, "failed to create onion packet");
|
ck_assert_msg(len != -1, "failed to create onion packet");
|
||||||
ck_assert_msg(sendpacket(net, &path->ip_port1, packet, len) == len, "failed to send onion packet");
|
ck_assert_msg(sendpacket(net, &path->ip_port1, packet, len) == len, "failed to send onion packet");
|
||||||
}
|
}
|
||||||
@ -264,7 +264,7 @@ static void test_basic(void)
|
|||||||
nodes[3] = n2;
|
nodes[3] = n2;
|
||||||
Onion_Path path;
|
Onion_Path path;
|
||||||
create_onion_path(rng, onion1->dht, &path, nodes);
|
create_onion_path(rng, onion1->dht, &path, nodes);
|
||||||
send_onion_packet(onion1->net, onion1->mem, rng, &path, &nodes[3].ip_port, req_packet, sizeof(req_packet));
|
send_onion_packet(onion1->net, rng, &path, &nodes[3].ip_port, req_packet, sizeof(req_packet));
|
||||||
|
|
||||||
handled_test_1 = 0;
|
handled_test_1 = 0;
|
||||||
|
|
||||||
@ -291,7 +291,7 @@ static void test_basic(void)
|
|||||||
uint64_t s;
|
uint64_t s;
|
||||||
memcpy(&s, sb_data, sizeof(uint64_t));
|
memcpy(&s, sb_data, sizeof(uint64_t));
|
||||||
memcpy(test_3_pub_key, nodes[3].public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(test_3_pub_key, nodes[3].public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
int ret = send_announce_request(log1, onion1->mem, onion1->net, rng, &path, &nodes[3],
|
int ret = send_announce_request(log1, onion1->net, rng, &path, &nodes[3],
|
||||||
dht_get_self_public_key(onion1->dht),
|
dht_get_self_public_key(onion1->dht),
|
||||||
dht_get_self_secret_key(onion1->dht),
|
dht_get_self_secret_key(onion1->dht),
|
||||||
zeroes,
|
zeroes,
|
||||||
@ -313,7 +313,7 @@ static void test_basic(void)
|
|||||||
memcpy(onion_announce_entry_public_key(onion2_a, 1), dht_get_self_public_key(onion2->dht), CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(onion_announce_entry_public_key(onion2_a, 1), dht_get_self_public_key(onion2->dht), CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
onion_announce_entry_set_time(onion2_a, 1, mono_time_get(mono_time2));
|
onion_announce_entry_set_time(onion2_a, 1, mono_time_get(mono_time2));
|
||||||
networking_registerhandler(onion1->net, NET_PACKET_ONION_DATA_RESPONSE, &handle_test_4, onion1);
|
networking_registerhandler(onion1->net, NET_PACKET_ONION_DATA_RESPONSE, &handle_test_4, onion1);
|
||||||
send_announce_request(log1, onion1->mem, onion1->net, rng, &path, &nodes[3],
|
send_announce_request(log1, onion1->net, rng, &path, &nodes[3],
|
||||||
dht_get_self_public_key(onion1->dht),
|
dht_get_self_public_key(onion1->dht),
|
||||||
dht_get_self_secret_key(onion1->dht),
|
dht_get_self_secret_key(onion1->dht),
|
||||||
test_3_ping_id,
|
test_3_ping_id,
|
||||||
@ -338,7 +338,7 @@ static void test_basic(void)
|
|||||||
ck_assert_msg((onion3 != nullptr), "Onion failed initializing.");
|
ck_assert_msg((onion3 != nullptr), "Onion failed initializing.");
|
||||||
|
|
||||||
random_nonce(rng, nonce);
|
random_nonce(rng, nonce);
|
||||||
ret = send_data_request(log3, onion3->mem, onion3->net, rng, &path, &nodes[3].ip_port,
|
ret = send_data_request(log3, onion3->net, rng, &path, &nodes[3].ip_port,
|
||||||
dht_get_self_public_key(onion1->dht),
|
dht_get_self_public_key(onion1->dht),
|
||||||
dht_get_self_public_key(onion1->dht),
|
dht_get_self_public_key(onion1->dht),
|
||||||
nonce, (const uint8_t *)"Install gentoo", sizeof("Install gentoo"));
|
nonce, (const uint8_t *)"Install gentoo", sizeof("Install gentoo"));
|
||||||
|
5
external/toxcore/c-toxcore/other/BUILD.bazel
vendored
5
external/toxcore/c-toxcore/other/BUILD.bazel
vendored
@ -21,15 +21,10 @@ cc_binary(
|
|||||||
"//c-toxcore/toxcore:Messenger",
|
"//c-toxcore/toxcore:Messenger",
|
||||||
"//c-toxcore/toxcore:TCP_server",
|
"//c-toxcore/toxcore:TCP_server",
|
||||||
"//c-toxcore/toxcore:ccompat",
|
"//c-toxcore/toxcore:ccompat",
|
||||||
"//c-toxcore/toxcore:crypto_core",
|
|
||||||
"//c-toxcore/toxcore:forwarding",
|
|
||||||
"//c-toxcore/toxcore:group_announce",
|
|
||||||
"//c-toxcore/toxcore:group_onion_announce",
|
"//c-toxcore/toxcore:group_onion_announce",
|
||||||
"//c-toxcore/toxcore:logger",
|
"//c-toxcore/toxcore:logger",
|
||||||
"//c-toxcore/toxcore:mem",
|
|
||||||
"//c-toxcore/toxcore:mono_time",
|
"//c-toxcore/toxcore:mono_time",
|
||||||
"//c-toxcore/toxcore:network",
|
"//c-toxcore/toxcore:network",
|
||||||
"//c-toxcore/toxcore:onion",
|
|
||||||
"//c-toxcore/toxcore:onion_announce",
|
"//c-toxcore/toxcore:onion_announce",
|
||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
],
|
],
|
||||||
|
@ -15,7 +15,7 @@ CPPFLAGS+=("-Itoxav")
|
|||||||
CPPFLAGS+=("-Itoxencryptsave")
|
CPPFLAGS+=("-Itoxencryptsave")
|
||||||
CPPFLAGS+=("-Ithird_party/cmp")
|
CPPFLAGS+=("-Ithird_party/cmp")
|
||||||
|
|
||||||
LDFLAGS=("-lopus" "-lsodium" "-lvpx" "-lpthread" "-lconfig" "-lgmock" "-lgtest" "-lbenchmark")
|
LDFLAGS=("-lopus" "-lsodium" "-lvpx" "-lpthread" "-lconfig" "-lgmock" "-lgtest")
|
||||||
LDFLAGS+=("-fuse-ld=gold")
|
LDFLAGS+=("-fuse-ld=gold")
|
||||||
LDFLAGS+=("-Wl,--detect-odr-violations")
|
LDFLAGS+=("-Wl,--detect-odr-violations")
|
||||||
LDFLAGS+=("-Wl,--warn-common")
|
LDFLAGS+=("-Wl,--warn-common")
|
||||||
@ -27,7 +27,7 @@ put() {
|
|||||||
if [ "$SKIP_LINES" = "" ]; then
|
if [ "$SKIP_LINES" = "" ]; then
|
||||||
echo "#line 1 \"$1\"" >>amalgamation.cc
|
echo "#line 1 \"$1\"" >>amalgamation.cc
|
||||||
fi
|
fi
|
||||||
grep -v '^BENCHMARK_MAIN' "$1" >>amalgamation.cc
|
cat "$1" >>amalgamation.cc
|
||||||
}
|
}
|
||||||
|
|
||||||
putmain() {
|
putmain() {
|
||||||
|
@ -6,24 +6,16 @@ cc_binary(
|
|||||||
"src/*.c",
|
"src/*.c",
|
||||||
"src/*.h",
|
"src/*.h",
|
||||||
]),
|
]),
|
||||||
tags = ["no-windows"],
|
|
||||||
deps = [
|
deps = [
|
||||||
"//c-toxcore/other:bootstrap_node_packets",
|
"//c-toxcore/other:bootstrap_node_packets",
|
||||||
"//c-toxcore/toxcore:DHT",
|
"//c-toxcore/toxcore:DHT",
|
||||||
"//c-toxcore/toxcore:LAN_discovery",
|
"//c-toxcore/toxcore:LAN_discovery",
|
||||||
"//c-toxcore/toxcore:TCP_server",
|
"//c-toxcore/toxcore:TCP_server",
|
||||||
"//c-toxcore/toxcore:announce",
|
"//c-toxcore/toxcore:announce",
|
||||||
"//c-toxcore/toxcore:attributes",
|
|
||||||
"//c-toxcore/toxcore:ccompat",
|
"//c-toxcore/toxcore:ccompat",
|
||||||
"//c-toxcore/toxcore:crypto_core",
|
|
||||||
"//c-toxcore/toxcore:forwarding",
|
|
||||||
"//c-toxcore/toxcore:group_announce",
|
|
||||||
"//c-toxcore/toxcore:group_onion_announce",
|
"//c-toxcore/toxcore:group_onion_announce",
|
||||||
"//c-toxcore/toxcore:logger",
|
"//c-toxcore/toxcore:logger",
|
||||||
"//c-toxcore/toxcore:mem",
|
|
||||||
"//c-toxcore/toxcore:mono_time",
|
"//c-toxcore/toxcore:mono_time",
|
||||||
"//c-toxcore/toxcore:network",
|
|
||||||
"//c-toxcore/toxcore:onion",
|
|
||||||
"//c-toxcore/toxcore:onion_announce",
|
"//c-toxcore/toxcore:onion_announce",
|
||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
"@libconfig",
|
"@libconfig",
|
||||||
|
@ -1,18 +1,18 @@
|
|||||||
################################################
|
################################################
|
||||||
# cmake-asan
|
# cmake-asan
|
||||||
FROM ubuntu:24.04
|
FROM ubuntu:20.04
|
||||||
|
|
||||||
RUN apt-get update && \
|
RUN apt-get update && \
|
||||||
DEBIAN_FRONTEND="noninteractive" apt-get install -y --no-install-recommends \
|
DEBIAN_FRONTEND="noninteractive" apt-get install -y --no-install-recommends \
|
||||||
clang \
|
clang \
|
||||||
cmake \
|
cmake \
|
||||||
libclang-rt-dev \
|
|
||||||
libconfig-dev \
|
libconfig-dev \
|
||||||
libgmock-dev \
|
libgmock-dev \
|
||||||
libgtest-dev \
|
libgtest-dev \
|
||||||
libopus-dev \
|
libopus-dev \
|
||||||
libsodium-dev \
|
libsodium-dev \
|
||||||
libvpx-dev \
|
libvpx-dev \
|
||||||
|
llvm-dev \
|
||||||
ninja-build \
|
ninja-build \
|
||||||
pkg-config \
|
pkg-config \
|
||||||
&& apt-get clean \
|
&& apt-get clean \
|
||||||
@ -22,8 +22,8 @@ COPY entrypoint.sh /
|
|||||||
RUN ["chmod", "755", "/entrypoint.sh"]
|
RUN ["chmod", "755", "/entrypoint.sh"]
|
||||||
|
|
||||||
WORKDIR /home/builder
|
WORKDIR /home/builder
|
||||||
RUN groupadd -r -g 987 builder \
|
RUN groupadd -r -g 1000 builder \
|
||||||
&& useradd --no-log-init -r -g builder -u 987 builder \
|
&& useradd --no-log-init -r -g builder -u 1000 builder \
|
||||||
&& chown builder:builder /home/builder
|
&& chown builder:builder /home/builder
|
||||||
USER builder
|
USER builder
|
||||||
|
|
||||||
|
@ -1,9 +1,9 @@
|
|||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
|
|
||||||
set -eux
|
set -eu
|
||||||
|
|
||||||
SANITIZER="${1:-asan}"
|
SANITIZER="${1:-asan}"
|
||||||
|
|
||||||
cp -a /c-toxcore .
|
cp -a /c-toxcore .
|
||||||
cd c-toxcore
|
cd c-toxcore
|
||||||
.circleci/cmake-"$SANITIZER" || (cat /home/builder/c-toxcore/_build/CMakeFiles/CMakeError.log && false)
|
.circleci/cmake-"$SANITIZER"
|
||||||
|
@ -1,14 +1,6 @@
|
|||||||
#!/bin/sh
|
#!/bin/sh
|
||||||
|
|
||||||
set -eux
|
|
||||||
|
|
||||||
SANITIZER="${1:-asan}"
|
SANITIZER="${1:-asan}"
|
||||||
|
|
||||||
if [ -t 0 ]; then
|
|
||||||
TTY=true
|
|
||||||
else
|
|
||||||
TTY=false
|
|
||||||
fi
|
|
||||||
|
|
||||||
docker build -t toxchat/c-toxcore:circleci other/docker/circleci
|
docker build -t toxchat/c-toxcore:circleci other/docker/circleci
|
||||||
docker run --name toxcore-circleci --rm --interactive="$TTY" --tty="$TTY" --volume "$PWD:/c-toxcore" toxchat/c-toxcore:circleci "$SANITIZER"
|
docker run --name toxcore-circleci --rm -it -v "$PWD:/c-toxcore" toxchat/c-toxcore:circleci "$SANITIZER"
|
||||||
|
@ -9,7 +9,6 @@ cc_binary(
|
|||||||
"main/tox_main.h",
|
"main/tox_main.h",
|
||||||
],
|
],
|
||||||
deps = [
|
deps = [
|
||||||
"//c-toxcore/toxcore:ccompat",
|
|
||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
"//c-toxcore/toxcore:tox_events",
|
"//c-toxcore/toxcore:tox_events",
|
||||||
],
|
],
|
||||||
|
@ -50,10 +50,6 @@ module "//c-toxcore/third_party:cmp" {
|
|||||||
module "//c-toxcore/toxencryptsave:defines" {
|
module "//c-toxcore/toxencryptsave:defines" {
|
||||||
header "toxencryptsave/defines.h"
|
header "toxencryptsave/defines.h"
|
||||||
}
|
}
|
||||||
module "@benchmark" {
|
|
||||||
textual header "/usr/include/benchmark/benchmark.h"
|
|
||||||
use std
|
|
||||||
}
|
|
||||||
module "@com_google_googletest//:gtest" {
|
module "@com_google_googletest//:gtest" {
|
||||||
textual header "/usr/include/gmock/gmock.h"
|
textual header "/usr/include/gmock/gmock.h"
|
||||||
textual header "/usr/include/gtest/gtest.h"
|
textual header "/usr/include/gtest/gtest.h"
|
||||||
@ -114,7 +110,7 @@ class Context:
|
|||||||
hdrs,
|
hdrs,
|
||||||
}
|
}
|
||||||
|
|
||||||
def bzl_cc_binary(
|
def bzl_cc_test(
|
||||||
self,
|
self,
|
||||||
name: str,
|
name: str,
|
||||||
srcs: Iterable[str] = tuple(),
|
srcs: Iterable[str] = tuple(),
|
||||||
@ -165,8 +161,7 @@ def main() -> None:
|
|||||||
"load": ctx.bzl_load,
|
"load": ctx.bzl_load,
|
||||||
"exports_files": ctx.bzl_exports_files,
|
"exports_files": ctx.bzl_exports_files,
|
||||||
"cc_library": ctx.bzl_cc_library,
|
"cc_library": ctx.bzl_cc_library,
|
||||||
"cc_binary": ctx.bzl_cc_binary,
|
"cc_test": ctx.bzl_cc_test,
|
||||||
"cc_test": ctx.bzl_cc_binary,
|
|
||||||
"cc_fuzz_test": ctx.bzl_cc_fuzz_test,
|
"cc_fuzz_test": ctx.bzl_cc_fuzz_test,
|
||||||
"select": ctx.bzl_select,
|
"select": ctx.bzl_select,
|
||||||
"glob": ctx.bzl_glob,
|
"glob": ctx.bzl_glob,
|
||||||
|
@ -3,7 +3,6 @@ FROM alpine:3.19.0
|
|||||||
|
|
||||||
RUN ["apk", "add", "--no-cache", \
|
RUN ["apk", "add", "--no-cache", \
|
||||||
"bash", \
|
"bash", \
|
||||||
"benchmark-dev", \
|
|
||||||
"clang", \
|
"clang", \
|
||||||
"gtest-dev", \
|
"gtest-dev", \
|
||||||
"libconfig-dev", \
|
"libconfig-dev", \
|
||||||
|
@ -4,5 +4,4 @@ cc_binary(
|
|||||||
name = "grencez_tok5",
|
name = "grencez_tok5",
|
||||||
srcs = ["grencez_tok5.c"],
|
srcs = ["grencez_tok5.c"],
|
||||||
copts = ["-Wno-unused-result"],
|
copts = ["-Wno-unused-result"],
|
||||||
tags = ["no-windows"],
|
|
||||||
)
|
)
|
||||||
|
@ -48,7 +48,6 @@ cc_fuzz_test(
|
|||||||
deps = [
|
deps = [
|
||||||
":fuzz_support",
|
":fuzz_support",
|
||||||
":fuzz_tox",
|
":fuzz_tox",
|
||||||
"//c-toxcore/toxcore:crypto_core",
|
|
||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
"//c-toxcore/toxcore:tox_dispatch",
|
"//c-toxcore/toxcore:tox_dispatch",
|
||||||
"//c-toxcore/toxcore:tox_events",
|
"//c-toxcore/toxcore:tox_events",
|
||||||
@ -103,7 +102,6 @@ cc_test(
|
|||||||
deps = [
|
deps = [
|
||||||
":fuzz_support",
|
":fuzz_support",
|
||||||
":fuzz_tox",
|
":fuzz_tox",
|
||||||
"//c-toxcore/toxcore:crypto_core",
|
|
||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
"//c-toxcore/toxcore:tox_dispatch",
|
"//c-toxcore/toxcore:tox_dispatch",
|
||||||
"//c-toxcore/toxcore:tox_events",
|
"//c-toxcore/toxcore:tox_events",
|
||||||
@ -119,7 +117,6 @@ cc_fuzz_test(
|
|||||||
deps = [
|
deps = [
|
||||||
":fuzz_support",
|
":fuzz_support",
|
||||||
":fuzz_tox",
|
":fuzz_tox",
|
||||||
"//c-toxcore/toxcore:crypto_core",
|
|
||||||
"//c-toxcore/toxcore:tox",
|
"//c-toxcore/toxcore:tox",
|
||||||
"//c-toxcore/toxcore:tox_dispatch",
|
"//c-toxcore/toxcore:tox_dispatch",
|
||||||
"//c-toxcore/toxcore:tox_events",
|
"//c-toxcore/toxcore:tox_events",
|
||||||
|
@ -4,15 +4,8 @@
|
|||||||
|
|
||||||
#include "fuzz_support.hh"
|
#include "fuzz_support.hh"
|
||||||
|
|
||||||
#ifdef _WIN32
|
|
||||||
#include <winsock2.h>
|
|
||||||
// Comment line here to avoid reordering by source code formatters.
|
|
||||||
#include <windows.h>
|
|
||||||
#include <ws2tcpip.h>
|
|
||||||
#else
|
|
||||||
#include <arpa/inet.h>
|
#include <arpa/inet.h>
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#endif
|
|
||||||
|
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
#include <cassert>
|
#include <cassert>
|
||||||
|
92
external/toxcore/c-toxcore/toxcore/BUILD.bazel
vendored
92
external/toxcore/c-toxcore/toxcore/BUILD.bazel
vendored
@ -1,4 +1,4 @@
|
|||||||
load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_library", "cc_test")
|
load("@rules_cc//cc:defs.bzl", "cc_library", "cc_test")
|
||||||
load("@rules_fuzzing//fuzzing:cc_defs.bzl", "cc_fuzz_test")
|
load("@rules_fuzzing//fuzzing:cc_defs.bzl", "cc_fuzz_test")
|
||||||
|
|
||||||
exports_files(
|
exports_files(
|
||||||
@ -100,58 +100,14 @@ cc_test(
|
|||||||
size = "small",
|
size = "small",
|
||||||
srcs = ["util_test.cc"],
|
srcs = ["util_test.cc"],
|
||||||
deps = [
|
deps = [
|
||||||
|
":crypto_core",
|
||||||
|
":crypto_core_test_util",
|
||||||
":util",
|
":util",
|
||||||
"@com_google_googletest//:gtest",
|
"@com_google_googletest//:gtest",
|
||||||
"@com_google_googletest//:gtest_main",
|
"@com_google_googletest//:gtest_main",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
cc_library(
|
|
||||||
name = "sort",
|
|
||||||
srcs = ["sort.c"],
|
|
||||||
hdrs = ["sort.h"],
|
|
||||||
deps = [
|
|
||||||
":attributes",
|
|
||||||
":ccompat",
|
|
||||||
":util",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
cc_library(
|
|
||||||
name = "sort_test_util",
|
|
||||||
testonly = True,
|
|
||||||
srcs = ["sort_test_util.cc"],
|
|
||||||
hdrs = ["sort_test_util.hh"],
|
|
||||||
deps = [
|
|
||||||
":sort",
|
|
||||||
":util",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
cc_test(
|
|
||||||
name = "sort_test",
|
|
||||||
size = "small",
|
|
||||||
srcs = ["sort_test.cc"],
|
|
||||||
deps = [
|
|
||||||
":sort",
|
|
||||||
":sort_test_util",
|
|
||||||
"@com_google_googletest//:gtest",
|
|
||||||
"@com_google_googletest//:gtest_main",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
cc_binary(
|
|
||||||
name = "sort_bench",
|
|
||||||
testonly = True,
|
|
||||||
srcs = ["sort_bench.cc"],
|
|
||||||
deps = [
|
|
||||||
":mem",
|
|
||||||
":sort",
|
|
||||||
":sort_test_util",
|
|
||||||
"@benchmark",
|
|
||||||
],
|
|
||||||
)
|
|
||||||
|
|
||||||
cc_library(
|
cc_library(
|
||||||
name = "logger",
|
name = "logger",
|
||||||
srcs = ["logger.c"],
|
srcs = ["logger.c"],
|
||||||
@ -215,7 +171,6 @@ cc_library(
|
|||||||
deps = [
|
deps = [
|
||||||
":attributes",
|
":attributes",
|
||||||
":ccompat",
|
":ccompat",
|
||||||
":mem",
|
|
||||||
":util",
|
":util",
|
||||||
"@libsodium",
|
"@libsodium",
|
||||||
],
|
],
|
||||||
@ -254,7 +209,6 @@ cc_test(
|
|||||||
deps = [
|
deps = [
|
||||||
":crypto_core",
|
":crypto_core",
|
||||||
":crypto_core_test_util",
|
":crypto_core_test_util",
|
||||||
":mem_test_util",
|
|
||||||
":util",
|
":util",
|
||||||
"@com_google_googletest//:gtest",
|
"@com_google_googletest//:gtest",
|
||||||
"@com_google_googletest//:gtest_main",
|
"@com_google_googletest//:gtest_main",
|
||||||
@ -346,6 +300,18 @@ cc_library(
|
|||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
cc_library(
|
||||||
|
name = "net_profile",
|
||||||
|
srcs = ["net_profile.c"],
|
||||||
|
hdrs = ["net_profile.h"],
|
||||||
|
deps = [
|
||||||
|
":attributes",
|
||||||
|
":ccompat",
|
||||||
|
":logger",
|
||||||
|
":mem",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
cc_library(
|
cc_library(
|
||||||
name = "network",
|
name = "network",
|
||||||
srcs = ["network.c"],
|
srcs = ["network.c"],
|
||||||
@ -365,6 +331,7 @@ cc_library(
|
|||||||
":logger",
|
":logger",
|
||||||
":mem",
|
":mem",
|
||||||
":mono_time",
|
":mono_time",
|
||||||
|
":net_profile",
|
||||||
":util",
|
":util",
|
||||||
"@libsodium",
|
"@libsodium",
|
||||||
"@psocket",
|
"@psocket",
|
||||||
@ -483,7 +450,6 @@ cc_library(
|
|||||||
":network",
|
":network",
|
||||||
":ping_array",
|
":ping_array",
|
||||||
":shared_key_cache",
|
":shared_key_cache",
|
||||||
":sort",
|
|
||||||
":state",
|
":state",
|
||||||
":util",
|
":util",
|
||||||
],
|
],
|
||||||
@ -527,12 +493,10 @@ cc_test(
|
|||||||
cc_fuzz_test(
|
cc_fuzz_test(
|
||||||
name = "DHT_fuzz_test",
|
name = "DHT_fuzz_test",
|
||||||
size = "small",
|
size = "small",
|
||||||
testonly = True,
|
|
||||||
srcs = ["DHT_fuzz_test.cc"],
|
srcs = ["DHT_fuzz_test.cc"],
|
||||||
corpus = ["//tools/toktok-fuzzer/corpus:DHT_fuzz_test"],
|
corpus = ["//tools/toktok-fuzzer/corpus:DHT_fuzz_test"],
|
||||||
deps = [
|
deps = [
|
||||||
":DHT",
|
":DHT",
|
||||||
":mem_test_util",
|
|
||||||
"//c-toxcore/testing/fuzzing:fuzz_support",
|
"//c-toxcore/testing/fuzzing:fuzz_support",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
@ -541,11 +505,7 @@ cc_library(
|
|||||||
name = "onion",
|
name = "onion",
|
||||||
srcs = ["onion.c"],
|
srcs = ["onion.c"],
|
||||||
hdrs = ["onion.h"],
|
hdrs = ["onion.h"],
|
||||||
visibility = [
|
visibility = ["//c-toxcore/auto_tests:__pkg__"],
|
||||||
"//c-toxcore/auto_tests:__pkg__",
|
|
||||||
"//c-toxcore/other:__pkg__",
|
|
||||||
"//c-toxcore/other/bootstrap_daemon:__pkg__",
|
|
||||||
],
|
|
||||||
deps = [
|
deps = [
|
||||||
":DHT",
|
":DHT",
|
||||||
":attributes",
|
":attributes",
|
||||||
@ -564,11 +524,7 @@ cc_library(
|
|||||||
name = "forwarding",
|
name = "forwarding",
|
||||||
srcs = ["forwarding.c"],
|
srcs = ["forwarding.c"],
|
||||||
hdrs = ["forwarding.h"],
|
hdrs = ["forwarding.h"],
|
||||||
visibility = [
|
visibility = ["//c-toxcore/auto_tests:__pkg__"],
|
||||||
"//c-toxcore/auto_tests:__pkg__",
|
|
||||||
"//c-toxcore/other:__pkg__",
|
|
||||||
"//c-toxcore/other/bootstrap_daemon:__pkg__",
|
|
||||||
],
|
|
||||||
deps = [
|
deps = [
|
||||||
":DHT",
|
":DHT",
|
||||||
":attributes",
|
":attributes",
|
||||||
@ -629,6 +585,7 @@ cc_library(
|
|||||||
":crypto_core",
|
":crypto_core",
|
||||||
":logger",
|
":logger",
|
||||||
":mem",
|
":mem",
|
||||||
|
":net_profile",
|
||||||
":network",
|
":network",
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
@ -656,6 +613,7 @@ cc_library(
|
|||||||
":logger",
|
":logger",
|
||||||
":mem",
|
":mem",
|
||||||
":mono_time",
|
":mono_time",
|
||||||
|
":net_profile",
|
||||||
":network",
|
":network",
|
||||||
":onion",
|
":onion",
|
||||||
":util",
|
":util",
|
||||||
@ -677,6 +635,7 @@ cc_library(
|
|||||||
":logger",
|
":logger",
|
||||||
":mem",
|
":mem",
|
||||||
":mono_time",
|
":mono_time",
|
||||||
|
":net_profile",
|
||||||
":network",
|
":network",
|
||||||
":util",
|
":util",
|
||||||
],
|
],
|
||||||
@ -699,6 +658,7 @@ cc_library(
|
|||||||
":logger",
|
":logger",
|
||||||
":mem",
|
":mem",
|
||||||
":mono_time",
|
":mono_time",
|
||||||
|
":net_profile",
|
||||||
":network",
|
":network",
|
||||||
":onion",
|
":onion",
|
||||||
":util",
|
":util",
|
||||||
@ -736,6 +696,7 @@ cc_library(
|
|||||||
":logger",
|
":logger",
|
||||||
":mem",
|
":mem",
|
||||||
":mono_time",
|
":mono_time",
|
||||||
|
":net_profile",
|
||||||
":network",
|
":network",
|
||||||
":util",
|
":util",
|
||||||
"@pthread",
|
"@pthread",
|
||||||
@ -763,7 +724,6 @@ cc_library(
|
|||||||
":network",
|
":network",
|
||||||
":onion",
|
":onion",
|
||||||
":shared_key_cache",
|
":shared_key_cache",
|
||||||
":sort",
|
|
||||||
":timed_auth",
|
":timed_auth",
|
||||||
":util",
|
":util",
|
||||||
],
|
],
|
||||||
@ -837,7 +797,6 @@ cc_library(
|
|||||||
":crypto_core",
|
":crypto_core",
|
||||||
":group_announce",
|
":group_announce",
|
||||||
":logger",
|
":logger",
|
||||||
":mem",
|
|
||||||
":mono_time",
|
":mono_time",
|
||||||
":network",
|
":network",
|
||||||
":onion_announce",
|
":onion_announce",
|
||||||
@ -867,7 +826,6 @@ cc_library(
|
|||||||
":onion",
|
":onion",
|
||||||
":onion_announce",
|
":onion_announce",
|
||||||
":ping_array",
|
":ping_array",
|
||||||
":sort",
|
|
||||||
":timed_auth",
|
":timed_auth",
|
||||||
":util",
|
":util",
|
||||||
],
|
],
|
||||||
@ -1033,11 +991,9 @@ cc_library(
|
|||||||
":crypto_core",
|
":crypto_core",
|
||||||
":friend_connection",
|
":friend_connection",
|
||||||
":logger",
|
":logger",
|
||||||
":mem",
|
|
||||||
":mono_time",
|
":mono_time",
|
||||||
":net_crypto",
|
":net_crypto",
|
||||||
":network",
|
":network",
|
||||||
":sort",
|
|
||||||
":state",
|
":state",
|
||||||
":util",
|
":util",
|
||||||
],
|
],
|
||||||
@ -1060,6 +1016,7 @@ cc_library(
|
|||||||
":DHT",
|
":DHT",
|
||||||
":Messenger",
|
":Messenger",
|
||||||
":TCP_client",
|
":TCP_client",
|
||||||
|
":TCP_server",
|
||||||
":attributes",
|
":attributes",
|
||||||
":ccompat",
|
":ccompat",
|
||||||
":crypto_core",
|
":crypto_core",
|
||||||
@ -1070,6 +1027,7 @@ cc_library(
|
|||||||
":mem",
|
":mem",
|
||||||
":mono_time",
|
":mono_time",
|
||||||
":net_crypto",
|
":net_crypto",
|
||||||
|
":net_profile",
|
||||||
":network",
|
":network",
|
||||||
":onion_client",
|
":onion_client",
|
||||||
":state",
|
":state",
|
||||||
|
188
external/toxcore/c-toxcore/toxcore/DHT.c
vendored
188
external/toxcore/c-toxcore/toxcore/DHT.c
vendored
@ -9,6 +9,7 @@
|
|||||||
#include "DHT.h"
|
#include "DHT.h"
|
||||||
|
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#include "LAN_discovery.h"
|
#include "LAN_discovery.h"
|
||||||
@ -23,9 +24,7 @@
|
|||||||
#include "ping.h"
|
#include "ping.h"
|
||||||
#include "ping_array.h"
|
#include "ping_array.h"
|
||||||
#include "shared_key_cache.h"
|
#include "shared_key_cache.h"
|
||||||
#include "sort.h"
|
|
||||||
#include "state.h"
|
#include "state.h"
|
||||||
#include "util.h"
|
|
||||||
|
|
||||||
/** The timeout after which a node is discarded completely. */
|
/** The timeout after which a node is discarded completely. */
|
||||||
#define KILL_NODE_TIMEOUT (BAD_NODE_TIMEOUT + PING_INTERVAL)
|
#define KILL_NODE_TIMEOUT (BAD_NODE_TIMEOUT + PING_INTERVAL)
|
||||||
@ -280,7 +279,7 @@ const uint8_t *dht_get_shared_key_sent(DHT *dht, const uint8_t *public_key)
|
|||||||
|
|
||||||
#define CRYPTO_SIZE (1 + CRYPTO_PUBLIC_KEY_SIZE * 2 + CRYPTO_NONCE_SIZE)
|
#define CRYPTO_SIZE (1 + CRYPTO_PUBLIC_KEY_SIZE * 2 + CRYPTO_NONCE_SIZE)
|
||||||
|
|
||||||
int create_request(const Memory *mem, const Random *rng, const uint8_t *send_public_key, const uint8_t *send_secret_key,
|
int create_request(const Random *rng, const uint8_t *send_public_key, const uint8_t *send_secret_key,
|
||||||
uint8_t *packet, const uint8_t *recv_public_key,
|
uint8_t *packet, const uint8_t *recv_public_key,
|
||||||
const uint8_t *data, uint32_t data_length, uint8_t request_id)
|
const uint8_t *data, uint32_t data_length, uint8_t request_id)
|
||||||
{
|
{
|
||||||
@ -297,7 +296,7 @@ int create_request(const Memory *mem, const Random *rng, const uint8_t *send_pub
|
|||||||
uint8_t temp[MAX_CRYPTO_REQUEST_SIZE] = {0};
|
uint8_t temp[MAX_CRYPTO_REQUEST_SIZE] = {0};
|
||||||
temp[0] = request_id;
|
temp[0] = request_id;
|
||||||
memcpy(temp + 1, data, data_length);
|
memcpy(temp + 1, data, data_length);
|
||||||
const int len = encrypt_data(mem, recv_public_key, send_secret_key, nonce, temp, data_length + 1,
|
const int len = encrypt_data(recv_public_key, send_secret_key, nonce, temp, data_length + 1,
|
||||||
packet + CRYPTO_SIZE);
|
packet + CRYPTO_SIZE);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
@ -313,7 +312,7 @@ int create_request(const Memory *mem, const Random *rng, const uint8_t *send_pub
|
|||||||
return len + CRYPTO_SIZE;
|
return len + CRYPTO_SIZE;
|
||||||
}
|
}
|
||||||
|
|
||||||
int handle_request(const Memory *mem, const uint8_t *self_public_key, const uint8_t *self_secret_key, uint8_t *public_key, uint8_t *data,
|
int handle_request(const uint8_t *self_public_key, const uint8_t *self_secret_key, uint8_t *public_key, uint8_t *data,
|
||||||
uint8_t *request_id, const uint8_t *packet, uint16_t packet_length)
|
uint8_t *request_id, const uint8_t *packet, uint16_t packet_length)
|
||||||
{
|
{
|
||||||
if (self_public_key == nullptr || public_key == nullptr || data == nullptr || request_id == nullptr
|
if (self_public_key == nullptr || public_key == nullptr || data == nullptr || request_id == nullptr
|
||||||
@ -332,7 +331,7 @@ int handle_request(const Memory *mem, const uint8_t *self_public_key, const uint
|
|||||||
memcpy(public_key, packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(public_key, packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
const uint8_t *const nonce = packet + 1 + CRYPTO_PUBLIC_KEY_SIZE * 2;
|
const uint8_t *const nonce = packet + 1 + CRYPTO_PUBLIC_KEY_SIZE * 2;
|
||||||
uint8_t temp[MAX_CRYPTO_REQUEST_SIZE];
|
uint8_t temp[MAX_CRYPTO_REQUEST_SIZE];
|
||||||
int32_t len1 = decrypt_data(mem, public_key, self_secret_key, nonce,
|
int32_t len1 = decrypt_data(public_key, self_secret_key, nonce,
|
||||||
packet + CRYPTO_SIZE, packet_length - CRYPTO_SIZE, temp);
|
packet + CRYPTO_SIZE, packet_length - CRYPTO_SIZE, temp);
|
||||||
|
|
||||||
if (len1 == -1 || len1 == 0) {
|
if (len1 == -1 || len1 == 0) {
|
||||||
@ -379,7 +378,7 @@ int dht_create_packet(const Memory *mem, const Random *rng,
|
|||||||
|
|
||||||
random_nonce(rng, nonce);
|
random_nonce(rng, nonce);
|
||||||
|
|
||||||
const int encrypted_length = encrypt_data_symmetric(mem, shared_key, nonce, plain, plain_length, encrypted);
|
const int encrypted_length = encrypt_data_symmetric(shared_key, nonce, plain, plain_length, encrypted);
|
||||||
|
|
||||||
if (encrypted_length < 0) {
|
if (encrypted_length < 0) {
|
||||||
mem_delete(mem, encrypted);
|
mem_delete(mem, encrypted);
|
||||||
@ -756,6 +755,49 @@ int get_close_nodes(
|
|||||||
is_lan, want_announce);
|
is_lan, want_announce);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
typedef struct DHT_Cmp_Data {
|
||||||
|
uint64_t cur_time;
|
||||||
|
const uint8_t *base_public_key;
|
||||||
|
Client_data entry;
|
||||||
|
} DHT_Cmp_Data;
|
||||||
|
|
||||||
|
non_null()
|
||||||
|
static int dht_cmp_entry(const void *a, const void *b)
|
||||||
|
{
|
||||||
|
const DHT_Cmp_Data *cmp1 = (const DHT_Cmp_Data *)a;
|
||||||
|
const DHT_Cmp_Data *cmp2 = (const DHT_Cmp_Data *)b;
|
||||||
|
const Client_data entry1 = cmp1->entry;
|
||||||
|
const Client_data entry2 = cmp2->entry;
|
||||||
|
const uint8_t *cmp_public_key = cmp1->base_public_key;
|
||||||
|
|
||||||
|
const bool t1 = assoc_timeout(cmp1->cur_time, &entry1.assoc4) && assoc_timeout(cmp1->cur_time, &entry1.assoc6);
|
||||||
|
const bool t2 = assoc_timeout(cmp2->cur_time, &entry2.assoc4) && assoc_timeout(cmp2->cur_time, &entry2.assoc6);
|
||||||
|
|
||||||
|
if (t1 && t2) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (t1) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (t2) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
const int closest = id_closest(cmp_public_key, entry1.public_key, entry2.public_key);
|
||||||
|
|
||||||
|
if (closest == 1) {
|
||||||
|
return 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (closest == 2) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
#ifdef CHECK_ANNOUNCE_NODE
|
#ifdef CHECK_ANNOUNCE_NODE
|
||||||
non_null()
|
non_null()
|
||||||
static void set_announce_node_in_list(Client_data *list, uint32_t list_len, const uint8_t *public_key)
|
static void set_announce_node_in_list(Client_data *list, uint32_t list_len, const uint8_t *public_key)
|
||||||
@ -828,7 +870,7 @@ static int handle_data_search_response(void *object, const IP_Port *source,
|
|||||||
const uint8_t *public_key = packet + 1;
|
const uint8_t *public_key = packet + 1;
|
||||||
const uint8_t *shared_key = dht_get_shared_key_recv(dht, public_key);
|
const uint8_t *shared_key = dht_get_shared_key_recv(dht, public_key);
|
||||||
|
|
||||||
if (decrypt_data_symmetric(dht->mem, shared_key,
|
if (decrypt_data_symmetric(shared_key,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
plain_len + CRYPTO_MAC_SIZE,
|
plain_len + CRYPTO_MAC_SIZE,
|
||||||
@ -872,117 +914,31 @@ static bool store_node_ok(const Client_data *client, uint64_t cur_time, const ui
|
|||||||
|| id_closest(comp_public_key, client->public_key, public_key) == 2;
|
|| id_closest(comp_public_key, client->public_key, public_key) == 2;
|
||||||
}
|
}
|
||||||
|
|
||||||
typedef struct Client_data_Cmp {
|
|
||||||
const Memory *mem;
|
|
||||||
uint64_t cur_time;
|
|
||||||
const uint8_t *comp_public_key;
|
|
||||||
} Client_data_Cmp;
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static int client_data_cmp(const Client_data_Cmp *cmp, const Client_data *entry1, const Client_data *entry2)
|
|
||||||
{
|
|
||||||
const bool t1 = assoc_timeout(cmp->cur_time, &entry1->assoc4) && assoc_timeout(cmp->cur_time, &entry1->assoc6);
|
|
||||||
const bool t2 = assoc_timeout(cmp->cur_time, &entry2->assoc4) && assoc_timeout(cmp->cur_time, &entry2->assoc6);
|
|
||||||
|
|
||||||
if (t1 && t2) {
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (t1) {
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (t2) {
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
const int closest = id_closest(cmp->comp_public_key, entry1->public_key, entry2->public_key);
|
|
||||||
|
|
||||||
if (closest == 1) {
|
|
||||||
return 1;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (closest == 2) {
|
|
||||||
return -1;
|
|
||||||
}
|
|
||||||
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static bool client_data_less_handler(const void *object, const void *a, const void *b)
|
|
||||||
{
|
|
||||||
const Client_data_Cmp *cmp = (const Client_data_Cmp *)object;
|
|
||||||
const Client_data *entry1 = (const Client_data *)a;
|
|
||||||
const Client_data *entry2 = (const Client_data *)b;
|
|
||||||
|
|
||||||
return client_data_cmp(cmp, entry1, entry2) < 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static const void *client_data_get_handler(const void *arr, uint32_t index)
|
|
||||||
{
|
|
||||||
const Client_data *entries = (const Client_data *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void client_data_set_handler(void *arr, uint32_t index, const void *val)
|
|
||||||
{
|
|
||||||
Client_data *entries = (Client_data *)arr;
|
|
||||||
const Client_data *entry = (const Client_data *)val;
|
|
||||||
entries[index] = *entry;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *client_data_subarr_handler(void *arr, uint32_t index, uint32_t size)
|
|
||||||
{
|
|
||||||
Client_data *entries = (Client_data *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *client_data_alloc_handler(const void *object, uint32_t size)
|
|
||||||
{
|
|
||||||
const Client_data_Cmp *cmp = (const Client_data_Cmp *)object;
|
|
||||||
Client_data *tmp = (Client_data *)mem_valloc(cmp->mem, size, sizeof(Client_data));
|
|
||||||
|
|
||||||
if (tmp == nullptr) {
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return tmp;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void client_data_delete_handler(const void *object, void *arr, uint32_t size)
|
|
||||||
{
|
|
||||||
const Client_data_Cmp *cmp = (const Client_data_Cmp *)object;
|
|
||||||
mem_delete(cmp->mem, arr);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const Sort_Funcs client_data_cmp_funcs = {
|
|
||||||
client_data_less_handler,
|
|
||||||
client_data_get_handler,
|
|
||||||
client_data_set_handler,
|
|
||||||
client_data_subarr_handler,
|
|
||||||
client_data_alloc_handler,
|
|
||||||
client_data_delete_handler,
|
|
||||||
};
|
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
static void sort_client_list(const Memory *mem, Client_data *list, uint64_t cur_time, unsigned int length,
|
static void sort_client_list(const Memory *mem, Client_data *list, uint64_t cur_time, unsigned int length,
|
||||||
const uint8_t *comp_public_key)
|
const uint8_t *comp_public_key)
|
||||||
{
|
{
|
||||||
// Pass comp_public_key to merge_sort with each Client_data entry, so the
|
// Pass comp_public_key to qsort with each Client_data entry, so the
|
||||||
// comparison function can use it as the base of comparison.
|
// comparison function can use it as the base of comparison.
|
||||||
const Client_data_Cmp cmp = {
|
DHT_Cmp_Data *cmp_list = (DHT_Cmp_Data *)mem_valloc(mem, length, sizeof(DHT_Cmp_Data));
|
||||||
mem,
|
|
||||||
cur_time,
|
|
||||||
comp_public_key,
|
|
||||||
};
|
|
||||||
|
|
||||||
merge_sort(list, length, &cmp, &client_data_cmp_funcs);
|
if (cmp_list == nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < length; ++i) {
|
||||||
|
cmp_list[i].cur_time = cur_time;
|
||||||
|
cmp_list[i].base_public_key = comp_public_key;
|
||||||
|
cmp_list[i].entry = list[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
qsort(cmp_list, length, sizeof(DHT_Cmp_Data), dht_cmp_entry);
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < length; ++i) {
|
||||||
|
list[i] = cmp_list[i].entry;
|
||||||
|
}
|
||||||
|
|
||||||
|
mem_delete(mem, cmp_list);
|
||||||
}
|
}
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
@ -1425,7 +1381,6 @@ static int handle_getnodes(void *object, const IP_Port *source, const uint8_t *p
|
|||||||
uint8_t plain[CRYPTO_NODE_SIZE];
|
uint8_t plain[CRYPTO_NODE_SIZE];
|
||||||
const uint8_t *shared_key = dht_get_shared_key_recv(dht, packet + 1);
|
const uint8_t *shared_key = dht_get_shared_key_recv(dht, packet + 1);
|
||||||
const int len = decrypt_data_symmetric(
|
const int len = decrypt_data_symmetric(
|
||||||
dht->mem,
|
|
||||||
shared_key,
|
shared_key,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
@ -1487,7 +1442,6 @@ static bool handle_sendnodes_core(void *object, const IP_Port *source, const uin
|
|||||||
VLA(uint8_t, plain, plain_size);
|
VLA(uint8_t, plain, plain_size);
|
||||||
const uint8_t *shared_key = dht_get_shared_key_sent(dht, packet + 1);
|
const uint8_t *shared_key = dht_get_shared_key_sent(dht, packet + 1);
|
||||||
const int len = decrypt_data_symmetric(
|
const int len = decrypt_data_symmetric(
|
||||||
dht->mem,
|
|
||||||
shared_key,
|
shared_key,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
@ -2169,7 +2123,7 @@ static int send_nat_ping(const DHT *dht, const uint8_t *public_key, uint64_t pin
|
|||||||
memcpy(data + 1, &ping_id, sizeof(uint64_t));
|
memcpy(data + 1, &ping_id, sizeof(uint64_t));
|
||||||
/* 254 is NAT ping request packet id */
|
/* 254 is NAT ping request packet id */
|
||||||
const int len = create_request(
|
const int len = create_request(
|
||||||
dht->mem, dht->rng, dht->self_public_key, dht->self_secret_key, packet_data, public_key,
|
dht->rng, dht->self_public_key, dht->self_secret_key, packet_data, public_key,
|
||||||
data, sizeof(uint64_t) + 1, CRYPTO_PACKET_NAT_PING);
|
data, sizeof(uint64_t) + 1, CRYPTO_PACKET_NAT_PING);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
@ -2504,7 +2458,7 @@ static int cryptopacket_handle(void *object, const IP_Port *source, const uint8_
|
|||||||
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE];
|
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE];
|
||||||
uint8_t data[MAX_CRYPTO_REQUEST_SIZE];
|
uint8_t data[MAX_CRYPTO_REQUEST_SIZE];
|
||||||
uint8_t number;
|
uint8_t number;
|
||||||
const int len = handle_request(dht->mem, dht->self_public_key, dht->self_secret_key, public_key,
|
const int len = handle_request(dht->self_public_key, dht->self_secret_key, public_key,
|
||||||
data, &number, packet, length);
|
data, &number, packet, length);
|
||||||
|
|
||||||
if (len == -1 || len == 0) {
|
if (len == -1 || len == 0) {
|
||||||
|
4
external/toxcore/c-toxcore/toxcore/DHT.h
vendored
4
external/toxcore/c-toxcore/toxcore/DHT.h
vendored
@ -99,7 +99,7 @@ extern "C" {
|
|||||||
* @return the length of the created packet on success.
|
* @return the length of the created packet on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int create_request(const Memory *mem, const Random *rng, const uint8_t *send_public_key, const uint8_t *send_secret_key,
|
int create_request(const Random *rng, const uint8_t *send_public_key, const uint8_t *send_secret_key,
|
||||||
uint8_t *packet, const uint8_t *recv_public_key,
|
uint8_t *packet, const uint8_t *recv_public_key,
|
||||||
const uint8_t *data, uint32_t data_length, uint8_t request_id);
|
const uint8_t *data, uint32_t data_length, uint8_t request_id);
|
||||||
|
|
||||||
@ -127,7 +127,7 @@ int create_request(const Memory *mem, const Random *rng, const uint8_t *send_pub
|
|||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int handle_request(
|
int handle_request(
|
||||||
const Memory *mem, const uint8_t *self_public_key, const uint8_t *self_secret_key, uint8_t *public_key, uint8_t *data,
|
const uint8_t *self_public_key, const uint8_t *self_secret_key, uint8_t *public_key, uint8_t *data,
|
||||||
uint8_t *request_id, const uint8_t *packet, uint16_t packet_length);
|
uint8_t *request_id, const uint8_t *packet, uint16_t packet_length);
|
||||||
|
|
||||||
typedef struct IPPTs {
|
typedef struct IPPTs {
|
||||||
|
@ -6,22 +6,19 @@
|
|||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "../testing/fuzzing/fuzz_support.hh"
|
#include "../testing/fuzzing/fuzz_support.hh"
|
||||||
#include "mem_test_util.hh"
|
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
void TestHandleRequest(Fuzz_Data &input)
|
void TestHandleRequest(Fuzz_Data &input)
|
||||||
{
|
{
|
||||||
const Test_Memory mem;
|
|
||||||
|
|
||||||
CONSUME_OR_RETURN(const uint8_t *self_public_key, input, CRYPTO_PUBLIC_KEY_SIZE);
|
CONSUME_OR_RETURN(const uint8_t *self_public_key, input, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
CONSUME_OR_RETURN(const uint8_t *self_secret_key, input, CRYPTO_SECRET_KEY_SIZE);
|
CONSUME_OR_RETURN(const uint8_t *self_secret_key, input, CRYPTO_SECRET_KEY_SIZE);
|
||||||
|
|
||||||
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE];
|
uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE];
|
||||||
uint8_t request[MAX_CRYPTO_REQUEST_SIZE];
|
uint8_t request[MAX_CRYPTO_REQUEST_SIZE];
|
||||||
uint8_t request_id;
|
uint8_t request_id;
|
||||||
handle_request(mem, self_public_key, self_secret_key, public_key, request, &request_id,
|
handle_request(self_public_key, self_secret_key, public_key, request, &request_id, input.data(),
|
||||||
input.data(), input.size());
|
input.size());
|
||||||
}
|
}
|
||||||
|
|
||||||
void TestUnpackNodes(Fuzz_Data &input)
|
void TestUnpackNodes(Fuzz_Data &input)
|
||||||
|
15
external/toxcore/c-toxcore/toxcore/DHT_test.cc
vendored
15
external/toxcore/c-toxcore/toxcore/DHT_test.cc
vendored
@ -274,7 +274,6 @@ TEST(AddToList, KeepsKeysInOrder)
|
|||||||
|
|
||||||
TEST(Request, CreateAndParse)
|
TEST(Request, CreateAndParse)
|
||||||
{
|
{
|
||||||
Test_Memory mem;
|
|
||||||
Test_Random rng;
|
Test_Random rng;
|
||||||
|
|
||||||
// Peers.
|
// Peers.
|
||||||
@ -294,32 +293,32 @@ TEST(Request, CreateAndParse)
|
|||||||
std::vector<uint8_t> outgoing(919);
|
std::vector<uint8_t> outgoing(919);
|
||||||
random_bytes(rng, outgoing.data(), outgoing.size());
|
random_bytes(rng, outgoing.data(), outgoing.size());
|
||||||
|
|
||||||
EXPECT_LT(create_request(mem, rng, sender.pk.data(), sender.sk.data(), packet.data(),
|
EXPECT_LT(create_request(rng, sender.pk.data(), sender.sk.data(), packet.data(),
|
||||||
receiver.pk.data(), outgoing.data(), outgoing.size(), sent_pkt_id),
|
receiver.pk.data(), outgoing.data(), outgoing.size(), sent_pkt_id),
|
||||||
0);
|
0);
|
||||||
|
|
||||||
// Pop one element so the payload is 918 bytes. Packing should now succeed.
|
// Pop one element so the payload is 918 bytes. Packing should now succeed.
|
||||||
outgoing.pop_back();
|
outgoing.pop_back();
|
||||||
|
|
||||||
const int max_sent_length = create_request(mem, rng, sender.pk.data(), sender.sk.data(),
|
const int max_sent_length = create_request(rng, sender.pk.data(), sender.sk.data(),
|
||||||
packet.data(), receiver.pk.data(), outgoing.data(), outgoing.size(), sent_pkt_id);
|
packet.data(), receiver.pk.data(), outgoing.data(), outgoing.size(), sent_pkt_id);
|
||||||
ASSERT_GT(max_sent_length, 0); // success.
|
ASSERT_GT(max_sent_length, 0); // success.
|
||||||
|
|
||||||
// Check that handle_request rejects packets larger than the maximum created packet size.
|
// Check that handle_request rejects packets larger than the maximum created packet size.
|
||||||
EXPECT_LT(handle_request(mem, receiver.pk.data(), receiver.sk.data(), pk.data(),
|
EXPECT_LT(handle_request(receiver.pk.data(), receiver.sk.data(), pk.data(), incoming.data(),
|
||||||
incoming.data(), &recvd_pkt_id, packet.data(), max_sent_length + 1),
|
&recvd_pkt_id, packet.data(), max_sent_length + 1),
|
||||||
0);
|
0);
|
||||||
|
|
||||||
// Now try all possible packet sizes from max (918) to 0.
|
// Now try all possible packet sizes from max (918) to 0.
|
||||||
while (!outgoing.empty()) {
|
while (!outgoing.empty()) {
|
||||||
// Pack:
|
// Pack:
|
||||||
const int sent_length = create_request(mem, rng, sender.pk.data(), sender.sk.data(),
|
const int sent_length = create_request(rng, sender.pk.data(), sender.sk.data(),
|
||||||
packet.data(), receiver.pk.data(), outgoing.data(), outgoing.size(), sent_pkt_id);
|
packet.data(), receiver.pk.data(), outgoing.data(), outgoing.size(), sent_pkt_id);
|
||||||
ASSERT_GT(sent_length, 0);
|
ASSERT_GT(sent_length, 0);
|
||||||
|
|
||||||
// Unpack:
|
// Unpack:
|
||||||
const int recvd_length = handle_request(mem, receiver.pk.data(), receiver.sk.data(),
|
const int recvd_length = handle_request(receiver.pk.data(), receiver.sk.data(), pk.data(),
|
||||||
pk.data(), incoming.data(), &recvd_pkt_id, packet.data(), sent_length);
|
incoming.data(), &recvd_pkt_id, packet.data(), sent_length);
|
||||||
ASSERT_GE(recvd_length, 0);
|
ASSERT_GE(recvd_length, 0);
|
||||||
|
|
||||||
EXPECT_EQ(
|
EXPECT_EQ(
|
||||||
|
@ -74,6 +74,8 @@ libtoxcore_la_SOURCES = ../third_party/cmp/cmp.c \
|
|||||||
../toxcore/ping_array.c \
|
../toxcore/ping_array.c \
|
||||||
../toxcore/net_crypto.h \
|
../toxcore/net_crypto.h \
|
||||||
../toxcore/net_crypto.c \
|
../toxcore/net_crypto.c \
|
||||||
|
../toxcore/net_profile.c \
|
||||||
|
../toxcore/net_profile.h \
|
||||||
../toxcore/friend_requests.h \
|
../toxcore/friend_requests.h \
|
||||||
../toxcore/friend_requests.c \
|
../toxcore/friend_requests.c \
|
||||||
../toxcore/LAN_discovery.h \
|
../toxcore/LAN_discovery.h \
|
||||||
@ -86,8 +88,6 @@ libtoxcore_la_SOURCES = ../third_party/cmp/cmp.c \
|
|||||||
../toxcore/ping.c \
|
../toxcore/ping.c \
|
||||||
../toxcore/shared_key_cache.h \
|
../toxcore/shared_key_cache.h \
|
||||||
../toxcore/shared_key_cache.c \
|
../toxcore/shared_key_cache.c \
|
||||||
../toxcore/sort.h \
|
|
||||||
../toxcore/sort.c \
|
|
||||||
../toxcore/state.h \
|
../toxcore/state.h \
|
||||||
../toxcore/state.c \
|
../toxcore/state.c \
|
||||||
../toxcore/tox.h \
|
../toxcore/tox.h \
|
||||||
|
10
external/toxcore/c-toxcore/toxcore/TCP_client.c
vendored
10
external/toxcore/c-toxcore/toxcore/TCP_client.c
vendored
@ -20,6 +20,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
@ -312,7 +313,7 @@ static int generate_handshake(TCP_Client_Connection *tcp_conn)
|
|||||||
memcpy(plain + CRYPTO_PUBLIC_KEY_SIZE, tcp_conn->con.sent_nonce, CRYPTO_NONCE_SIZE);
|
memcpy(plain + CRYPTO_PUBLIC_KEY_SIZE, tcp_conn->con.sent_nonce, CRYPTO_NONCE_SIZE);
|
||||||
memcpy(tcp_conn->con.last_packet, tcp_conn->self_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(tcp_conn->con.last_packet, tcp_conn->self_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
random_nonce(tcp_conn->con.rng, tcp_conn->con.last_packet + CRYPTO_PUBLIC_KEY_SIZE);
|
random_nonce(tcp_conn->con.rng, tcp_conn->con.last_packet + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
const int len = encrypt_data_symmetric(tcp_conn->con.mem, tcp_conn->con.shared_key, tcp_conn->con.last_packet + CRYPTO_PUBLIC_KEY_SIZE, plain,
|
const int len = encrypt_data_symmetric(tcp_conn->con.shared_key, tcp_conn->con.last_packet + CRYPTO_PUBLIC_KEY_SIZE, plain,
|
||||||
sizeof(plain), tcp_conn->con.last_packet + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
sizeof(plain), tcp_conn->con.last_packet + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != sizeof(plain) + CRYPTO_MAC_SIZE) {
|
if (len != sizeof(plain) + CRYPTO_MAC_SIZE) {
|
||||||
@ -334,7 +335,7 @@ non_null()
|
|||||||
static int handle_handshake(TCP_Client_Connection *tcp_conn, const uint8_t *data)
|
static int handle_handshake(TCP_Client_Connection *tcp_conn, const uint8_t *data)
|
||||||
{
|
{
|
||||||
uint8_t plain[CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE];
|
uint8_t plain[CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE];
|
||||||
const int len = decrypt_data_symmetric(tcp_conn->con.mem, tcp_conn->con.shared_key, data, data + CRYPTO_NONCE_SIZE,
|
const int len = decrypt_data_symmetric(tcp_conn->con.shared_key, data, data + CRYPTO_NONCE_SIZE,
|
||||||
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, plain);
|
TCP_SERVER_HANDSHAKE_SIZE - CRYPTO_NONCE_SIZE, plain);
|
||||||
|
|
||||||
if (len != sizeof(plain)) {
|
if (len != sizeof(plain)) {
|
||||||
@ -582,7 +583,7 @@ void forwarding_handler(TCP_Client_Connection *con, forwarded_response_cb *forwa
|
|||||||
TCP_Client_Connection *new_tcp_connection(
|
TCP_Client_Connection *new_tcp_connection(
|
||||||
const Logger *logger, const Memory *mem, const Mono_Time *mono_time, const Random *rng, const Network *ns,
|
const Logger *logger, const Memory *mem, const Mono_Time *mono_time, const Random *rng, const Network *ns,
|
||||||
const IP_Port *ip_port, const uint8_t *public_key, const uint8_t *self_public_key, const uint8_t *self_secret_key,
|
const IP_Port *ip_port, const uint8_t *public_key, const uint8_t *self_public_key, const uint8_t *self_secret_key,
|
||||||
const TCP_Proxy_Info *proxy_info)
|
const TCP_Proxy_Info *proxy_info, Net_Profile *net_profile)
|
||||||
{
|
{
|
||||||
assert(logger != nullptr);
|
assert(logger != nullptr);
|
||||||
assert(mem != nullptr);
|
assert(mem != nullptr);
|
||||||
@ -634,6 +635,7 @@ TCP_Client_Connection *new_tcp_connection(
|
|||||||
temp->con.rng = rng;
|
temp->con.rng = rng;
|
||||||
temp->con.sock = sock;
|
temp->con.sock = sock;
|
||||||
temp->con.ip_port = *ip_port;
|
temp->con.ip_port = *ip_port;
|
||||||
|
temp->con.net_profile = net_profile;
|
||||||
memcpy(temp->public_key, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(temp->public_key, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
memcpy(temp->self_public_key, self_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(temp->self_public_key, self_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
encrypt_precompute(temp->public_key, self_secret_key, temp->con.shared_key);
|
encrypt_precompute(temp->public_key, self_secret_key, temp->con.shared_key);
|
||||||
@ -819,6 +821,8 @@ static int handle_tcp_client_packet(const Logger *logger, TCP_Client_Connection
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
netprof_record_packet(conn->con.net_profile, data[0], length, PACKET_DIRECTION_RECV);
|
||||||
|
|
||||||
switch (data[0]) {
|
switch (data[0]) {
|
||||||
case TCP_PACKET_ROUTING_RESPONSE:
|
case TCP_PACKET_ROUTING_RESPONSE:
|
||||||
return handle_tcp_client_routing_response(conn, data, length);
|
return handle_tcp_client_routing_response(conn, data, length);
|
||||||
|
@ -15,6 +15,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
|
|
||||||
#define TCP_CONNECTION_TIMEOUT 10
|
#define TCP_CONNECTION_TIMEOUT 10
|
||||||
@ -60,11 +61,11 @@ non_null()
|
|||||||
void tcp_con_set_custom_uint(TCP_Client_Connection *con, uint32_t value);
|
void tcp_con_set_custom_uint(TCP_Client_Connection *con, uint32_t value);
|
||||||
|
|
||||||
/** Create new TCP connection to ip_port/public_key */
|
/** Create new TCP connection to ip_port/public_key */
|
||||||
non_null(1, 2, 3, 4, 5, 6, 7, 8, 9) nullable(10)
|
non_null(1, 2, 3, 4, 5, 6, 7, 8, 9) nullable(10, 11)
|
||||||
TCP_Client_Connection *new_tcp_connection(
|
TCP_Client_Connection *new_tcp_connection(
|
||||||
const Logger *logger, const Memory *mem, const Mono_Time *mono_time, const Random *rng, const Network *ns,
|
const Logger *logger, const Memory *mem, const Mono_Time *mono_time, const Random *rng, const Network *ns,
|
||||||
const IP_Port *ip_port, const uint8_t *public_key, const uint8_t *self_public_key, const uint8_t *self_secret_key,
|
const IP_Port *ip_port, const uint8_t *public_key, const uint8_t *self_public_key, const uint8_t *self_secret_key,
|
||||||
const TCP_Proxy_Info *proxy_info);
|
const TCP_Proxy_Info *proxy_info, Net_Profile *net_profile);
|
||||||
|
|
||||||
/** Run the TCP connection */
|
/** Run the TCP connection */
|
||||||
non_null(1, 2, 3) nullable(4)
|
non_null(1, 2, 3) nullable(4)
|
||||||
|
14
external/toxcore/c-toxcore/toxcore/TCP_common.c
vendored
14
external/toxcore/c-toxcore/toxcore/TCP_common.c
vendored
@ -35,7 +35,8 @@ int send_pending_data_nonpriority(const Logger *logger, TCP_Connection *con)
|
|||||||
}
|
}
|
||||||
|
|
||||||
const uint16_t left = con->last_packet_length - con->last_packet_sent;
|
const uint16_t left = con->last_packet_length - con->last_packet_sent;
|
||||||
const int len = net_send(con->ns, logger, con->sock, con->last_packet + con->last_packet_sent, left, &con->ip_port);
|
const int len = net_send(con->ns, logger, con->sock, con->last_packet + con->last_packet_sent, left, &con->ip_port,
|
||||||
|
con->net_profile);
|
||||||
|
|
||||||
if (len <= 0) {
|
if (len <= 0) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -66,7 +67,7 @@ int send_pending_data(const Logger *logger, TCP_Connection *con)
|
|||||||
|
|
||||||
while (p != nullptr) {
|
while (p != nullptr) {
|
||||||
const uint16_t left = p->size - p->sent;
|
const uint16_t left = p->size - p->sent;
|
||||||
const int len = net_send(con->ns, logger, con->sock, p->data + p->sent, left, &con->ip_port);
|
const int len = net_send(con->ns, logger, con->sock, p->data + p->sent, left, &con->ip_port, con->net_profile);
|
||||||
|
|
||||||
if (len != left) {
|
if (len != left) {
|
||||||
if (len > 0) {
|
if (len > 0) {
|
||||||
@ -157,14 +158,15 @@ int write_packet_tcp_secure_connection(const Logger *logger, TCP_Connection *con
|
|||||||
|
|
||||||
uint16_t c_length = net_htons(length + CRYPTO_MAC_SIZE);
|
uint16_t c_length = net_htons(length + CRYPTO_MAC_SIZE);
|
||||||
memcpy(packet, &c_length, sizeof(uint16_t));
|
memcpy(packet, &c_length, sizeof(uint16_t));
|
||||||
int len = encrypt_data_symmetric(con->mem, con->shared_key, con->sent_nonce, data, length, packet + sizeof(uint16_t));
|
int len = encrypt_data_symmetric(con->shared_key, con->sent_nonce, data, length, packet + sizeof(uint16_t));
|
||||||
|
|
||||||
if ((unsigned int)len != (packet_size - sizeof(uint16_t))) {
|
if ((unsigned int)len != (packet_size - sizeof(uint16_t))) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (priority) {
|
if (priority) {
|
||||||
len = sendpriority ? net_send(con->ns, logger, con->sock, packet, packet_size, &con->ip_port) : 0;
|
len = sendpriority ? net_send(con->ns, logger, con->sock, packet, packet_size, &con->ip_port,
|
||||||
|
con->net_profile) : 0;
|
||||||
|
|
||||||
if (len <= 0) {
|
if (len <= 0) {
|
||||||
len = 0;
|
len = 0;
|
||||||
@ -179,7 +181,7 @@ int write_packet_tcp_secure_connection(const Logger *logger, TCP_Connection *con
|
|||||||
return add_priority(con, packet, packet_size, len) ? 1 : 0;
|
return add_priority(con, packet, packet_size, len) ? 1 : 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
len = net_send(con->ns, logger, con->sock, packet, packet_size, &con->ip_port);
|
len = net_send(con->ns, logger, con->sock, packet, packet_size, &con->ip_port, con->net_profile);
|
||||||
|
|
||||||
if (len <= 0) {
|
if (len <= 0) {
|
||||||
return 0;
|
return 0;
|
||||||
@ -305,7 +307,7 @@ int read_packet_tcp_secure_connection(
|
|||||||
|
|
||||||
*next_packet_length = 0;
|
*next_packet_length = 0;
|
||||||
|
|
||||||
const int len = decrypt_data_symmetric(mem, shared_key, recv_nonce, data_encrypted, len_packet, data);
|
const int len = decrypt_data_symmetric(shared_key, recv_nonce, data_encrypted, len_packet, data);
|
||||||
|
|
||||||
if (len + CRYPTO_MAC_SIZE != len_packet) {
|
if (len + CRYPTO_MAC_SIZE != len_packet) {
|
||||||
LOGGER_ERROR(logger, "decrypted length %d does not match expected length %d", len + CRYPTO_MAC_SIZE, len_packet);
|
LOGGER_ERROR(logger, "decrypted length %d does not match expected length %d", len + CRYPTO_MAC_SIZE, len_packet);
|
||||||
|
@ -10,6 +10,7 @@
|
|||||||
#include "crypto_core.h"
|
#include "crypto_core.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
|
|
||||||
typedef struct TCP_Priority_List TCP_Priority_List;
|
typedef struct TCP_Priority_List TCP_Priority_List;
|
||||||
@ -66,6 +67,10 @@ typedef struct TCP_Connection {
|
|||||||
|
|
||||||
TCP_Priority_List *priority_queue_start;
|
TCP_Priority_List *priority_queue_start;
|
||||||
TCP_Priority_List *priority_queue_end;
|
TCP_Priority_List *priority_queue_end;
|
||||||
|
|
||||||
|
// This is a shared pointer to the parent's respective Net_Profile object
|
||||||
|
// (either TCP_Server for TCP server packets or TCP_Connections for TCP client packets).
|
||||||
|
Net_Profile *net_profile;
|
||||||
} TCP_Connection;
|
} TCP_Connection;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
|
@ -20,6 +20,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
@ -56,6 +57,9 @@ struct TCP_Connections {
|
|||||||
|
|
||||||
bool onion_status;
|
bool onion_status;
|
||||||
uint16_t onion_num_conns;
|
uint16_t onion_num_conns;
|
||||||
|
|
||||||
|
/* Network profile for all TCP client packets. */
|
||||||
|
Net_Profile *net_profile;
|
||||||
};
|
};
|
||||||
|
|
||||||
static const TCP_Connection_to empty_tcp_connection_to = {0};
|
static const TCP_Connection_to empty_tcp_connection_to = {0};
|
||||||
@ -928,7 +932,8 @@ static int reconnect_tcp_relay_connection(TCP_Connections *tcp_c, int tcp_connec
|
|||||||
uint8_t relay_pk[CRYPTO_PUBLIC_KEY_SIZE];
|
uint8_t relay_pk[CRYPTO_PUBLIC_KEY_SIZE];
|
||||||
memcpy(relay_pk, tcp_con_public_key(tcp_con->connection), CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(relay_pk, tcp_con_public_key(tcp_con->connection), CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
kill_tcp_connection(tcp_con->connection);
|
kill_tcp_connection(tcp_con->connection);
|
||||||
tcp_con->connection = new_tcp_connection(tcp_c->logger, tcp_c->mem, tcp_c->mono_time, tcp_c->rng, tcp_c->ns, &ip_port, relay_pk, tcp_c->self_public_key, tcp_c->self_secret_key, &tcp_c->proxy_info);
|
tcp_con->connection = new_tcp_connection(tcp_c->logger, tcp_c->mem, tcp_c->mono_time, tcp_c->rng, tcp_c->ns, &ip_port, relay_pk, tcp_c->self_public_key, tcp_c->self_secret_key, &tcp_c->proxy_info,
|
||||||
|
tcp_c->net_profile);
|
||||||
|
|
||||||
if (tcp_con->connection == nullptr) {
|
if (tcp_con->connection == nullptr) {
|
||||||
kill_tcp_relay_connection(tcp_c, tcp_connections_number);
|
kill_tcp_relay_connection(tcp_c, tcp_connections_number);
|
||||||
@ -1017,7 +1022,7 @@ static int unsleep_tcp_relay_connection(TCP_Connections *tcp_c, int tcp_connecti
|
|||||||
|
|
||||||
tcp_con->connection = new_tcp_connection(
|
tcp_con->connection = new_tcp_connection(
|
||||||
tcp_c->logger, tcp_c->mem, tcp_c->mono_time, tcp_c->rng, tcp_c->ns, &tcp_con->ip_port,
|
tcp_c->logger, tcp_c->mem, tcp_c->mono_time, tcp_c->rng, tcp_c->ns, &tcp_con->ip_port,
|
||||||
tcp_con->relay_pk, tcp_c->self_public_key, tcp_c->self_secret_key, &tcp_c->proxy_info);
|
tcp_con->relay_pk, tcp_c->self_public_key, tcp_c->self_secret_key, &tcp_c->proxy_info, tcp_c->net_profile);
|
||||||
|
|
||||||
if (tcp_con->connection == nullptr) {
|
if (tcp_con->connection == nullptr) {
|
||||||
kill_tcp_relay_connection(tcp_c, tcp_connections_number);
|
kill_tcp_relay_connection(tcp_c, tcp_connections_number);
|
||||||
@ -1315,7 +1320,7 @@ static int add_tcp_relay_instance(TCP_Connections *tcp_c, const IP_Port *ip_port
|
|||||||
|
|
||||||
tcp_con->connection = new_tcp_connection(
|
tcp_con->connection = new_tcp_connection(
|
||||||
tcp_c->logger, tcp_c->mem, tcp_c->mono_time, tcp_c->rng, tcp_c->ns, &ipp_copy,
|
tcp_c->logger, tcp_c->mem, tcp_c->mono_time, tcp_c->rng, tcp_c->ns, &ipp_copy,
|
||||||
relay_pk, tcp_c->self_public_key, tcp_c->self_secret_key, &tcp_c->proxy_info);
|
relay_pk, tcp_c->self_public_key, tcp_c->self_secret_key, &tcp_c->proxy_info, tcp_c->net_profile);
|
||||||
|
|
||||||
if (tcp_con->connection == nullptr) {
|
if (tcp_con->connection == nullptr) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -1609,6 +1614,14 @@ TCP_Connections *new_tcp_connections(const Logger *logger, const Memory *mem, co
|
|||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Net_Profile *np = netprof_new(logger, mem);
|
||||||
|
|
||||||
|
if (np == nullptr) {
|
||||||
|
mem_delete(mem, temp);
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
temp->net_profile = np;
|
||||||
temp->logger = logger;
|
temp->logger = logger;
|
||||||
temp->mem = mem;
|
temp->mem = mem;
|
||||||
temp->rng = rng;
|
temp->rng = rng;
|
||||||
@ -1723,7 +1736,17 @@ void kill_tcp_connections(TCP_Connections *tcp_c)
|
|||||||
|
|
||||||
crypto_memzero(tcp_c->self_secret_key, sizeof(tcp_c->self_secret_key));
|
crypto_memzero(tcp_c->self_secret_key, sizeof(tcp_c->self_secret_key));
|
||||||
|
|
||||||
|
netprof_kill(tcp_c->mem, tcp_c->net_profile);
|
||||||
mem_delete(tcp_c->mem, tcp_c->tcp_connections);
|
mem_delete(tcp_c->mem, tcp_c->tcp_connections);
|
||||||
mem_delete(tcp_c->mem, tcp_c->connections);
|
mem_delete(tcp_c->mem, tcp_c->connections);
|
||||||
mem_delete(tcp_c->mem, tcp_c);
|
mem_delete(tcp_c->mem, tcp_c);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const Net_Profile *tcp_connection_get_client_net_profile(const TCP_Connections *tcp_c)
|
||||||
|
{
|
||||||
|
if (tcp_c == nullptr) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return tcp_c->net_profile;
|
||||||
|
}
|
||||||
|
@ -21,6 +21,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
|
|
||||||
#define TCP_CONN_NONE 0
|
#define TCP_CONN_NONE 0
|
||||||
@ -317,4 +318,11 @@ void do_tcp_connections(const Logger *logger, TCP_Connections *tcp_c, void *user
|
|||||||
nullable(1)
|
nullable(1)
|
||||||
void kill_tcp_connections(TCP_Connections *tcp_c);
|
void kill_tcp_connections(TCP_Connections *tcp_c);
|
||||||
|
|
||||||
|
/** @brief a pointer to the tcp client net profile associated with tcp_c.
|
||||||
|
*
|
||||||
|
* @retval null if tcp_c is null.
|
||||||
|
*/
|
||||||
|
non_null()
|
||||||
|
const Net_Profile *tcp_connection_get_client_net_profile(const TCP_Connections *tcp_c);
|
||||||
|
|
||||||
#endif /* C_TOXCORE_TOXCORE_TCP_CONNECTION_H */
|
#endif /* C_TOXCORE_TOXCORE_TCP_CONNECTION_H */
|
||||||
|
33
external/toxcore/c-toxcore/toxcore/TCP_server.c
vendored
33
external/toxcore/c-toxcore/toxcore/TCP_server.c
vendored
@ -27,6 +27,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "onion.h"
|
#include "onion.h"
|
||||||
|
|
||||||
@ -91,6 +92,9 @@ struct TCP_Server {
|
|||||||
uint64_t counter;
|
uint64_t counter;
|
||||||
|
|
||||||
BS_List accepted_key_list;
|
BS_List accepted_key_list;
|
||||||
|
|
||||||
|
/* Network profile for all TCP server packets. */
|
||||||
|
Net_Profile *net_profile;
|
||||||
};
|
};
|
||||||
|
|
||||||
static_assert(sizeof(TCP_Server) < 7 * 1024 * 1024,
|
static_assert(sizeof(TCP_Server) < 7 * 1024 * 1024,
|
||||||
@ -236,6 +240,7 @@ static int add_accepted(TCP_Server *tcp_server, const Mono_Time *mono_time, TCP_
|
|||||||
tcp_server->accepted_connection_array[index].identifier = ++tcp_server->counter;
|
tcp_server->accepted_connection_array[index].identifier = ++tcp_server->counter;
|
||||||
tcp_server->accepted_connection_array[index].last_pinged = mono_time_get(mono_time);
|
tcp_server->accepted_connection_array[index].last_pinged = mono_time_get(mono_time);
|
||||||
tcp_server->accepted_connection_array[index].ping_id = 0;
|
tcp_server->accepted_connection_array[index].ping_id = 0;
|
||||||
|
tcp_server->accepted_connection_array[index].con.net_profile = tcp_server->net_profile;
|
||||||
|
|
||||||
return index;
|
return index;
|
||||||
}
|
}
|
||||||
@ -327,7 +332,7 @@ static int handle_tcp_handshake(const Logger *logger, TCP_Secure_Connection *con
|
|||||||
uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE];
|
uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE];
|
||||||
encrypt_precompute(data, self_secret_key, shared_key);
|
encrypt_precompute(data, self_secret_key, shared_key);
|
||||||
uint8_t plain[TCP_HANDSHAKE_PLAIN_SIZE];
|
uint8_t plain[TCP_HANDSHAKE_PLAIN_SIZE];
|
||||||
int len = decrypt_data_symmetric(con->con.mem, shared_key, data + CRYPTO_PUBLIC_KEY_SIZE,
|
int len = decrypt_data_symmetric(shared_key, data + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
data + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE, TCP_HANDSHAKE_PLAIN_SIZE + CRYPTO_MAC_SIZE, plain);
|
data + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE, TCP_HANDSHAKE_PLAIN_SIZE + CRYPTO_MAC_SIZE, plain);
|
||||||
|
|
||||||
if (len != TCP_HANDSHAKE_PLAIN_SIZE) {
|
if (len != TCP_HANDSHAKE_PLAIN_SIZE) {
|
||||||
@ -347,7 +352,7 @@ static int handle_tcp_handshake(const Logger *logger, TCP_Secure_Connection *con
|
|||||||
uint8_t response[TCP_SERVER_HANDSHAKE_SIZE];
|
uint8_t response[TCP_SERVER_HANDSHAKE_SIZE];
|
||||||
random_nonce(con->con.rng, response);
|
random_nonce(con->con.rng, response);
|
||||||
|
|
||||||
len = encrypt_data_symmetric(con->con.mem, shared_key, response, resp_plain, TCP_HANDSHAKE_PLAIN_SIZE,
|
len = encrypt_data_symmetric(shared_key, response, resp_plain, TCP_HANDSHAKE_PLAIN_SIZE,
|
||||||
response + CRYPTO_NONCE_SIZE);
|
response + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != TCP_HANDSHAKE_PLAIN_SIZE + CRYPTO_MAC_SIZE) {
|
if (len != TCP_HANDSHAKE_PLAIN_SIZE + CRYPTO_MAC_SIZE) {
|
||||||
@ -357,7 +362,7 @@ static int handle_tcp_handshake(const Logger *logger, TCP_Secure_Connection *con
|
|||||||
|
|
||||||
const IP_Port ipp = {{{0}}};
|
const IP_Port ipp = {{{0}}};
|
||||||
|
|
||||||
if (TCP_SERVER_HANDSHAKE_SIZE != net_send(con->con.ns, logger, con->con.sock, response, TCP_SERVER_HANDSHAKE_SIZE, &ipp)) {
|
if (TCP_SERVER_HANDSHAKE_SIZE != net_send(con->con.ns, logger, con->con.sock, response, TCP_SERVER_HANDSHAKE_SIZE, &ipp, con->con.net_profile)) {
|
||||||
crypto_memzero(shared_key, sizeof(shared_key));
|
crypto_memzero(shared_key, sizeof(shared_key));
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
@ -680,6 +685,7 @@ static int handle_tcp_packet(TCP_Server *tcp_server, uint32_t con_id, const uint
|
|||||||
}
|
}
|
||||||
|
|
||||||
TCP_Secure_Connection *const con = &tcp_server->accepted_connection_array[con_id];
|
TCP_Secure_Connection *const con = &tcp_server->accepted_connection_array[con_id];
|
||||||
|
netprof_record_packet(con->con.net_profile, data[0], length, PACKET_DIRECTION_RECV);
|
||||||
|
|
||||||
switch (data[0]) {
|
switch (data[0]) {
|
||||||
case TCP_PACKET_ROUTING_REQUEST: {
|
case TCP_PACKET_ROUTING_REQUEST: {
|
||||||
@ -969,6 +975,14 @@ TCP_Server *new_tcp_server(const Logger *logger, const Memory *mem, const Random
|
|||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Net_Profile *np = netprof_new(logger, mem);
|
||||||
|
|
||||||
|
if (np == nullptr) {
|
||||||
|
mem_delete(mem, temp);
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
temp->net_profile = np;
|
||||||
temp->logger = logger;
|
temp->logger = logger;
|
||||||
temp->mem = mem;
|
temp->mem = mem;
|
||||||
temp->ns = ns;
|
temp->ns = ns;
|
||||||
@ -978,6 +992,7 @@ TCP_Server *new_tcp_server(const Logger *logger, const Memory *mem, const Random
|
|||||||
|
|
||||||
if (socks_listening == nullptr) {
|
if (socks_listening == nullptr) {
|
||||||
LOGGER_ERROR(logger, "socket allocation failed");
|
LOGGER_ERROR(logger, "socket allocation failed");
|
||||||
|
netprof_kill(mem, temp->net_profile);
|
||||||
mem_delete(mem, temp);
|
mem_delete(mem, temp);
|
||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
@ -989,6 +1004,7 @@ TCP_Server *new_tcp_server(const Logger *logger, const Memory *mem, const Random
|
|||||||
|
|
||||||
if (temp->efd == -1) {
|
if (temp->efd == -1) {
|
||||||
LOGGER_ERROR(logger, "epoll initialisation failed");
|
LOGGER_ERROR(logger, "epoll initialisation failed");
|
||||||
|
netprof_kill(mem, temp->net_profile);
|
||||||
mem_delete(mem, socks_listening);
|
mem_delete(mem, socks_listening);
|
||||||
mem_delete(mem, temp);
|
mem_delete(mem, temp);
|
||||||
return nullptr;
|
return nullptr;
|
||||||
@ -1022,6 +1038,7 @@ TCP_Server *new_tcp_server(const Logger *logger, const Memory *mem, const Random
|
|||||||
}
|
}
|
||||||
|
|
||||||
if (temp->num_listening_socks == 0) {
|
if (temp->num_listening_socks == 0) {
|
||||||
|
netprof_kill(mem, temp->net_profile);
|
||||||
mem_delete(mem, temp->socks_listening);
|
mem_delete(mem, temp->socks_listening);
|
||||||
mem_delete(mem, temp);
|
mem_delete(mem, temp);
|
||||||
return nullptr;
|
return nullptr;
|
||||||
@ -1422,6 +1439,16 @@ void kill_tcp_server(TCP_Server *tcp_server)
|
|||||||
|
|
||||||
crypto_memzero(tcp_server->secret_key, sizeof(tcp_server->secret_key));
|
crypto_memzero(tcp_server->secret_key, sizeof(tcp_server->secret_key));
|
||||||
|
|
||||||
|
netprof_kill(tcp_server->mem, tcp_server->net_profile);
|
||||||
mem_delete(tcp_server->mem, tcp_server->socks_listening);
|
mem_delete(tcp_server->mem, tcp_server->socks_listening);
|
||||||
mem_delete(tcp_server->mem, tcp_server);
|
mem_delete(tcp_server->mem, tcp_server);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const Net_Profile *tcp_server_get_net_profile(const TCP_Server *tcp_server)
|
||||||
|
{
|
||||||
|
if (tcp_server == nullptr) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return tcp_server->net_profile;
|
||||||
|
}
|
||||||
|
@ -15,6 +15,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "onion.h"
|
#include "onion.h"
|
||||||
|
|
||||||
@ -52,4 +53,11 @@ void do_tcp_server(TCP_Server *tcp_server, const Mono_Time *mono_time);
|
|||||||
nullable(1)
|
nullable(1)
|
||||||
void kill_tcp_server(TCP_Server *tcp_server);
|
void kill_tcp_server(TCP_Server *tcp_server);
|
||||||
|
|
||||||
|
/** @brief Returns a pointer to the net profile associated with `tcp_server`.
|
||||||
|
*
|
||||||
|
* Returns null if `tcp_server` is null.
|
||||||
|
*/
|
||||||
|
nullable(1)
|
||||||
|
const Net_Profile *tcp_server_get_net_profile(const TCP_Server *tcp_server);
|
||||||
|
|
||||||
#endif /* C_TOXCORE_TOXCORE_TCP_SERVER_H */
|
#endif /* C_TOXCORE_TOXCORE_TCP_SERVER_H */
|
||||||
|
@ -451,7 +451,7 @@ static int create_reply_plain_store_announce_request(Announcements *announce,
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (decrypt_data_symmetric(announce->mem, shared_key,
|
if (decrypt_data_symmetric(shared_key,
|
||||||
data + CRYPTO_PUBLIC_KEY_SIZE,
|
data + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
data + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
data + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
plain_len + CRYPTO_MAC_SIZE,
|
plain_len + CRYPTO_MAC_SIZE,
|
||||||
@ -568,7 +568,7 @@ static int create_reply(Announcements *announce, const IP_Port *source,
|
|||||||
VLA(uint8_t, plain, plain_len);
|
VLA(uint8_t, plain, plain_len);
|
||||||
const uint8_t *shared_key = dht_get_shared_key_recv(announce->dht, data + 1);
|
const uint8_t *shared_key = dht_get_shared_key_recv(announce->dht, data + 1);
|
||||||
|
|
||||||
if (decrypt_data_symmetric(announce->mem, shared_key,
|
if (decrypt_data_symmetric(shared_key,
|
||||||
data + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
data + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
data + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
data + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
plain_len + CRYPTO_MAC_SIZE,
|
plain_len + CRYPTO_MAC_SIZE,
|
||||||
|
60
external/toxcore/c-toxcore/toxcore/crypto_core.c
vendored
60
external/toxcore/c-toxcore/toxcore/crypto_core.c
vendored
@ -13,7 +13,6 @@
|
|||||||
|
|
||||||
#include "attributes.h"
|
#include "attributes.h"
|
||||||
#include "ccompat.h"
|
#include "ccompat.h"
|
||||||
#include "mem.h"
|
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
static_assert(CRYPTO_PUBLIC_KEY_SIZE == crypto_box_PUBLICKEYBYTES,
|
static_assert(CRYPTO_PUBLIC_KEY_SIZE == crypto_box_PUBLICKEYBYTES,
|
||||||
@ -89,10 +88,9 @@ const uint8_t *get_chat_id(const Extended_Public_Key *key)
|
|||||||
}
|
}
|
||||||
|
|
||||||
#if !defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION)
|
#if !defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION)
|
||||||
non_null()
|
static uint8_t *crypto_malloc(size_t bytes)
|
||||||
static uint8_t *crypto_malloc(const Memory *mem, size_t bytes)
|
|
||||||
{
|
{
|
||||||
uint8_t *ptr = (uint8_t *)mem_balloc(mem, bytes);
|
uint8_t *ptr = (uint8_t *)malloc(bytes);
|
||||||
|
|
||||||
if (ptr != nullptr) {
|
if (ptr != nullptr) {
|
||||||
crypto_memlock(ptr, bytes);
|
crypto_memlock(ptr, bytes);
|
||||||
@ -101,15 +99,15 @@ static uint8_t *crypto_malloc(const Memory *mem, size_t bytes)
|
|||||||
return ptr;
|
return ptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
non_null(1) nullable(2)
|
nullable(1)
|
||||||
static void crypto_free(const Memory *mem, uint8_t *ptr, size_t bytes)
|
static void crypto_free(uint8_t *ptr, size_t bytes)
|
||||||
{
|
{
|
||||||
if (ptr != nullptr) {
|
if (ptr != nullptr) {
|
||||||
crypto_memzero(ptr, bytes);
|
crypto_memzero(ptr, bytes);
|
||||||
crypto_memunlock(ptr, bytes);
|
crypto_memunlock(ptr, bytes);
|
||||||
}
|
}
|
||||||
|
|
||||||
mem_delete(mem, ptr);
|
free(ptr);
|
||||||
}
|
}
|
||||||
#endif /* !defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION) */
|
#endif /* !defined(FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION) */
|
||||||
|
|
||||||
@ -242,8 +240,7 @@ int32_t encrypt_precompute(const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|||||||
#endif /* FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION */
|
#endif /* FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION */
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t encrypt_data_symmetric(const Memory *mem,
|
int32_t encrypt_data_symmetric(const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
||||||
const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *plain, size_t length, uint8_t *encrypted)
|
const uint8_t *plain, size_t length, uint8_t *encrypted)
|
||||||
{
|
{
|
||||||
@ -261,12 +258,12 @@ int32_t encrypt_data_symmetric(const Memory *mem,
|
|||||||
const size_t size_temp_plain = length + crypto_box_ZEROBYTES;
|
const size_t size_temp_plain = length + crypto_box_ZEROBYTES;
|
||||||
const size_t size_temp_encrypted = length + crypto_box_MACBYTES + crypto_box_BOXZEROBYTES;
|
const size_t size_temp_encrypted = length + crypto_box_MACBYTES + crypto_box_BOXZEROBYTES;
|
||||||
|
|
||||||
uint8_t *temp_plain = crypto_malloc(mem, size_temp_plain);
|
uint8_t *temp_plain = crypto_malloc(size_temp_plain);
|
||||||
uint8_t *temp_encrypted = crypto_malloc(mem, size_temp_encrypted);
|
uint8_t *temp_encrypted = crypto_malloc(size_temp_encrypted);
|
||||||
|
|
||||||
if (temp_plain == nullptr || temp_encrypted == nullptr) {
|
if (temp_plain == nullptr || temp_encrypted == nullptr) {
|
||||||
crypto_free(mem, temp_plain, size_temp_plain);
|
crypto_free(temp_plain, size_temp_plain);
|
||||||
crypto_free(mem, temp_encrypted, size_temp_encrypted);
|
crypto_free(temp_encrypted, size_temp_encrypted);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -281,23 +278,22 @@ int32_t encrypt_data_symmetric(const Memory *mem,
|
|||||||
|
|
||||||
if (crypto_box_afternm(temp_encrypted, temp_plain, length + crypto_box_ZEROBYTES, nonce,
|
if (crypto_box_afternm(temp_encrypted, temp_plain, length + crypto_box_ZEROBYTES, nonce,
|
||||||
shared_key) != 0) {
|
shared_key) != 0) {
|
||||||
crypto_free(mem, temp_plain, size_temp_plain);
|
crypto_free(temp_plain, size_temp_plain);
|
||||||
crypto_free(mem, temp_encrypted, size_temp_encrypted);
|
crypto_free(temp_encrypted, size_temp_encrypted);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Unpad the encrypted message.
|
// Unpad the encrypted message.
|
||||||
memcpy(encrypted, temp_encrypted + crypto_box_BOXZEROBYTES, length + crypto_box_MACBYTES);
|
memcpy(encrypted, temp_encrypted + crypto_box_BOXZEROBYTES, length + crypto_box_MACBYTES);
|
||||||
|
|
||||||
crypto_free(mem, temp_plain, size_temp_plain);
|
crypto_free(temp_plain, size_temp_plain);
|
||||||
crypto_free(mem, temp_encrypted, size_temp_encrypted);
|
crypto_free(temp_encrypted, size_temp_encrypted);
|
||||||
#endif /* FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION */
|
#endif /* FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION */
|
||||||
assert(length < INT32_MAX - crypto_box_MACBYTES);
|
assert(length < INT32_MAX - crypto_box_MACBYTES);
|
||||||
return (int32_t)(length + crypto_box_MACBYTES);
|
return (int32_t)(length + crypto_box_MACBYTES);
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t decrypt_data_symmetric(const Memory *mem,
|
int32_t decrypt_data_symmetric(const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
||||||
const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *encrypted, size_t length, uint8_t *plain)
|
const uint8_t *encrypted, size_t length, uint8_t *plain)
|
||||||
{
|
{
|
||||||
@ -314,12 +310,12 @@ int32_t decrypt_data_symmetric(const Memory *mem,
|
|||||||
const size_t size_temp_plain = length + crypto_box_ZEROBYTES;
|
const size_t size_temp_plain = length + crypto_box_ZEROBYTES;
|
||||||
const size_t size_temp_encrypted = length + crypto_box_BOXZEROBYTES;
|
const size_t size_temp_encrypted = length + crypto_box_BOXZEROBYTES;
|
||||||
|
|
||||||
uint8_t *temp_plain = crypto_malloc(mem, size_temp_plain);
|
uint8_t *temp_plain = crypto_malloc(size_temp_plain);
|
||||||
uint8_t *temp_encrypted = crypto_malloc(mem, size_temp_encrypted);
|
uint8_t *temp_encrypted = crypto_malloc(size_temp_encrypted);
|
||||||
|
|
||||||
if (temp_plain == nullptr || temp_encrypted == nullptr) {
|
if (temp_plain == nullptr || temp_encrypted == nullptr) {
|
||||||
crypto_free(mem, temp_plain, size_temp_plain);
|
crypto_free(temp_plain, size_temp_plain);
|
||||||
crypto_free(mem, temp_encrypted, size_temp_encrypted);
|
crypto_free(temp_encrypted, size_temp_encrypted);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -334,23 +330,22 @@ int32_t decrypt_data_symmetric(const Memory *mem,
|
|||||||
|
|
||||||
if (crypto_box_open_afternm(temp_plain, temp_encrypted, length + crypto_box_BOXZEROBYTES, nonce,
|
if (crypto_box_open_afternm(temp_plain, temp_encrypted, length + crypto_box_BOXZEROBYTES, nonce,
|
||||||
shared_key) != 0) {
|
shared_key) != 0) {
|
||||||
crypto_free(mem, temp_plain, size_temp_plain);
|
crypto_free(temp_plain, size_temp_plain);
|
||||||
crypto_free(mem, temp_encrypted, size_temp_encrypted);
|
crypto_free(temp_encrypted, size_temp_encrypted);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
memcpy(plain, temp_plain + crypto_box_ZEROBYTES, length - crypto_box_MACBYTES);
|
memcpy(plain, temp_plain + crypto_box_ZEROBYTES, length - crypto_box_MACBYTES);
|
||||||
|
|
||||||
crypto_free(mem, temp_plain, size_temp_plain);
|
crypto_free(temp_plain, size_temp_plain);
|
||||||
crypto_free(mem, temp_encrypted, size_temp_encrypted);
|
crypto_free(temp_encrypted, size_temp_encrypted);
|
||||||
#endif /* FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION */
|
#endif /* FUZZING_BUILD_MODE_UNSAFE_FOR_PRODUCTION */
|
||||||
assert(length > crypto_box_MACBYTES);
|
assert(length > crypto_box_MACBYTES);
|
||||||
assert(length < INT32_MAX);
|
assert(length < INT32_MAX);
|
||||||
return (int32_t)(length - crypto_box_MACBYTES);
|
return (int32_t)(length - crypto_box_MACBYTES);
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t encrypt_data(const Memory *mem,
|
int32_t encrypt_data(const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
||||||
const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|
||||||
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *plain, size_t length, uint8_t *encrypted)
|
const uint8_t *plain, size_t length, uint8_t *encrypted)
|
||||||
@ -361,13 +356,12 @@ int32_t encrypt_data(const Memory *mem,
|
|||||||
|
|
||||||
uint8_t k[crypto_box_BEFORENMBYTES];
|
uint8_t k[crypto_box_BEFORENMBYTES];
|
||||||
encrypt_precompute(public_key, secret_key, k);
|
encrypt_precompute(public_key, secret_key, k);
|
||||||
const int ret = encrypt_data_symmetric(mem, k, nonce, plain, length, encrypted);
|
const int ret = encrypt_data_symmetric(k, nonce, plain, length, encrypted);
|
||||||
crypto_memzero(k, sizeof(k));
|
crypto_memzero(k, sizeof(k));
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t decrypt_data(const Memory *mem,
|
int32_t decrypt_data(const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
||||||
const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|
||||||
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *encrypted, size_t length, uint8_t *plain)
|
const uint8_t *encrypted, size_t length, uint8_t *plain)
|
||||||
@ -378,7 +372,7 @@ int32_t decrypt_data(const Memory *mem,
|
|||||||
|
|
||||||
uint8_t k[crypto_box_BEFORENMBYTES];
|
uint8_t k[crypto_box_BEFORENMBYTES];
|
||||||
encrypt_precompute(public_key, secret_key, k);
|
encrypt_precompute(public_key, secret_key, k);
|
||||||
const int ret = decrypt_data_symmetric(mem, k, nonce, encrypted, length, plain);
|
const int ret = decrypt_data_symmetric(k, nonce, encrypted, length, plain);
|
||||||
crypto_memzero(k, sizeof(k));
|
crypto_memzero(k, sizeof(k));
|
||||||
return ret;
|
return ret;
|
||||||
}
|
}
|
||||||
|
13
external/toxcore/c-toxcore/toxcore/crypto_core.h
vendored
13
external/toxcore/c-toxcore/toxcore/crypto_core.h
vendored
@ -16,7 +16,6 @@
|
|||||||
#include <stdint.h>
|
#include <stdint.h>
|
||||||
|
|
||||||
#include "attributes.h"
|
#include "attributes.h"
|
||||||
#include "mem.h"
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
@ -387,8 +386,7 @@ void crypto_derive_public_key(uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|||||||
* @return length of encrypted data if everything was fine.
|
* @return length of encrypted data if everything was fine.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int32_t encrypt_data(const Memory *mem,
|
int32_t encrypt_data(const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
||||||
const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|
||||||
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *plain, size_t length, uint8_t *encrypted);
|
const uint8_t *plain, size_t length, uint8_t *encrypted);
|
||||||
@ -405,8 +403,7 @@ int32_t encrypt_data(const Memory *mem,
|
|||||||
* @return length of plain text data if everything was fine.
|
* @return length of plain text data if everything was fine.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int32_t decrypt_data(const Memory *mem,
|
int32_t decrypt_data(const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
||||||
const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|
||||||
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
const uint8_t secret_key[CRYPTO_SECRET_KEY_SIZE],
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *encrypted, size_t length, uint8_t *plain);
|
const uint8_t *encrypted, size_t length, uint8_t *plain);
|
||||||
@ -434,8 +431,7 @@ int32_t encrypt_precompute(const uint8_t public_key[CRYPTO_PUBLIC_KEY_SIZE],
|
|||||||
* @return length of encrypted data if everything was fine.
|
* @return length of encrypted data if everything was fine.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int32_t encrypt_data_symmetric(const Memory *mem,
|
int32_t encrypt_data_symmetric(const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
||||||
const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *plain, size_t length, uint8_t *encrypted);
|
const uint8_t *plain, size_t length, uint8_t *encrypted);
|
||||||
|
|
||||||
@ -450,8 +446,7 @@ int32_t encrypt_data_symmetric(const Memory *mem,
|
|||||||
* @return length of plain data if everything was fine.
|
* @return length of plain data if everything was fine.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int32_t decrypt_data_symmetric(const Memory *mem,
|
int32_t decrypt_data_symmetric(const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
||||||
const uint8_t shared_key[CRYPTO_SHARED_KEY_SIZE],
|
|
||||||
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
const uint8_t nonce[CRYPTO_NONCE_SIZE],
|
||||||
const uint8_t *encrypted, size_t length, uint8_t *plain);
|
const uint8_t *encrypted, size_t length, uint8_t *plain);
|
||||||
|
|
||||||
|
@ -7,7 +7,6 @@
|
|||||||
#include <vector>
|
#include <vector>
|
||||||
|
|
||||||
#include "crypto_core_test_util.hh"
|
#include "crypto_core_test_util.hh"
|
||||||
#include "mem_test_util.hh"
|
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
@ -18,38 +17,8 @@ using SecretKey = std::array<uint8_t, CRYPTO_SECRET_KEY_SIZE>;
|
|||||||
using Signature = std::array<uint8_t, CRYPTO_SIGNATURE_SIZE>;
|
using Signature = std::array<uint8_t, CRYPTO_SIGNATURE_SIZE>;
|
||||||
using Nonce = std::array<uint8_t, CRYPTO_NONCE_SIZE>;
|
using Nonce = std::array<uint8_t, CRYPTO_NONCE_SIZE>;
|
||||||
|
|
||||||
TEST(PkEqual, TwoRandomIdsAreNotEqual)
|
|
||||||
{
|
|
||||||
std::mt19937 rng;
|
|
||||||
std::uniform_int_distribution<unsigned short> dist{0, UINT8_MAX};
|
|
||||||
|
|
||||||
uint8_t pk1[CRYPTO_PUBLIC_KEY_SIZE];
|
|
||||||
uint8_t pk2[CRYPTO_PUBLIC_KEY_SIZE];
|
|
||||||
|
|
||||||
std::generate(std::begin(pk1), std::end(pk1), [&]() { return dist(rng); });
|
|
||||||
std::generate(std::begin(pk2), std::end(pk2), [&]() { return dist(rng); });
|
|
||||||
|
|
||||||
EXPECT_FALSE(pk_equal(pk1, pk2));
|
|
||||||
}
|
|
||||||
|
|
||||||
TEST(PkEqual, IdCopyMakesKeysEqual)
|
|
||||||
{
|
|
||||||
std::mt19937 rng;
|
|
||||||
std::uniform_int_distribution<unsigned short> dist{0, UINT8_MAX};
|
|
||||||
|
|
||||||
uint8_t pk1[CRYPTO_PUBLIC_KEY_SIZE];
|
|
||||||
uint8_t pk2[CRYPTO_PUBLIC_KEY_SIZE] = {0};
|
|
||||||
|
|
||||||
std::generate(std::begin(pk1), std::end(pk1), [&]() { return dist(rng); });
|
|
||||||
|
|
||||||
pk_copy(pk2, pk1);
|
|
||||||
|
|
||||||
EXPECT_TRUE(pk_equal(pk1, pk2));
|
|
||||||
}
|
|
||||||
|
|
||||||
TEST(CryptoCore, EncryptLargeData)
|
TEST(CryptoCore, EncryptLargeData)
|
||||||
{
|
{
|
||||||
Test_Memory mem;
|
|
||||||
Test_Random rng;
|
Test_Random rng;
|
||||||
|
|
||||||
Nonce nonce{};
|
Nonce nonce{};
|
||||||
@ -61,8 +30,7 @@ TEST(CryptoCore, EncryptLargeData)
|
|||||||
std::vector<uint8_t> plain(100 * 1024 * 1024);
|
std::vector<uint8_t> plain(100 * 1024 * 1024);
|
||||||
std::vector<uint8_t> encrypted(plain.size() + CRYPTO_MAC_SIZE);
|
std::vector<uint8_t> encrypted(plain.size() + CRYPTO_MAC_SIZE);
|
||||||
|
|
||||||
encrypt_data(
|
encrypt_data(pk.data(), sk.data(), nonce.data(), plain.data(), plain.size(), encrypted.data());
|
||||||
mem, pk.data(), sk.data(), nonce.data(), plain.data(), plain.size(), encrypted.data());
|
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST(CryptoCore, IncrementNonce)
|
TEST(CryptoCore, IncrementNonce)
|
||||||
|
71
external/toxcore/c-toxcore/toxcore/group.c
vendored
71
external/toxcore/c-toxcore/toxcore/group.c
vendored
@ -9,7 +9,7 @@
|
|||||||
#include "group.h"
|
#include "group.h"
|
||||||
|
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
#include <stdlib.h> // calloc, free
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#include "DHT.h"
|
#include "DHT.h"
|
||||||
@ -20,11 +20,9 @@
|
|||||||
#include "friend_connection.h"
|
#include "friend_connection.h"
|
||||||
#include "group_common.h"
|
#include "group_common.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
#include "net_crypto.h"
|
#include "net_crypto.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "sort.h"
|
|
||||||
#include "state.h"
|
#include "state.h"
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
@ -959,75 +957,24 @@ static bool delpeer(Group_Chats *g_c, uint32_t groupnumber, int peer_index, void
|
|||||||
|
|
||||||
/** Order peers with friends first and with more recently active earlier */
|
/** Order peers with friends first and with more recently active earlier */
|
||||||
non_null()
|
non_null()
|
||||||
static bool group_peer_less_handler(const void *object, const void *a, const void *b)
|
static int cmp_frozen(const void *a, const void *b)
|
||||||
{
|
{
|
||||||
const Group_Peer *pa = (const Group_Peer *)a;
|
const Group_Peer *pa = (const Group_Peer *)a;
|
||||||
const Group_Peer *pb = (const Group_Peer *)b;
|
const Group_Peer *pb = (const Group_Peer *)b;
|
||||||
|
|
||||||
if (((pa->is_friend ? 1 : 0) ^ (pb->is_friend ? 1 : 0)) != 0) {
|
if (pa->is_friend ^ pb->is_friend) {
|
||||||
return pa->is_friend;
|
return pa->is_friend ? -1 : 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
return cmp_uint(pb->last_active, pa->last_active) < 0;
|
return cmp_uint(pb->last_active, pa->last_active);
|
||||||
}
|
}
|
||||||
|
|
||||||
non_null()
|
|
||||||
static const void *group_peer_get_handler(const void *arr, uint32_t index)
|
|
||||||
{
|
|
||||||
const Group_Peer *entries = (const Group_Peer *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void group_peer_set_handler(void *arr, uint32_t index, const void *val)
|
|
||||||
{
|
|
||||||
Group_Peer *entries = (Group_Peer *)arr;
|
|
||||||
const Group_Peer *entry = (const Group_Peer *)val;
|
|
||||||
entries[index] = *entry;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *group_peer_subarr_handler(void *arr, uint32_t index, uint32_t size)
|
|
||||||
{
|
|
||||||
Group_Peer *entries = (Group_Peer *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *group_peer_alloc_handler(const void *object, uint32_t size)
|
|
||||||
{
|
|
||||||
const Memory *mem = (const Memory *)object;
|
|
||||||
Group_Peer *tmp = (Group_Peer *)mem_valloc(mem, size, sizeof(Group_Peer));
|
|
||||||
|
|
||||||
if (tmp == nullptr) {
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return tmp;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void group_peer_delete_handler(const void *object, void *arr, uint32_t size)
|
|
||||||
{
|
|
||||||
const Memory *mem = (const Memory *)object;
|
|
||||||
mem_delete(mem, arr);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const Sort_Funcs group_peer_cmp_funcs = {
|
|
||||||
group_peer_less_handler,
|
|
||||||
group_peer_get_handler,
|
|
||||||
group_peer_set_handler,
|
|
||||||
group_peer_subarr_handler,
|
|
||||||
group_peer_alloc_handler,
|
|
||||||
group_peer_delete_handler,
|
|
||||||
};
|
|
||||||
|
|
||||||
/** @brief Delete frozen peers as necessary to ensure at most `g->maxfrozen` remain.
|
/** @brief Delete frozen peers as necessary to ensure at most `g->maxfrozen` remain.
|
||||||
*
|
*
|
||||||
* @retval true if any frozen peers are removed.
|
* @retval true if any frozen peers are removed.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
static bool delete_old_frozen(Group_c *g, const Memory *mem)
|
static bool delete_old_frozen(Group_c *g)
|
||||||
{
|
{
|
||||||
if (g->numfrozen <= g->maxfrozen) {
|
if (g->numfrozen <= g->maxfrozen) {
|
||||||
return false;
|
return false;
|
||||||
@ -1040,7 +987,7 @@ static bool delete_old_frozen(Group_c *g, const Memory *mem)
|
|||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
merge_sort(g->frozen, g->numfrozen, mem, &group_peer_cmp_funcs);
|
qsort(g->frozen, g->numfrozen, sizeof(Group_Peer), cmp_frozen);
|
||||||
|
|
||||||
Group_Peer *temp = (Group_Peer *)realloc(g->frozen, g->maxfrozen * sizeof(Group_Peer));
|
Group_Peer *temp = (Group_Peer *)realloc(g->frozen, g->maxfrozen * sizeof(Group_Peer));
|
||||||
|
|
||||||
@ -1085,7 +1032,7 @@ static bool freeze_peer(Group_Chats *g_c, uint32_t groupnumber, int peer_index,
|
|||||||
|
|
||||||
++g->numfrozen;
|
++g->numfrozen;
|
||||||
|
|
||||||
delete_old_frozen(g, g_c->m->mem);
|
delete_old_frozen(g);
|
||||||
|
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
@ -1572,7 +1519,7 @@ int group_set_max_frozen(const Group_Chats *g_c, uint32_t groupnumber, uint32_t
|
|||||||
}
|
}
|
||||||
|
|
||||||
g->maxfrozen = maxfrozen;
|
g->maxfrozen = maxfrozen;
|
||||||
delete_old_frozen(g, g_c->m->mem);
|
delete_old_frozen(g);
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
29
external/toxcore/c-toxcore/toxcore/group_chats.c
vendored
29
external/toxcore/c-toxcore/toxcore/group_chats.c
vendored
@ -30,7 +30,6 @@
|
|||||||
#include "group_moderation.h"
|
#include "group_moderation.h"
|
||||||
#include "group_pack.h"
|
#include "group_pack.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
#include "net_crypto.h"
|
#include "net_crypto.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
@ -1474,8 +1473,8 @@ static bool sign_gc_shared_state(GC_Chat *chat)
|
|||||||
* Return -2 on decryption failure.
|
* Return -2 on decryption failure.
|
||||||
* Return -3 if plaintext payload length is invalid.
|
* Return -3 if plaintext payload length is invalid.
|
||||||
*/
|
*/
|
||||||
non_null(1, 2, 3, 4, 6, 7) nullable(5)
|
non_null(1, 2, 3, 5, 6) nullable(4)
|
||||||
static int group_packet_unwrap(const Logger *log, const Memory *mem, const GC_Connection *gconn, uint8_t *data, uint64_t *message_id,
|
static int group_packet_unwrap(const Logger *log, const GC_Connection *gconn, uint8_t *data, uint64_t *message_id,
|
||||||
uint8_t *packet_type, const uint8_t *packet, uint16_t length)
|
uint8_t *packet_type, const uint8_t *packet, uint16_t length)
|
||||||
{
|
{
|
||||||
assert(data != nullptr);
|
assert(data != nullptr);
|
||||||
@ -1493,7 +1492,7 @@ static int group_packet_unwrap(const Logger *log, const Memory *mem, const GC_Co
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int plain_len = decrypt_data_symmetric(mem, gconn->session_shared_key, packet, packet + CRYPTO_NONCE_SIZE,
|
int plain_len = decrypt_data_symmetric(gconn->session_shared_key, packet, packet + CRYPTO_NONCE_SIZE,
|
||||||
length - CRYPTO_NONCE_SIZE, plain);
|
length - CRYPTO_NONCE_SIZE, plain);
|
||||||
|
|
||||||
if (plain_len <= 0) {
|
if (plain_len <= 0) {
|
||||||
@ -1534,7 +1533,7 @@ static int group_packet_unwrap(const Logger *log, const Memory *mem, const GC_Co
|
|||||||
}
|
}
|
||||||
|
|
||||||
int group_packet_wrap(
|
int group_packet_wrap(
|
||||||
const Logger *log, const Memory *mem, const Random *rng, const uint8_t *self_pk, const uint8_t *shared_key, uint8_t *packet,
|
const Logger *log, const Random *rng, const uint8_t *self_pk, const uint8_t *shared_key, uint8_t *packet,
|
||||||
uint16_t packet_size, const uint8_t *data, uint16_t length, uint64_t message_id,
|
uint16_t packet_size, const uint8_t *data, uint16_t length, uint64_t message_id,
|
||||||
uint8_t gp_packet_type, Net_Packet_Type net_packet_type)
|
uint8_t gp_packet_type, Net_Packet_Type net_packet_type)
|
||||||
{
|
{
|
||||||
@ -1589,7 +1588,7 @@ int group_packet_wrap(
|
|||||||
return -2;
|
return -2;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int enc_len = encrypt_data_symmetric(mem, shared_key, nonce, plain, plain_len, encrypt);
|
const int enc_len = encrypt_data_symmetric(shared_key, nonce, plain, plain_len, encrypt);
|
||||||
|
|
||||||
free(plain);
|
free(plain);
|
||||||
|
|
||||||
@ -1635,7 +1634,7 @@ static bool send_lossy_group_packet(const GC_Chat *chat, const GC_Connection *gc
|
|||||||
}
|
}
|
||||||
|
|
||||||
const int len = group_packet_wrap(
|
const int len = group_packet_wrap(
|
||||||
chat->log, chat->mem, chat->rng, chat->self_public_key.enc, gconn->session_shared_key, packet,
|
chat->log, chat->rng, chat->self_public_key.enc, gconn->session_shared_key, packet,
|
||||||
packet_size, data, length, 0, packet_type, NET_PACKET_GC_LOSSY);
|
packet_size, data, length, 0, packet_type, NET_PACKET_GC_LOSSY);
|
||||||
|
|
||||||
if (len < 0) {
|
if (len < 0) {
|
||||||
@ -5509,7 +5508,7 @@ static int handle_gc_broadcast(const GC_Session *c, GC_Chat *chat, uint32_t peer
|
|||||||
* Return -2 if decryption fails.
|
* Return -2 if decryption fails.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
static int unwrap_group_handshake_packet(const Logger *log, const Memory *mem, const uint8_t *self_sk, const uint8_t *sender_pk,
|
static int unwrap_group_handshake_packet(const Logger *log, const uint8_t *self_sk, const uint8_t *sender_pk,
|
||||||
uint8_t *plain, size_t plain_size, const uint8_t *packet, uint16_t length)
|
uint8_t *plain, size_t plain_size, const uint8_t *packet, uint16_t length)
|
||||||
{
|
{
|
||||||
if (length <= CRYPTO_NONCE_SIZE) {
|
if (length <= CRYPTO_NONCE_SIZE) {
|
||||||
@ -5517,7 +5516,7 @@ static int unwrap_group_handshake_packet(const Logger *log, const Memory *mem, c
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int plain_len = decrypt_data(mem, sender_pk, self_sk, packet, packet + CRYPTO_NONCE_SIZE,
|
const int plain_len = decrypt_data(sender_pk, self_sk, packet, packet + CRYPTO_NONCE_SIZE,
|
||||||
length - CRYPTO_NONCE_SIZE, plain);
|
length - CRYPTO_NONCE_SIZE, plain);
|
||||||
|
|
||||||
if (plain_len < 0 || (uint32_t)plain_len != plain_size) {
|
if (plain_len < 0 || (uint32_t)plain_len != plain_size) {
|
||||||
@ -5540,7 +5539,7 @@ static int unwrap_group_handshake_packet(const Logger *log, const Memory *mem, c
|
|||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
static int wrap_group_handshake_packet(
|
static int wrap_group_handshake_packet(
|
||||||
const Logger *log, const Memory *mem, const Random *rng, const uint8_t *self_pk, const uint8_t *self_sk,
|
const Logger *log, const Random *rng, const uint8_t *self_pk, const uint8_t *self_sk,
|
||||||
const uint8_t *target_pk, uint8_t *packet, uint32_t packet_size,
|
const uint8_t *target_pk, uint8_t *packet, uint32_t packet_size,
|
||||||
const uint8_t *data, uint16_t length)
|
const uint8_t *data, uint16_t length)
|
||||||
{
|
{
|
||||||
@ -5559,7 +5558,7 @@ static int wrap_group_handshake_packet(
|
|||||||
return -2;
|
return -2;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int enc_len = encrypt_data(mem, target_pk, self_sk, nonce, data, length, encrypt);
|
const int enc_len = encrypt_data(target_pk, self_sk, nonce, data, length, encrypt);
|
||||||
|
|
||||||
if (enc_len < 0 || (size_t)enc_len != encrypt_buf_size) {
|
if (enc_len < 0 || (size_t)enc_len != encrypt_buf_size) {
|
||||||
LOGGER_ERROR(log, "Failed to encrypt group handshake packet (len: %d)", enc_len);
|
LOGGER_ERROR(log, "Failed to encrypt group handshake packet (len: %d)", enc_len);
|
||||||
@ -5623,7 +5622,7 @@ static int make_gc_handshake_packet(const GC_Chat *chat, const GC_Connection *gc
|
|||||||
}
|
}
|
||||||
|
|
||||||
const int enc_len = wrap_group_handshake_packet(
|
const int enc_len = wrap_group_handshake_packet(
|
||||||
chat->log, chat->mem, chat->rng, chat->self_public_key.enc, chat->self_secret_key.enc,
|
chat->log, chat->rng, chat->self_public_key.enc, chat->self_secret_key.enc,
|
||||||
gconn->addr.public_key.enc, packet, (uint16_t)packet_size, data, length);
|
gconn->addr.public_key.enc, packet, (uint16_t)packet_size, data, length);
|
||||||
|
|
||||||
if (enc_len != GC_MIN_ENCRYPTED_HS_PAYLOAD_SIZE + nodes_size) {
|
if (enc_len != GC_MIN_ENCRYPTED_HS_PAYLOAD_SIZE + nodes_size) {
|
||||||
@ -5952,7 +5951,7 @@ static int handle_gc_handshake_packet(GC_Chat *chat, const uint8_t *sender_pk, c
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int plain_len = unwrap_group_handshake_packet(chat->log, chat->mem, chat->self_secret_key.enc, sender_pk, data,
|
const int plain_len = unwrap_group_handshake_packet(chat->log, chat->self_secret_key.enc, sender_pk, data,
|
||||||
data_buf_size, packet, length);
|
data_buf_size, packet, length);
|
||||||
|
|
||||||
if (plain_len < GC_MIN_HS_PACKET_PAYLOAD_SIZE) {
|
if (plain_len < GC_MIN_HS_PACKET_PAYLOAD_SIZE) {
|
||||||
@ -6182,7 +6181,7 @@ static bool handle_gc_lossless_packet(const GC_Session *c, GC_Chat *chat, const
|
|||||||
uint8_t packet_type;
|
uint8_t packet_type;
|
||||||
uint64_t message_id;
|
uint64_t message_id;
|
||||||
|
|
||||||
const int len = group_packet_unwrap(chat->log, chat->mem, gconn, data, &message_id, &packet_type, packet, length);
|
const int len = group_packet_unwrap(chat->log, gconn, data, &message_id, &packet_type, packet, length);
|
||||||
|
|
||||||
if (len < 0) {
|
if (len < 0) {
|
||||||
Ip_Ntoa ip_str;
|
Ip_Ntoa ip_str;
|
||||||
@ -6335,7 +6334,7 @@ static bool handle_gc_lossy_packet(const GC_Session *c, GC_Chat *chat, const uin
|
|||||||
|
|
||||||
uint8_t packet_type;
|
uint8_t packet_type;
|
||||||
|
|
||||||
const int len = group_packet_unwrap(chat->log, chat->mem, gconn, data, nullptr, &packet_type, packet, length);
|
const int len = group_packet_unwrap(chat->log, gconn, data, nullptr, &packet_type, packet, length);
|
||||||
|
|
||||||
if (len <= 0) {
|
if (len <= 0) {
|
||||||
Ip_Ntoa ip_str;
|
Ip_Ntoa ip_str;
|
||||||
|
@ -22,7 +22,6 @@
|
|||||||
#include "group_common.h"
|
#include "group_common.h"
|
||||||
#include "group_connection.h"
|
#include "group_connection.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
|
|
||||||
#define GC_PING_TIMEOUT 12
|
#define GC_PING_TIMEOUT 12
|
||||||
@ -142,9 +141,9 @@ int get_peer_number_of_enc_pk(const GC_Chat *chat, const uint8_t *public_enc_key
|
|||||||
* Return -2 if malloc fails.
|
* Return -2 if malloc fails.
|
||||||
* Return -3 if encryption fails.
|
* Return -3 if encryption fails.
|
||||||
*/
|
*/
|
||||||
non_null(1, 2, 3, 4, 5, 6) nullable(8)
|
non_null(1, 2, 3, 4, 5) nullable(7)
|
||||||
int group_packet_wrap(
|
int group_packet_wrap(
|
||||||
const Logger *log, const Memory *mem, const Random *rng, const uint8_t *self_pk, const uint8_t *shared_key, uint8_t *packet,
|
const Logger *log, const Random *rng, const uint8_t *self_pk, const uint8_t *shared_key, uint8_t *packet,
|
||||||
uint16_t packet_size, const uint8_t *data, uint16_t length, uint64_t message_id,
|
uint16_t packet_size, const uint8_t *data, uint16_t length, uint64_t message_id,
|
||||||
uint8_t gp_packet_type, Net_Packet_Type net_packet_type);
|
uint8_t gp_packet_type, Net_Packet_Type net_packet_type);
|
||||||
|
|
||||||
|
@ -52,7 +52,7 @@
|
|||||||
#define MAX_GC_PACKET_SIZE (MAX_GC_PACKET_CHUNK_SIZE * 100)
|
#define MAX_GC_PACKET_SIZE (MAX_GC_PACKET_CHUNK_SIZE * 100)
|
||||||
|
|
||||||
/* Max number of messages to store in the send/recv arrays */
|
/* Max number of messages to store in the send/recv arrays */
|
||||||
#define GCC_BUFFER_SIZE 2048
|
#define GCC_BUFFER_SIZE 8192
|
||||||
|
|
||||||
/** Self UDP status. Must correspond to return values from `ipport_self_copy()`. */
|
/** Self UDP status. Must correspond to return values from `ipport_self_copy()`. */
|
||||||
typedef enum Self_UDP_Status {
|
typedef enum Self_UDP_Status {
|
||||||
|
@ -629,7 +629,7 @@ int gcc_encrypt_and_send_lossless_packet(const GC_Chat *chat, const GC_Connectio
|
|||||||
}
|
}
|
||||||
|
|
||||||
const int enc_len = group_packet_wrap(
|
const int enc_len = group_packet_wrap(
|
||||||
chat->log, chat->mem, chat->rng, chat->self_public_key.enc, gconn->session_shared_key, packet,
|
chat->log, chat->rng, chat->self_public_key.enc, gconn->session_shared_key, packet,
|
||||||
packet_size, data, length, message_id, packet_type, NET_PACKET_GC_LOSSLESS);
|
packet_size, data, length, message_id, packet_type, NET_PACKET_GC_LOSSLESS);
|
||||||
|
|
||||||
if (enc_len < 0) {
|
if (enc_len < 0) {
|
||||||
|
@ -14,7 +14,6 @@
|
|||||||
#include "crypto_core.h"
|
#include "crypto_core.h"
|
||||||
#include "group_announce.h"
|
#include "group_announce.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "onion_announce.h"
|
#include "onion_announce.h"
|
||||||
@ -77,7 +76,7 @@ void gca_onion_init(GC_Announces_List *group_announce, Onion_Announce *onion_a)
|
|||||||
}
|
}
|
||||||
|
|
||||||
int create_gca_announce_request(
|
int create_gca_announce_request(
|
||||||
const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
||||||
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id,
|
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id,
|
||||||
const uint8_t *client_id, const uint8_t *data_public_key, uint64_t sendback_data,
|
const uint8_t *client_id, const uint8_t *data_public_key, uint64_t sendback_data,
|
||||||
const uint8_t *gc_data, uint16_t gc_data_length)
|
const uint8_t *gc_data, uint16_t gc_data_length)
|
||||||
@ -109,7 +108,7 @@ int create_gca_announce_request(
|
|||||||
random_nonce(rng, packet + 1);
|
random_nonce(rng, packet + 1);
|
||||||
memcpy(packet + 1 + CRYPTO_NONCE_SIZE, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(packet + 1 + CRYPTO_NONCE_SIZE, public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
const int len = encrypt_data(mem, dest_client_id, secret_key, packet + 1, plain,
|
const int len = encrypt_data(dest_client_id, secret_key, packet + 1, plain,
|
||||||
encrypted_size, packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
encrypted_size, packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
const uint32_t full_length = (uint32_t)len + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE;
|
const uint32_t full_length = (uint32_t)len + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE;
|
||||||
|
@ -9,7 +9,6 @@
|
|||||||
#include "attributes.h"
|
#include "attributes.h"
|
||||||
#include "crypto_core.h"
|
#include "crypto_core.h"
|
||||||
#include "group_announce.h"
|
#include "group_announce.h"
|
||||||
#include "mem.h"
|
|
||||||
#include "onion_announce.h"
|
#include "onion_announce.h"
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
@ -17,7 +16,7 @@ void gca_onion_init(GC_Announces_List *group_announce, Onion_Announce *onion_a);
|
|||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
int create_gca_announce_request(
|
int create_gca_announce_request(
|
||||||
const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
||||||
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id,
|
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id,
|
||||||
const uint8_t *client_id, const uint8_t *data_public_key, uint64_t sendback_data,
|
const uint8_t *client_id, const uint8_t *data_public_key, uint64_t sendback_data,
|
||||||
const uint8_t *gc_data, uint16_t gc_data_length);
|
const uint8_t *gc_data, uint16_t gc_data_length);
|
||||||
|
50
external/toxcore/c-toxcore/toxcore/net_crypto.c
vendored
50
external/toxcore/c-toxcore/toxcore/net_crypto.c
vendored
@ -23,6 +23,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
@ -230,7 +231,7 @@ static int create_cookie_request(const Net_Crypto *c, uint8_t *packet, const uin
|
|||||||
packet[0] = NET_PACKET_COOKIE_REQUEST;
|
packet[0] = NET_PACKET_COOKIE_REQUEST;
|
||||||
memcpy(packet + 1, dht_get_self_public_key(c->dht), CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(packet + 1, dht_get_self_public_key(c->dht), CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
memcpy(packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, nonce, CRYPTO_NONCE_SIZE);
|
memcpy(packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, nonce, CRYPTO_NONCE_SIZE);
|
||||||
const int len = encrypt_data_symmetric(c->mem, shared_key, nonce, plain, sizeof(plain),
|
const int len = encrypt_data_symmetric(shared_key, nonce, plain, sizeof(plain),
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != COOKIE_REQUEST_PLAIN_LENGTH + CRYPTO_MAC_SIZE) {
|
if (len != COOKIE_REQUEST_PLAIN_LENGTH + CRYPTO_MAC_SIZE) {
|
||||||
@ -246,7 +247,7 @@ static int create_cookie_request(const Net_Crypto *c, uint8_t *packet, const uin
|
|||||||
* @retval 0 on success.
|
* @retval 0 on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
static int create_cookie(const Memory *mem, const Random *rng, const Mono_Time *mono_time, uint8_t *cookie, const uint8_t *bytes,
|
static int create_cookie(const Random *rng, const Mono_Time *mono_time, uint8_t *cookie, const uint8_t *bytes,
|
||||||
const uint8_t *encryption_key)
|
const uint8_t *encryption_key)
|
||||||
{
|
{
|
||||||
uint8_t contents[COOKIE_CONTENTS_LENGTH];
|
uint8_t contents[COOKIE_CONTENTS_LENGTH];
|
||||||
@ -254,7 +255,7 @@ static int create_cookie(const Memory *mem, const Random *rng, const Mono_Time *
|
|||||||
memcpy(contents, &temp_time, sizeof(temp_time));
|
memcpy(contents, &temp_time, sizeof(temp_time));
|
||||||
memcpy(contents + sizeof(temp_time), bytes, COOKIE_DATA_LENGTH);
|
memcpy(contents + sizeof(temp_time), bytes, COOKIE_DATA_LENGTH);
|
||||||
random_nonce(rng, cookie);
|
random_nonce(rng, cookie);
|
||||||
const int len = encrypt_data_symmetric(mem, encryption_key, cookie, contents, sizeof(contents), cookie + CRYPTO_NONCE_SIZE);
|
const int len = encrypt_data_symmetric(encryption_key, cookie, contents, sizeof(contents), cookie + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != COOKIE_LENGTH - CRYPTO_NONCE_SIZE) {
|
if (len != COOKIE_LENGTH - CRYPTO_NONCE_SIZE) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -269,11 +270,11 @@ static int create_cookie(const Memory *mem, const Random *rng, const Mono_Time *
|
|||||||
* @retval 0 on success.
|
* @retval 0 on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
static int open_cookie(const Memory *mem, const Mono_Time *mono_time, uint8_t *bytes, const uint8_t *cookie,
|
static int open_cookie(const Mono_Time *mono_time, uint8_t *bytes, const uint8_t *cookie,
|
||||||
const uint8_t *encryption_key)
|
const uint8_t *encryption_key)
|
||||||
{
|
{
|
||||||
uint8_t contents[COOKIE_CONTENTS_LENGTH];
|
uint8_t contents[COOKIE_CONTENTS_LENGTH];
|
||||||
const int len = decrypt_data_symmetric(mem, encryption_key, cookie, cookie + CRYPTO_NONCE_SIZE,
|
const int len = decrypt_data_symmetric(encryption_key, cookie, cookie + CRYPTO_NONCE_SIZE,
|
||||||
COOKIE_LENGTH - CRYPTO_NONCE_SIZE, contents);
|
COOKIE_LENGTH - CRYPTO_NONCE_SIZE, contents);
|
||||||
|
|
||||||
if (len != sizeof(contents)) {
|
if (len != sizeof(contents)) {
|
||||||
@ -308,14 +309,14 @@ static int create_cookie_response(const Net_Crypto *c, uint8_t *packet, const ui
|
|||||||
memcpy(cookie_plain + CRYPTO_PUBLIC_KEY_SIZE, dht_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(cookie_plain + CRYPTO_PUBLIC_KEY_SIZE, dht_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
uint8_t plain[COOKIE_LENGTH + sizeof(uint64_t)];
|
uint8_t plain[COOKIE_LENGTH + sizeof(uint64_t)];
|
||||||
|
|
||||||
if (create_cookie(c->mem, c->rng, c->mono_time, plain, cookie_plain, c->secret_symmetric_key) != 0) {
|
if (create_cookie(c->rng, c->mono_time, plain, cookie_plain, c->secret_symmetric_key) != 0) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
memcpy(plain + COOKIE_LENGTH, request_plain + COOKIE_DATA_LENGTH, sizeof(uint64_t));
|
memcpy(plain + COOKIE_LENGTH, request_plain + COOKIE_DATA_LENGTH, sizeof(uint64_t));
|
||||||
packet[0] = NET_PACKET_COOKIE_RESPONSE;
|
packet[0] = NET_PACKET_COOKIE_RESPONSE;
|
||||||
random_nonce(c->rng, packet + 1);
|
random_nonce(c->rng, packet + 1);
|
||||||
const int len = encrypt_data_symmetric(c->mem, shared_key, packet + 1, plain, sizeof(plain), packet + 1 + CRYPTO_NONCE_SIZE);
|
const int len = encrypt_data_symmetric(shared_key, packet + 1, plain, sizeof(plain), packet + 1 + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != COOKIE_RESPONSE_LENGTH - (1 + CRYPTO_NONCE_SIZE)) {
|
if (len != COOKIE_RESPONSE_LENGTH - (1 + CRYPTO_NONCE_SIZE)) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -342,7 +343,7 @@ static int handle_cookie_request(const Net_Crypto *c, uint8_t *request_plain, ui
|
|||||||
memcpy(dht_public_key, packet + 1, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(dht_public_key, packet + 1, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
const uint8_t *tmp_shared_key = dht_get_shared_key_sent(c->dht, dht_public_key);
|
const uint8_t *tmp_shared_key = dht_get_shared_key_sent(c->dht, dht_public_key);
|
||||||
memcpy(shared_key, tmp_shared_key, CRYPTO_SHARED_KEY_SIZE);
|
memcpy(shared_key, tmp_shared_key, CRYPTO_SHARED_KEY_SIZE);
|
||||||
const int len = decrypt_data_symmetric(c->mem, shared_key, packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
const int len = decrypt_data_symmetric(shared_key, packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE, COOKIE_REQUEST_PLAIN_LENGTH + CRYPTO_MAC_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE, COOKIE_REQUEST_PLAIN_LENGTH + CRYPTO_MAC_SIZE,
|
||||||
request_plain);
|
request_plain);
|
||||||
|
|
||||||
@ -439,7 +440,7 @@ static int tcp_oob_handle_cookie_request(const Net_Crypto *c, unsigned int tcp_c
|
|||||||
* @retval COOKIE_LENGTH on success.
|
* @retval COOKIE_LENGTH on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
static int handle_cookie_response(const Memory *mem, uint8_t *cookie, uint64_t *number,
|
static int handle_cookie_response(uint8_t *cookie, uint64_t *number,
|
||||||
const uint8_t *packet, uint16_t length,
|
const uint8_t *packet, uint16_t length,
|
||||||
const uint8_t *shared_key)
|
const uint8_t *shared_key)
|
||||||
{
|
{
|
||||||
@ -448,7 +449,7 @@ static int handle_cookie_response(const Memory *mem, uint8_t *cookie, uint64_t *
|
|||||||
}
|
}
|
||||||
|
|
||||||
uint8_t plain[COOKIE_LENGTH + sizeof(uint64_t)];
|
uint8_t plain[COOKIE_LENGTH + sizeof(uint64_t)];
|
||||||
const int len = decrypt_data_symmetric(mem, shared_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
const int len = decrypt_data_symmetric(shared_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
||||||
length - (1 + CRYPTO_NONCE_SIZE), plain);
|
length - (1 + CRYPTO_NONCE_SIZE), plain);
|
||||||
|
|
||||||
if (len != sizeof(plain)) {
|
if (len != sizeof(plain)) {
|
||||||
@ -481,13 +482,13 @@ static int create_crypto_handshake(const Net_Crypto *c, uint8_t *packet, const u
|
|||||||
memcpy(cookie_plain, peer_real_pk, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(cookie_plain, peer_real_pk, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
memcpy(cookie_plain + CRYPTO_PUBLIC_KEY_SIZE, peer_dht_pubkey, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(cookie_plain + CRYPTO_PUBLIC_KEY_SIZE, peer_dht_pubkey, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (create_cookie(c->mem, c->rng, c->mono_time, plain + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_SHA512_SIZE,
|
if (create_cookie(c->rng, c->mono_time, plain + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_SHA512_SIZE,
|
||||||
cookie_plain, c->secret_symmetric_key) != 0) {
|
cookie_plain, c->secret_symmetric_key) != 0) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
random_nonce(c->rng, packet + 1 + COOKIE_LENGTH);
|
random_nonce(c->rng, packet + 1 + COOKIE_LENGTH);
|
||||||
const int len = encrypt_data(c->mem, peer_real_pk, c->self_secret_key, packet + 1 + COOKIE_LENGTH, plain, sizeof(plain),
|
const int len = encrypt_data(peer_real_pk, c->self_secret_key, packet + 1 + COOKIE_LENGTH, plain, sizeof(plain),
|
||||||
packet + 1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE);
|
packet + 1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != HANDSHAKE_PACKET_LENGTH - (1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE)) {
|
if (len != HANDSHAKE_PACKET_LENGTH - (1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE)) {
|
||||||
@ -528,7 +529,7 @@ static bool handle_crypto_handshake(const Net_Crypto *c, uint8_t *nonce, uint8_t
|
|||||||
|
|
||||||
uint8_t cookie_plain[COOKIE_DATA_LENGTH];
|
uint8_t cookie_plain[COOKIE_DATA_LENGTH];
|
||||||
|
|
||||||
if (open_cookie(c->mem, c->mono_time, cookie_plain, packet + 1, c->secret_symmetric_key) != 0) {
|
if (open_cookie(c->mono_time, cookie_plain, packet + 1, c->secret_symmetric_key) != 0) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -540,7 +541,7 @@ static bool handle_crypto_handshake(const Net_Crypto *c, uint8_t *nonce, uint8_t
|
|||||||
crypto_sha512(cookie_hash, packet + 1, COOKIE_LENGTH);
|
crypto_sha512(cookie_hash, packet + 1, COOKIE_LENGTH);
|
||||||
|
|
||||||
uint8_t plain[CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_SHA512_SIZE + COOKIE_LENGTH];
|
uint8_t plain[CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_SHA512_SIZE + COOKIE_LENGTH];
|
||||||
const int len = decrypt_data(c->mem, cookie_plain, c->self_secret_key, packet + 1 + COOKIE_LENGTH,
|
const int len = decrypt_data(cookie_plain, c->self_secret_key, packet + 1 + COOKIE_LENGTH,
|
||||||
packet + 1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE,
|
packet + 1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE,
|
||||||
HANDSHAKE_PACKET_LENGTH - (1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE), plain);
|
HANDSHAKE_PACKET_LENGTH - (1 + COOKIE_LENGTH + CRYPTO_NONCE_SIZE), plain);
|
||||||
|
|
||||||
@ -1084,7 +1085,7 @@ static int send_data_packet(Net_Crypto *c, int crypt_connection_id, const uint8_
|
|||||||
VLA(uint8_t, packet, packet_size);
|
VLA(uint8_t, packet, packet_size);
|
||||||
packet[0] = NET_PACKET_CRYPTO_DATA;
|
packet[0] = NET_PACKET_CRYPTO_DATA;
|
||||||
memcpy(packet + 1, conn->sent_nonce + (CRYPTO_NONCE_SIZE - sizeof(uint16_t)), sizeof(uint16_t));
|
memcpy(packet + 1, conn->sent_nonce + (CRYPTO_NONCE_SIZE - sizeof(uint16_t)), sizeof(uint16_t));
|
||||||
const int len = encrypt_data_symmetric(c->mem, conn->shared_key, conn->sent_nonce, data, length, packet + 1 + sizeof(uint16_t));
|
const int len = encrypt_data_symmetric(conn->shared_key, conn->sent_nonce, data, length, packet + 1 + sizeof(uint16_t));
|
||||||
|
|
||||||
if (len + 1 + sizeof(uint16_t) != packet_size) {
|
if (len + 1 + sizeof(uint16_t) != packet_size) {
|
||||||
LOGGER_ERROR(c->log, "encryption failed: %d", len);
|
LOGGER_ERROR(c->log, "encryption failed: %d", len);
|
||||||
@ -1255,7 +1256,7 @@ static int handle_data_packet(const Net_Crypto *c, int crypt_connection_id, uint
|
|||||||
net_unpack_u16(packet + 1, &num);
|
net_unpack_u16(packet + 1, &num);
|
||||||
const uint16_t diff = num - num_cur_nonce;
|
const uint16_t diff = num - num_cur_nonce;
|
||||||
increment_nonce_number(nonce, diff);
|
increment_nonce_number(nonce, diff);
|
||||||
const int len = decrypt_data_symmetric(c->mem, conn->shared_key, nonce, packet + 1 + sizeof(uint16_t),
|
const int len = decrypt_data_symmetric(conn->shared_key, nonce, packet + 1 + sizeof(uint16_t),
|
||||||
length - (1 + sizeof(uint16_t)), data);
|
length - (1 + sizeof(uint16_t)), data);
|
||||||
|
|
||||||
if ((unsigned int)len != length - crypto_packet_overhead) {
|
if ((unsigned int)len != length - crypto_packet_overhead) {
|
||||||
@ -1662,7 +1663,7 @@ static int handle_packet_cookie_response(Net_Crypto *c, int crypt_connection_id,
|
|||||||
uint8_t cookie[COOKIE_LENGTH];
|
uint8_t cookie[COOKIE_LENGTH];
|
||||||
uint64_t number;
|
uint64_t number;
|
||||||
|
|
||||||
if (handle_cookie_response(c->mem, cookie, &number, packet, length, conn->shared_key) != sizeof(cookie)) {
|
if (handle_cookie_response(cookie, &number, packet, length, conn->shared_key) != sizeof(cookie)) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -3096,3 +3097,18 @@ void kill_net_crypto(Net_Crypto *c)
|
|||||||
crypto_memzero(c, sizeof(Net_Crypto));
|
crypto_memzero(c, sizeof(Net_Crypto));
|
||||||
mem_delete(mem, c);
|
mem_delete(mem, c);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const Net_Profile *nc_get_tcp_client_net_profile(const Net_Crypto *c)
|
||||||
|
{
|
||||||
|
if (c == nullptr) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
const TCP_Connections *tcp_c = nc_get_tcp_c(c);
|
||||||
|
|
||||||
|
if (tcp_c == nullptr) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return tcp_connection_get_client_net_profile(tcp_c);
|
||||||
|
}
|
||||||
|
@ -20,6 +20,7 @@
|
|||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "mono_time.h"
|
#include "mono_time.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
|
|
||||||
/*** Crypto payloads. */
|
/*** Crypto payloads. */
|
||||||
@ -417,4 +418,11 @@ void do_net_crypto(Net_Crypto *c, void *userdata);
|
|||||||
nullable(1)
|
nullable(1)
|
||||||
void kill_net_crypto(Net_Crypto *c);
|
void kill_net_crypto(Net_Crypto *c);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a pointer to the net profile object for the TCP client associated with `c`.
|
||||||
|
* Returns null if `c` is null or the TCP_Connections associated with `c` is null.
|
||||||
|
*/
|
||||||
|
non_null()
|
||||||
|
const Net_Profile *nc_get_tcp_client_net_profile(const Net_Crypto *c);
|
||||||
|
|
||||||
#endif /* C_TOXCORE_TOXCORE_NET_CRYPTO_H */
|
#endif /* C_TOXCORE_TOXCORE_NET_CRYPTO_H */
|
||||||
|
157
external/toxcore/c-toxcore/toxcore/net_profile.c
vendored
Normal file
157
external/toxcore/c-toxcore/toxcore/net_profile.c
vendored
Normal file
@ -0,0 +1,157 @@
|
|||||||
|
/* SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
* Copyright © 2023-2024 The TokTok team.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Functions for the network profile.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "net_profile.h"
|
||||||
|
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
#include "attributes.h"
|
||||||
|
#include "logger.h"
|
||||||
|
#include "mem.h"
|
||||||
|
|
||||||
|
#include "ccompat.h"
|
||||||
|
|
||||||
|
#define NETPROF_TCP_DATA_PACKET_ID 0x10
|
||||||
|
|
||||||
|
typedef struct Net_Profile {
|
||||||
|
uint64_t packets_recv[NET_PROF_MAX_PACKET_IDS];
|
||||||
|
uint64_t packets_sent[NET_PROF_MAX_PACKET_IDS];
|
||||||
|
|
||||||
|
uint64_t total_packets_recv;
|
||||||
|
uint64_t total_packets_sent;
|
||||||
|
|
||||||
|
uint64_t bytes_recv[NET_PROF_MAX_PACKET_IDS];
|
||||||
|
uint64_t bytes_sent[NET_PROF_MAX_PACKET_IDS];
|
||||||
|
|
||||||
|
uint64_t total_bytes_recv;
|
||||||
|
uint64_t total_bytes_sent;
|
||||||
|
} Net_Profile;
|
||||||
|
|
||||||
|
/** Returns the number of sent or received packets for all ID's between `start_id` and `end_id`. */
|
||||||
|
nullable(1)
|
||||||
|
static uint64_t netprof_get_packet_count_id_range(const Net_Profile *profile, uint8_t start_id, uint8_t end_id,
|
||||||
|
Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
const uint64_t *arr = dir == PACKET_DIRECTION_SEND ? profile->packets_sent : profile->packets_recv;
|
||||||
|
uint64_t count = 0;
|
||||||
|
|
||||||
|
for (size_t i = start_id; i <= end_id; ++i) {
|
||||||
|
count += arr[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
/** Returns the number of sent or received bytes for all ID's between `start_id` and `end_id`. */
|
||||||
|
nullable(1)
|
||||||
|
static uint64_t netprof_get_bytes_id_range(const Net_Profile *profile, uint8_t start_id, uint8_t end_id,
|
||||||
|
Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
const uint64_t *arr = dir == PACKET_DIRECTION_SEND ? profile->bytes_sent : profile->bytes_recv;
|
||||||
|
uint64_t bytes = 0;
|
||||||
|
|
||||||
|
for (size_t i = start_id; i <= end_id; ++i) {
|
||||||
|
bytes += arr[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
return bytes;
|
||||||
|
}
|
||||||
|
|
||||||
|
void netprof_record_packet(Net_Profile *profile, uint8_t id, size_t length, Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dir == PACKET_DIRECTION_SEND) {
|
||||||
|
++profile->total_packets_sent;
|
||||||
|
++profile->packets_sent[id];
|
||||||
|
|
||||||
|
profile->total_bytes_sent += length;
|
||||||
|
profile->bytes_sent[id] += length;
|
||||||
|
} else {
|
||||||
|
++profile->total_packets_recv;
|
||||||
|
++profile->packets_recv[id];
|
||||||
|
|
||||||
|
profile->total_bytes_recv += length;
|
||||||
|
profile->bytes_recv[id] += length;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t netprof_get_packet_count_id(const Net_Profile *profile, uint8_t id, Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Special case - TCP data packets can have any ID between 0x10 and 0xff
|
||||||
|
if (id == NETPROF_TCP_DATA_PACKET_ID) {
|
||||||
|
return netprof_get_packet_count_id_range(profile, id, UINT8_MAX, dir);
|
||||||
|
}
|
||||||
|
|
||||||
|
return dir == PACKET_DIRECTION_SEND ? profile->packets_sent[id] : profile->packets_recv[id];
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t netprof_get_packet_count_total(const Net_Profile *profile, Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return dir == PACKET_DIRECTION_SEND ? profile->total_packets_sent : profile->total_packets_recv;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t netprof_get_bytes_id(const Net_Profile *profile, uint8_t id, Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Special case - TCP data packets can have any ID between 0x10 and 0xff
|
||||||
|
if (id == NETPROF_TCP_DATA_PACKET_ID) {
|
||||||
|
return netprof_get_bytes_id_range(profile, id, 0xff, dir);
|
||||||
|
}
|
||||||
|
|
||||||
|
return dir == PACKET_DIRECTION_SEND ? profile->bytes_sent[id] : profile->bytes_recv[id];
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t netprof_get_bytes_total(const Net_Profile *profile, Packet_Direction dir)
|
||||||
|
{
|
||||||
|
if (profile == nullptr) {
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return dir == PACKET_DIRECTION_SEND ? profile->total_bytes_sent : profile->total_bytes_recv;
|
||||||
|
}
|
||||||
|
|
||||||
|
Net_Profile *netprof_new(const Logger *log, const Memory *mem)
|
||||||
|
{
|
||||||
|
Net_Profile *np = (Net_Profile *)mem_alloc(mem, sizeof(Net_Profile));
|
||||||
|
|
||||||
|
if (np == nullptr) {
|
||||||
|
LOGGER_ERROR(log, "failed to allocate memory for net profiler");
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return np;
|
||||||
|
}
|
||||||
|
|
||||||
|
void netprof_kill(const Memory *mem, Net_Profile *net_profile)
|
||||||
|
{
|
||||||
|
if (net_profile != nullptr) {
|
||||||
|
mem_delete(mem, net_profile);
|
||||||
|
}
|
||||||
|
}
|
73
external/toxcore/c-toxcore/toxcore/net_profile.h
vendored
Normal file
73
external/toxcore/c-toxcore/toxcore/net_profile.h
vendored
Normal file
@ -0,0 +1,73 @@
|
|||||||
|
/* SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
|
* Copyright © 2023-2024 The TokTok team.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Functions for the network profile.
|
||||||
|
*/
|
||||||
|
#ifndef C_TOXCORE_TOXCORE_NET_PROFILE_H
|
||||||
|
#define C_TOXCORE_TOXCORE_NET_PROFILE_H
|
||||||
|
|
||||||
|
#include <stddef.h>
|
||||||
|
#include <stdint.h>
|
||||||
|
|
||||||
|
#include "attributes.h"
|
||||||
|
#include "logger.h"
|
||||||
|
#include "mem.h"
|
||||||
|
|
||||||
|
/* The max number of packet ID's (must fit inside one byte) */
|
||||||
|
#define NET_PROF_MAX_PACKET_IDS 256
|
||||||
|
|
||||||
|
/* If passed to a netprof function as a nullptr the function will have no effect. */
|
||||||
|
typedef struct Net_Profile Net_Profile;
|
||||||
|
|
||||||
|
/** Specifies whether the query is for sent or received packets. */
|
||||||
|
typedef enum Packet_Direction {
|
||||||
|
PACKET_DIRECTION_SEND,
|
||||||
|
PACKET_DIRECTION_RECV,
|
||||||
|
} Packet_Direction;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Records a sent or received packet of type `id` and size `length` to the given profile.
|
||||||
|
*/
|
||||||
|
nullable(1)
|
||||||
|
void netprof_record_packet(Net_Profile *profile, uint8_t id, size_t length, Packet_Direction dir);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the number of sent or received packets of type `id` for the given profile.
|
||||||
|
*/
|
||||||
|
nullable(1)
|
||||||
|
uint64_t netprof_get_packet_count_id(const Net_Profile *profile, uint8_t id, Packet_Direction dir);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the total number of sent or received packets for the given profile.
|
||||||
|
*/
|
||||||
|
nullable(1)
|
||||||
|
uint64_t netprof_get_packet_count_total(const Net_Profile *profile, Packet_Direction dir);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the number of bytes sent or received of packet type `id` for the given profile.
|
||||||
|
*/
|
||||||
|
nullable(1)
|
||||||
|
uint64_t netprof_get_bytes_id(const Net_Profile *profile, uint8_t id, Packet_Direction dir);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the total number of bytes sent or received for the given profile.
|
||||||
|
*/
|
||||||
|
nullable(1)
|
||||||
|
uint64_t netprof_get_bytes_total(const Net_Profile *profile, Packet_Direction dir);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns a new net_profile object. The caller is responsible for freeing the
|
||||||
|
* returned memory via `netprof_kill`.
|
||||||
|
*/
|
||||||
|
non_null()
|
||||||
|
Net_Profile *netprof_new(const Logger *log, const Memory *mem);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Kills a net_profile object and frees all associated memory.
|
||||||
|
*/
|
||||||
|
non_null(1) nullable(2)
|
||||||
|
void netprof_kill(const Memory *mem, Net_Profile *net_profile);
|
||||||
|
|
||||||
|
#endif /* C_TOXCORE_TOXCORE_NET_PROFILE_H */
|
36
external/toxcore/c-toxcore/toxcore/network.c
vendored
36
external/toxcore/c-toxcore/toxcore/network.c
vendored
@ -86,6 +86,7 @@
|
|||||||
#include "ccompat.h"
|
#include "ccompat.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
// Disable MSG_NOSIGNAL on systems not supporting it, e.g. Windows, FreeBSD
|
// Disable MSG_NOSIGNAL on systems not supporting it, e.g. Windows, FreeBSD
|
||||||
@ -907,9 +908,14 @@ static void loglogdata(const Logger *log, const char *message, const uint8_t *bu
|
|||||||
}
|
}
|
||||||
|
|
||||||
int net_send(const Network *ns, const Logger *log,
|
int net_send(const Network *ns, const Logger *log,
|
||||||
Socket sock, const uint8_t *buf, size_t len, const IP_Port *ip_port)
|
Socket sock, const uint8_t *buf, size_t len, const IP_Port *ip_port, Net_Profile *net_profile)
|
||||||
{
|
{
|
||||||
const int res = ns->funcs->send(ns->obj, sock, buf, len);
|
const int res = ns->funcs->send(ns->obj, sock, buf, len);
|
||||||
|
|
||||||
|
if (res > 0) {
|
||||||
|
netprof_record_packet(net_profile, buf[0], res, PACKET_DIRECTION_SEND);
|
||||||
|
}
|
||||||
|
|
||||||
loglogdata(log, "T=>", buf, len, ip_port, res);
|
loglogdata(log, "T=>", buf, len, ip_port, res);
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
@ -1013,6 +1019,8 @@ struct Networking_Core {
|
|||||||
uint16_t port;
|
uint16_t port;
|
||||||
/* Our UDP socket. */
|
/* Our UDP socket. */
|
||||||
Socket sock;
|
Socket sock;
|
||||||
|
|
||||||
|
Net_Profile *udp_net_profile;
|
||||||
};
|
};
|
||||||
|
|
||||||
Family net_family(const Networking_Core *net)
|
Family net_family(const Networking_Core *net)
|
||||||
@ -1098,6 +1106,11 @@ int send_packet(const Networking_Core *net, const IP_Port *ip_port, Packet packe
|
|||||||
loglogdata(net->log, "O=>", packet.data, packet.length, ip_port, res);
|
loglogdata(net->log, "O=>", packet.data, packet.length, ip_port, res);
|
||||||
|
|
||||||
assert(res <= INT_MAX);
|
assert(res <= INT_MAX);
|
||||||
|
|
||||||
|
if (res == packet.length && packet.data != nullptr) {
|
||||||
|
netprof_record_packet(net->udp_net_profile, packet.data[0], packet.length, PACKET_DIRECTION_SEND);
|
||||||
|
}
|
||||||
|
|
||||||
return (int)res;
|
return (int)res;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1202,6 +1215,8 @@ void networking_poll(const Networking_Core *net, void *userdata)
|
|||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
netprof_record_packet(net->udp_net_profile, data[0], length, PACKET_DIRECTION_RECV);
|
||||||
|
|
||||||
const Packet_Handler *const handler = &net->packethandlers[data[0]];
|
const Packet_Handler *const handler = &net->packethandlers[data[0]];
|
||||||
|
|
||||||
if (handler->function == nullptr) {
|
if (handler->function == nullptr) {
|
||||||
@ -1262,6 +1277,14 @@ Networking_Core *new_networking_ex(
|
|||||||
return nullptr;
|
return nullptr;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Net_Profile *np = netprof_new(log, mem);
|
||||||
|
|
||||||
|
if (np == nullptr) {
|
||||||
|
free(temp);
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
temp->udp_net_profile = np;
|
||||||
temp->ns = ns;
|
temp->ns = ns;
|
||||||
temp->log = log;
|
temp->log = log;
|
||||||
temp->mem = mem;
|
temp->mem = mem;
|
||||||
@ -1278,6 +1301,7 @@ Networking_Core *new_networking_ex(
|
|||||||
char *strerror = net_new_strerror(neterror);
|
char *strerror = net_new_strerror(neterror);
|
||||||
LOGGER_ERROR(log, "failed to get a socket?! %d, %s", neterror, strerror);
|
LOGGER_ERROR(log, "failed to get a socket?! %d, %s", neterror, strerror);
|
||||||
net_kill_strerror(strerror);
|
net_kill_strerror(strerror);
|
||||||
|
netprof_kill(mem, temp->udp_net_profile);
|
||||||
mem_delete(mem, temp);
|
mem_delete(mem, temp);
|
||||||
|
|
||||||
if (error != nullptr) {
|
if (error != nullptr) {
|
||||||
@ -1485,6 +1509,7 @@ void kill_networking(Networking_Core *net)
|
|||||||
kill_sock(net->ns, net->sock);
|
kill_sock(net->ns, net->sock);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
netprof_kill(net->mem, net->udp_net_profile);
|
||||||
mem_delete(net->mem, net);
|
mem_delete(net->mem, net);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -2393,3 +2418,12 @@ void net_kill_strerror(char *strerror)
|
|||||||
free(strerror);
|
free(strerror);
|
||||||
#endif /* OS_WIN32 */
|
#endif /* OS_WIN32 */
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const Net_Profile *net_get_net_profile(const Networking_Core *net)
|
||||||
|
{
|
||||||
|
if (net == nullptr) {
|
||||||
|
return nullptr;
|
||||||
|
}
|
||||||
|
|
||||||
|
return net->udp_net_profile;
|
||||||
|
}
|
||||||
|
13
external/toxcore/c-toxcore/toxcore/network.h
vendored
13
external/toxcore/c-toxcore/toxcore/network.h
vendored
@ -17,6 +17,7 @@
|
|||||||
#include "bin_pack.h"
|
#include "bin_pack.h"
|
||||||
#include "logger.h"
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
|
#include "net_profile.h"
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
@ -236,8 +237,9 @@ Socket net_invalid_socket(void);
|
|||||||
/**
|
/**
|
||||||
* Calls send(sockfd, buf, len, MSG_NOSIGNAL).
|
* Calls send(sockfd, buf, len, MSG_NOSIGNAL).
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null(1, 2, 4, 6) nullable(7)
|
||||||
int net_send(const Network *ns, const Logger *log, Socket sock, const uint8_t *buf, size_t len, const IP_Port *ip_port);
|
int net_send(const Network *ns, const Logger *log, Socket sock, const uint8_t *buf, size_t len, const IP_Port *ip_port,
|
||||||
|
Net_Profile *net_profile);
|
||||||
/**
|
/**
|
||||||
* Calls recv(sockfd, buf, len, MSG_NOSIGNAL).
|
* Calls recv(sockfd, buf, len, MSG_NOSIGNAL).
|
||||||
*/
|
*/
|
||||||
@ -614,6 +616,13 @@ Networking_Core *new_networking_no_udp(const Logger *log, const Memory *mem, con
|
|||||||
nullable(1)
|
nullable(1)
|
||||||
void kill_networking(Networking_Core *net);
|
void kill_networking(Networking_Core *net);
|
||||||
|
|
||||||
|
/** @brief Returns a pointer to the network net_profile object associated with `net`.
|
||||||
|
*
|
||||||
|
* Returns null if `net` is null.
|
||||||
|
*/
|
||||||
|
non_null()
|
||||||
|
const Net_Profile *net_get_net_profile(const Networking_Core *net);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
} /* extern "C" */
|
} /* extern "C" */
|
||||||
#endif
|
#endif
|
||||||
|
32
external/toxcore/c-toxcore/toxcore/onion.c
vendored
32
external/toxcore/c-toxcore/toxcore/onion.c
vendored
@ -180,7 +180,7 @@ int onion_path_to_nodes(Node_format *nodes, unsigned int num_nodes, const Onion_
|
|||||||
* return -1 on failure.
|
* return -1 on failure.
|
||||||
* return length of created packet on success.
|
* return length of created packet on success.
|
||||||
*/
|
*/
|
||||||
int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
int create_onion_packet(const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
||||||
const Onion_Path *path, const IP_Port *dest,
|
const Onion_Path *path, const IP_Port *dest,
|
||||||
const uint8_t *data, uint16_t length)
|
const uint8_t *data, uint16_t length)
|
||||||
{
|
{
|
||||||
@ -202,7 +202,7 @@ int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
ipport_pack(step2, &path->ip_port3);
|
ipport_pack(step2, &path->ip_port3);
|
||||||
memcpy(step2 + SIZE_IPPORT, path->public_key3, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(step2 + SIZE_IPPORT, path->public_key3, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
int len = encrypt_data_symmetric(mem, path->shared_key3, nonce, step1, step1_size,
|
int len = encrypt_data_symmetric(path->shared_key3, nonce, step1, step1_size,
|
||||||
step2 + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
step2 + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (len != SIZE_IPPORT + length + CRYPTO_MAC_SIZE) {
|
if (len != SIZE_IPPORT + length + CRYPTO_MAC_SIZE) {
|
||||||
@ -213,7 +213,7 @@ int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
VLA(uint8_t, step3, step3_size);
|
VLA(uint8_t, step3, step3_size);
|
||||||
ipport_pack(step3, &path->ip_port2);
|
ipport_pack(step3, &path->ip_port2);
|
||||||
memcpy(step3 + SIZE_IPPORT, path->public_key2, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(step3 + SIZE_IPPORT, path->public_key2, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
len = encrypt_data_symmetric(mem, path->shared_key2, nonce, step2, step2_size,
|
len = encrypt_data_symmetric(path->shared_key2, nonce, step2, step2_size,
|
||||||
step3 + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
step3 + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (len != SIZE_IPPORT + SEND_BASE + length + CRYPTO_MAC_SIZE) {
|
if (len != SIZE_IPPORT + SEND_BASE + length + CRYPTO_MAC_SIZE) {
|
||||||
@ -224,7 +224,7 @@ int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
memcpy(packet + 1, nonce, CRYPTO_NONCE_SIZE);
|
memcpy(packet + 1, nonce, CRYPTO_NONCE_SIZE);
|
||||||
memcpy(packet + 1 + CRYPTO_NONCE_SIZE, path->public_key1, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(packet + 1 + CRYPTO_NONCE_SIZE, path->public_key1, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
len = encrypt_data_symmetric(mem, path->shared_key1, nonce, step3, step3_size,
|
len = encrypt_data_symmetric(path->shared_key1, nonce, step3, step3_size,
|
||||||
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (len != SIZE_IPPORT + SEND_BASE * 2 + length + CRYPTO_MAC_SIZE) {
|
if (len != SIZE_IPPORT + SEND_BASE * 2 + length + CRYPTO_MAC_SIZE) {
|
||||||
@ -243,7 +243,7 @@ int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
* return -1 on failure.
|
* return -1 on failure.
|
||||||
* return length of created packet on success.
|
* return length of created packet on success.
|
||||||
*/
|
*/
|
||||||
int create_onion_packet_tcp(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
int create_onion_packet_tcp(const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
||||||
const Onion_Path *path, const IP_Port *dest,
|
const Onion_Path *path, const IP_Port *dest,
|
||||||
const uint8_t *data, uint16_t length)
|
const uint8_t *data, uint16_t length)
|
||||||
{
|
{
|
||||||
@ -265,7 +265,7 @@ int create_onion_packet_tcp(const Memory *mem, const Random *rng, uint8_t *packe
|
|||||||
ipport_pack(step2, &path->ip_port3);
|
ipport_pack(step2, &path->ip_port3);
|
||||||
memcpy(step2 + SIZE_IPPORT, path->public_key3, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(step2 + SIZE_IPPORT, path->public_key3, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
int len = encrypt_data_symmetric(mem, path->shared_key3, nonce, step1, step1_size,
|
int len = encrypt_data_symmetric(path->shared_key3, nonce, step1, step1_size,
|
||||||
step2 + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
step2 + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (len != SIZE_IPPORT + length + CRYPTO_MAC_SIZE) {
|
if (len != SIZE_IPPORT + length + CRYPTO_MAC_SIZE) {
|
||||||
@ -274,7 +274,7 @@ int create_onion_packet_tcp(const Memory *mem, const Random *rng, uint8_t *packe
|
|||||||
|
|
||||||
ipport_pack(packet + CRYPTO_NONCE_SIZE, &path->ip_port2);
|
ipport_pack(packet + CRYPTO_NONCE_SIZE, &path->ip_port2);
|
||||||
memcpy(packet + CRYPTO_NONCE_SIZE + SIZE_IPPORT, path->public_key2, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(packet + CRYPTO_NONCE_SIZE + SIZE_IPPORT, path->public_key2, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
len = encrypt_data_symmetric(mem, path->shared_key2, nonce, step2, step2_size,
|
len = encrypt_data_symmetric(path->shared_key2, nonce, step2, step2_size,
|
||||||
packet + CRYPTO_NONCE_SIZE + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
packet + CRYPTO_NONCE_SIZE + SIZE_IPPORT + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (len != SIZE_IPPORT + SEND_BASE + length + CRYPTO_MAC_SIZE) {
|
if (len != SIZE_IPPORT + SEND_BASE + length + CRYPTO_MAC_SIZE) {
|
||||||
@ -355,7 +355,7 @@ static int handle_send_initial(void *object, const IP_Port *source, const uint8_
|
|||||||
}
|
}
|
||||||
|
|
||||||
const int len = decrypt_data_symmetric(
|
const int len = decrypt_data_symmetric(
|
||||||
onion->mem, shared_key, &packet[nonce_start], &packet[ciphertext_start], ciphertext_length, plain);
|
shared_key, &packet[nonce_start], &packet[ciphertext_start], ciphertext_length, plain);
|
||||||
|
|
||||||
if (len != plaintext_length) {
|
if (len != plaintext_length) {
|
||||||
LOGGER_TRACE(onion->log, "decrypt failed: %d != %d", len, plaintext_length);
|
LOGGER_TRACE(onion->log, "decrypt failed: %d != %d", len, plaintext_length);
|
||||||
@ -393,7 +393,7 @@ int onion_send_1(const Onion *onion, const uint8_t *plain, uint16_t len, const I
|
|||||||
uint16_t data_len = 1 + CRYPTO_NONCE_SIZE + (len - SIZE_IPPORT);
|
uint16_t data_len = 1 + CRYPTO_NONCE_SIZE + (len - SIZE_IPPORT);
|
||||||
uint8_t *ret_part = data + data_len;
|
uint8_t *ret_part = data + data_len;
|
||||||
random_nonce(onion->rng, ret_part);
|
random_nonce(onion->rng, ret_part);
|
||||||
len = encrypt_data_symmetric(onion->mem, onion->secret_symmetric_key, ret_part, ip_port, SIZE_IPPORT,
|
len = encrypt_data_symmetric(onion->secret_symmetric_key, ret_part, ip_port, SIZE_IPPORT,
|
||||||
ret_part + CRYPTO_NONCE_SIZE);
|
ret_part + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != SIZE_IPPORT + CRYPTO_MAC_SIZE) {
|
if (len != SIZE_IPPORT + CRYPTO_MAC_SIZE) {
|
||||||
@ -436,7 +436,7 @@ static int handle_send_1(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int len = decrypt_data_symmetric(onion->mem, shared_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE,
|
int len = decrypt_data_symmetric(shared_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_1), plain);
|
length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_1), plain);
|
||||||
|
|
||||||
if (len != length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_1 + CRYPTO_MAC_SIZE)) {
|
if (len != length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_1 + CRYPTO_MAC_SIZE)) {
|
||||||
@ -459,7 +459,7 @@ static int handle_send_1(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
uint8_t ret_data[RETURN_1 + SIZE_IPPORT];
|
uint8_t ret_data[RETURN_1 + SIZE_IPPORT];
|
||||||
ipport_pack(ret_data, source);
|
ipport_pack(ret_data, source);
|
||||||
memcpy(ret_data + SIZE_IPPORT, packet + (length - RETURN_1), RETURN_1);
|
memcpy(ret_data + SIZE_IPPORT, packet + (length - RETURN_1), RETURN_1);
|
||||||
len = encrypt_data_symmetric(onion->mem, onion->secret_symmetric_key, ret_part, ret_data, sizeof(ret_data),
|
len = encrypt_data_symmetric(onion->secret_symmetric_key, ret_part, ret_data, sizeof(ret_data),
|
||||||
ret_part + CRYPTO_NONCE_SIZE);
|
ret_part + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != RETURN_2 - CRYPTO_NONCE_SIZE) {
|
if (len != RETURN_2 - CRYPTO_NONCE_SIZE) {
|
||||||
@ -502,7 +502,7 @@ static int handle_send_2(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int len = decrypt_data_symmetric(onion->mem, shared_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE,
|
int len = decrypt_data_symmetric(shared_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_2), plain);
|
length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_2), plain);
|
||||||
|
|
||||||
if (len != length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_2 + CRYPTO_MAC_SIZE)) {
|
if (len != length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + RETURN_2 + CRYPTO_MAC_SIZE)) {
|
||||||
@ -532,7 +532,7 @@ static int handle_send_2(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
uint8_t ret_data[RETURN_2 + SIZE_IPPORT];
|
uint8_t ret_data[RETURN_2 + SIZE_IPPORT];
|
||||||
ipport_pack(ret_data, source);
|
ipport_pack(ret_data, source);
|
||||||
memcpy(ret_data + SIZE_IPPORT, packet + (length - RETURN_2), RETURN_2);
|
memcpy(ret_data + SIZE_IPPORT, packet + (length - RETURN_2), RETURN_2);
|
||||||
len = encrypt_data_symmetric(onion->mem, onion->secret_symmetric_key, ret_part, ret_data, sizeof(ret_data),
|
len = encrypt_data_symmetric(onion->secret_symmetric_key, ret_part, ret_data, sizeof(ret_data),
|
||||||
ret_part + CRYPTO_NONCE_SIZE);
|
ret_part + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != RETURN_3 - CRYPTO_NONCE_SIZE) {
|
if (len != RETURN_3 - CRYPTO_NONCE_SIZE) {
|
||||||
@ -574,7 +574,7 @@ static int handle_recv_3(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
change_symmetric_key(onion);
|
change_symmetric_key(onion);
|
||||||
|
|
||||||
uint8_t plain[SIZE_IPPORT + RETURN_2];
|
uint8_t plain[SIZE_IPPORT + RETURN_2];
|
||||||
const int len = decrypt_data_symmetric(onion->mem, onion->secret_symmetric_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
const int len = decrypt_data_symmetric(onion->secret_symmetric_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
||||||
SIZE_IPPORT + RETURN_2 + CRYPTO_MAC_SIZE, plain);
|
SIZE_IPPORT + RETURN_2 + CRYPTO_MAC_SIZE, plain);
|
||||||
|
|
||||||
if ((uint32_t)len != sizeof(plain)) {
|
if ((uint32_t)len != sizeof(plain)) {
|
||||||
@ -627,7 +627,7 @@ static int handle_recv_2(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
change_symmetric_key(onion);
|
change_symmetric_key(onion);
|
||||||
|
|
||||||
uint8_t plain[SIZE_IPPORT + RETURN_1];
|
uint8_t plain[SIZE_IPPORT + RETURN_1];
|
||||||
const int len = decrypt_data_symmetric(onion->mem, onion->secret_symmetric_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
const int len = decrypt_data_symmetric(onion->secret_symmetric_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
||||||
SIZE_IPPORT + RETURN_1 + CRYPTO_MAC_SIZE, plain);
|
SIZE_IPPORT + RETURN_1 + CRYPTO_MAC_SIZE, plain);
|
||||||
|
|
||||||
if ((uint32_t)len != sizeof(plain)) {
|
if ((uint32_t)len != sizeof(plain)) {
|
||||||
@ -679,7 +679,7 @@ static int handle_recv_1(void *object, const IP_Port *source, const uint8_t *pac
|
|||||||
change_symmetric_key(onion);
|
change_symmetric_key(onion);
|
||||||
|
|
||||||
uint8_t plain[SIZE_IPPORT];
|
uint8_t plain[SIZE_IPPORT];
|
||||||
const int len = decrypt_data_symmetric(onion->mem, onion->secret_symmetric_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
const int len = decrypt_data_symmetric(onion->secret_symmetric_key, packet + 1, packet + 1 + CRYPTO_NONCE_SIZE,
|
||||||
SIZE_IPPORT + CRYPTO_MAC_SIZE, plain);
|
SIZE_IPPORT + CRYPTO_MAC_SIZE, plain);
|
||||||
|
|
||||||
if ((uint32_t)len != SIZE_IPPORT) {
|
if ((uint32_t)len != SIZE_IPPORT) {
|
||||||
|
4
external/toxcore/c-toxcore/toxcore/onion.h
vendored
4
external/toxcore/c-toxcore/toxcore/onion.h
vendored
@ -105,7 +105,7 @@ int onion_path_to_nodes(Node_format *nodes, unsigned int num_nodes, const Onion_
|
|||||||
* return length of created packet on success.
|
* return length of created packet on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
int create_onion_packet(const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
||||||
const Onion_Path *path, const IP_Port *dest,
|
const Onion_Path *path, const IP_Port *dest,
|
||||||
const uint8_t *data, uint16_t length);
|
const uint8_t *data, uint16_t length);
|
||||||
|
|
||||||
@ -119,7 +119,7 @@ int create_onion_packet(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
* return length of created packet on success.
|
* return length of created packet on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int create_onion_packet_tcp(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
int create_onion_packet_tcp(const Random *rng, uint8_t *packet, uint16_t max_packet_length,
|
||||||
const Onion_Path *path, const IP_Port *dest,
|
const Onion_Path *path, const IP_Port *dest,
|
||||||
const uint8_t *data, uint16_t length);
|
const uint8_t *data, uint16_t length);
|
||||||
|
|
||||||
|
135
external/toxcore/c-toxcore/toxcore/onion_announce.c
vendored
135
external/toxcore/c-toxcore/toxcore/onion_announce.c
vendored
@ -9,6 +9,7 @@
|
|||||||
#include "onion_announce.h"
|
#include "onion_announce.h"
|
||||||
|
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#include "DHT.h"
|
#include "DHT.h"
|
||||||
@ -22,7 +23,6 @@
|
|||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "onion.h"
|
#include "onion.h"
|
||||||
#include "shared_key_cache.h"
|
#include "shared_key_cache.h"
|
||||||
#include "sort.h"
|
|
||||||
#include "timed_auth.h"
|
#include "timed_auth.h"
|
||||||
|
|
||||||
#define PING_ID_TIMEOUT ONION_ANNOUNCE_TIMEOUT
|
#define PING_ID_TIMEOUT ONION_ANNOUNCE_TIMEOUT
|
||||||
@ -103,7 +103,7 @@ void onion_announce_entry_set_time(Onion_Announce *onion_a, uint32_t entry, uint
|
|||||||
* return -1 on failure.
|
* return -1 on failure.
|
||||||
* return packet length on success.
|
* return packet length on success.
|
||||||
*/
|
*/
|
||||||
int create_announce_request(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
int create_announce_request(const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
||||||
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id, const uint8_t *client_id,
|
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id, const uint8_t *client_id,
|
||||||
const uint8_t *data_public_key, uint64_t sendback_data)
|
const uint8_t *data_public_key, uint64_t sendback_data)
|
||||||
{
|
{
|
||||||
@ -122,7 +122,7 @@ int create_announce_request(const Memory *mem, const Random *rng, uint8_t *packe
|
|||||||
packet[0] = NET_PACKET_ANNOUNCE_REQUEST_OLD;
|
packet[0] = NET_PACKET_ANNOUNCE_REQUEST_OLD;
|
||||||
random_nonce(rng, packet + 1);
|
random_nonce(rng, packet + 1);
|
||||||
|
|
||||||
const int len = encrypt_data(mem, dest_client_id, secret_key, packet + 1, plain, sizeof(plain),
|
const int len = encrypt_data(dest_client_id, secret_key, packet + 1, plain, sizeof(plain),
|
||||||
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if ((uint32_t)len + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE != ONION_ANNOUNCE_REQUEST_MIN_SIZE) {
|
if ((uint32_t)len + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE != ONION_ANNOUNCE_REQUEST_MIN_SIZE) {
|
||||||
@ -146,7 +146,7 @@ int create_announce_request(const Memory *mem, const Random *rng, uint8_t *packe
|
|||||||
* return -1 on failure.
|
* return -1 on failure.
|
||||||
* return 0 on success.
|
* return 0 on success.
|
||||||
*/
|
*/
|
||||||
int create_data_request(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *public_key,
|
int create_data_request(const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *public_key,
|
||||||
const uint8_t *encrypt_public_key, const uint8_t *nonce, const uint8_t *data, uint16_t length)
|
const uint8_t *encrypt_public_key, const uint8_t *nonce, const uint8_t *data, uint16_t length)
|
||||||
{
|
{
|
||||||
if (DATA_REQUEST_MIN_SIZE + length > max_packet_length) {
|
if (DATA_REQUEST_MIN_SIZE + length > max_packet_length) {
|
||||||
@ -167,7 +167,7 @@ int create_data_request(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
|
|
||||||
memcpy(packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE, random_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE, random_public_key, CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
const int len = encrypt_data(mem, encrypt_public_key, random_secret_key, packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, data, length,
|
const int len = encrypt_data(encrypt_public_key, random_secret_key, packet + 1 + CRYPTO_PUBLIC_KEY_SIZE, data, length,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
if (1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + len != DATA_REQUEST_MIN_SIZE +
|
if (1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE + len != DATA_REQUEST_MIN_SIZE +
|
||||||
@ -193,14 +193,14 @@ int create_data_request(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
* return 0 on success.
|
* return 0 on success.
|
||||||
*/
|
*/
|
||||||
int send_announce_request(
|
int send_announce_request(
|
||||||
const Logger *log, const Memory *mem, const Networking_Core *net, const Random *rng,
|
const Logger *log, const Networking_Core *net, const Random *rng,
|
||||||
const Onion_Path *path, const Node_format *dest,
|
const Onion_Path *path, const Node_format *dest,
|
||||||
const uint8_t *public_key, const uint8_t *secret_key,
|
const uint8_t *public_key, const uint8_t *secret_key,
|
||||||
const uint8_t *ping_id, const uint8_t *client_id,
|
const uint8_t *ping_id, const uint8_t *client_id,
|
||||||
const uint8_t *data_public_key, uint64_t sendback_data)
|
const uint8_t *data_public_key, uint64_t sendback_data)
|
||||||
{
|
{
|
||||||
uint8_t request[ONION_ANNOUNCE_REQUEST_MIN_SIZE];
|
uint8_t request[ONION_ANNOUNCE_REQUEST_MIN_SIZE];
|
||||||
int len = create_announce_request(mem, rng, request, sizeof(request), dest->public_key, public_key, secret_key, ping_id,
|
int len = create_announce_request(rng, request, sizeof(request), dest->public_key, public_key, secret_key, ping_id,
|
||||||
client_id, data_public_key, sendback_data);
|
client_id, data_public_key, sendback_data);
|
||||||
|
|
||||||
if (len != sizeof(request)) {
|
if (len != sizeof(request)) {
|
||||||
@ -208,7 +208,7 @@ int send_announce_request(
|
|||||||
}
|
}
|
||||||
|
|
||||||
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
||||||
len = create_onion_packet(mem, rng, packet, sizeof(packet), path, &dest->ip_port, request, sizeof(request));
|
len = create_onion_packet(rng, packet, sizeof(packet), path, &dest->ip_port, request, sizeof(request));
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -238,19 +238,19 @@ int send_announce_request(
|
|||||||
* return 0 on success.
|
* return 0 on success.
|
||||||
*/
|
*/
|
||||||
int send_data_request(
|
int send_data_request(
|
||||||
const Logger *log, const Memory *mem, const Networking_Core *net, const Random *rng, const Onion_Path *path, const IP_Port *dest,
|
const Logger *log, const Networking_Core *net, const Random *rng, const Onion_Path *path, const IP_Port *dest,
|
||||||
const uint8_t *public_key, const uint8_t *encrypt_public_key, const uint8_t *nonce,
|
const uint8_t *public_key, const uint8_t *encrypt_public_key, const uint8_t *nonce,
|
||||||
const uint8_t *data, uint16_t length)
|
const uint8_t *data, uint16_t length)
|
||||||
{
|
{
|
||||||
uint8_t request[ONION_MAX_DATA_SIZE];
|
uint8_t request[ONION_MAX_DATA_SIZE];
|
||||||
int len = create_data_request(mem, rng, request, sizeof(request), public_key, encrypt_public_key, nonce, data, length);
|
int len = create_data_request(rng, request, sizeof(request), public_key, encrypt_public_key, nonce, data, length);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
||||||
len = create_onion_packet(mem, rng, packet, sizeof(packet), path, dest, request, len);
|
len = create_onion_packet(rng, packet, sizeof(packet), path, dest, request, len);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -281,17 +281,23 @@ static int in_entries(const Onion_Announce *onion_a, const uint8_t *public_key)
|
|||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
typedef struct Onion_Announce_Entry_Cmp {
|
typedef struct Cmp_Data {
|
||||||
const Memory *mem;
|
|
||||||
const Mono_Time *mono_time;
|
const Mono_Time *mono_time;
|
||||||
const uint8_t *comp_public_key;
|
const uint8_t *base_public_key;
|
||||||
} Onion_Announce_Entry_Cmp;
|
Onion_Announce_Entry entry;
|
||||||
|
} Cmp_Data;
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
static int onion_announce_entry_cmp(const Onion_Announce_Entry_Cmp *cmp, const Onion_Announce_Entry *entry1, const Onion_Announce_Entry *entry2)
|
static int cmp_entry(const void *a, const void *b)
|
||||||
{
|
{
|
||||||
const bool t1 = mono_time_is_timeout(cmp->mono_time, entry1->announce_time, ONION_ANNOUNCE_TIMEOUT);
|
const Cmp_Data *cmp1 = (const Cmp_Data *)a;
|
||||||
const bool t2 = mono_time_is_timeout(cmp->mono_time, entry2->announce_time, ONION_ANNOUNCE_TIMEOUT);
|
const Cmp_Data *cmp2 = (const Cmp_Data *)b;
|
||||||
|
const Onion_Announce_Entry entry1 = cmp1->entry;
|
||||||
|
const Onion_Announce_Entry entry2 = cmp2->entry;
|
||||||
|
const uint8_t *cmp_public_key = cmp1->base_public_key;
|
||||||
|
|
||||||
|
const bool t1 = mono_time_is_timeout(cmp1->mono_time, entry1.announce_time, ONION_ANNOUNCE_TIMEOUT);
|
||||||
|
const bool t2 = mono_time_is_timeout(cmp1->mono_time, entry2.announce_time, ONION_ANNOUNCE_TIMEOUT);
|
||||||
|
|
||||||
if (t1 && t2) {
|
if (t1 && t2) {
|
||||||
return 0;
|
return 0;
|
||||||
@ -305,7 +311,7 @@ static int onion_announce_entry_cmp(const Onion_Announce_Entry_Cmp *cmp, const O
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int closest = id_closest(cmp->comp_public_key, entry1->public_key, entry2->public_key);
|
const int closest = id_closest(cmp_public_key, entry1.public_key, entry2.public_key);
|
||||||
|
|
||||||
if (closest == 1) {
|
if (closest == 1) {
|
||||||
return 1;
|
return 1;
|
||||||
@ -318,81 +324,32 @@ static int onion_announce_entry_cmp(const Onion_Announce_Entry_Cmp *cmp, const O
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
non_null()
|
|
||||||
static bool onion_announce_entry_less_handler(const void *object, const void *a, const void *b)
|
|
||||||
{
|
|
||||||
const Onion_Announce_Entry_Cmp *cmp = (const Onion_Announce_Entry_Cmp *)object;
|
|
||||||
const Onion_Announce_Entry *entry1 = (const Onion_Announce_Entry *)a;
|
|
||||||
const Onion_Announce_Entry *entry2 = (const Onion_Announce_Entry *)b;
|
|
||||||
|
|
||||||
return onion_announce_entry_cmp(cmp, entry1, entry2) < 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static const void *onion_announce_entry_get_handler(const void *arr, uint32_t index)
|
|
||||||
{
|
|
||||||
const Onion_Announce_Entry *entries = (const Onion_Announce_Entry *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void onion_announce_entry_set_handler(void *arr, uint32_t index, const void *val)
|
|
||||||
{
|
|
||||||
Onion_Announce_Entry *entries = (Onion_Announce_Entry *)arr;
|
|
||||||
const Onion_Announce_Entry *entry = (const Onion_Announce_Entry *)val;
|
|
||||||
entries[index] = *entry;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *onion_announce_entry_subarr_handler(void *arr, uint32_t index, uint32_t size)
|
|
||||||
{
|
|
||||||
Onion_Announce_Entry *entries = (Onion_Announce_Entry *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *onion_announce_entry_alloc_handler(const void *object, uint32_t size)
|
|
||||||
{
|
|
||||||
const Onion_Announce_Entry_Cmp *cmp = (const Onion_Announce_Entry_Cmp *)object;
|
|
||||||
Onion_Announce_Entry *tmp = (Onion_Announce_Entry *)mem_valloc(cmp->mem, size, sizeof(Onion_Announce_Entry));
|
|
||||||
|
|
||||||
if (tmp == nullptr) {
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return tmp;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void onion_announce_entry_delete_handler(const void *object, void *arr, uint32_t size)
|
|
||||||
{
|
|
||||||
const Onion_Announce_Entry_Cmp *cmp = (const Onion_Announce_Entry_Cmp *)object;
|
|
||||||
mem_delete(cmp->mem, arr);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const Sort_Funcs onion_announce_entry_cmp_funcs = {
|
|
||||||
onion_announce_entry_less_handler,
|
|
||||||
onion_announce_entry_get_handler,
|
|
||||||
onion_announce_entry_set_handler,
|
|
||||||
onion_announce_entry_subarr_handler,
|
|
||||||
onion_announce_entry_alloc_handler,
|
|
||||||
onion_announce_entry_delete_handler,
|
|
||||||
};
|
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
static void sort_onion_announce_list(const Memory *mem, const Mono_Time *mono_time,
|
static void sort_onion_announce_list(const Memory *mem, const Mono_Time *mono_time,
|
||||||
Onion_Announce_Entry *list, unsigned int length,
|
Onion_Announce_Entry *list, unsigned int length,
|
||||||
const uint8_t *comp_public_key)
|
const uint8_t *comp_public_key)
|
||||||
{
|
{
|
||||||
// Pass comp_public_key to sort with each Onion_Announce_Entry entry, so the
|
// Pass comp_public_key to qsort with each Client_data entry, so the
|
||||||
// comparison function can use it as the base of comparison.
|
// comparison function can use it as the base of comparison.
|
||||||
const Onion_Announce_Entry_Cmp cmp = {
|
Cmp_Data *cmp_list = (Cmp_Data *)mem_valloc(mem, length, sizeof(Cmp_Data));
|
||||||
mem,
|
|
||||||
mono_time,
|
|
||||||
comp_public_key,
|
|
||||||
};
|
|
||||||
|
|
||||||
merge_sort(list, length, &cmp, &onion_announce_entry_cmp_funcs);
|
if (cmp_list == nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < length; ++i) {
|
||||||
|
cmp_list[i].mono_time = mono_time;
|
||||||
|
cmp_list[i].base_public_key = comp_public_key;
|
||||||
|
cmp_list[i].entry = list[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
qsort(cmp_list, length, sizeof(Cmp_Data), cmp_entry);
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < length; ++i) {
|
||||||
|
list[i] = cmp_list[i].entry;
|
||||||
|
}
|
||||||
|
|
||||||
|
mem_delete(mem, cmp_list);
|
||||||
}
|
}
|
||||||
|
|
||||||
/** @brief add entry to entries list
|
/** @brief add entry to entries list
|
||||||
@ -498,7 +455,7 @@ static int handle_announce_request_common(
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int decrypted_len = decrypt_data_symmetric(onion_a->mem, shared_key, packet + 1,
|
const int decrypted_len = decrypt_data_symmetric(shared_key, packet + 1,
|
||||||
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE, plain_size + CRYPTO_MAC_SIZE, plain);
|
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE, plain_size + CRYPTO_MAC_SIZE, plain);
|
||||||
|
|
||||||
if ((uint32_t)decrypted_len != plain_size) {
|
if ((uint32_t)decrypted_len != plain_size) {
|
||||||
@ -585,7 +542,7 @@ static int handle_announce_request_common(
|
|||||||
offset += extra_size;
|
offset += extra_size;
|
||||||
|
|
||||||
uint8_t data[ONION_ANNOUNCE_RESPONSE_MAX_SIZE];
|
uint8_t data[ONION_ANNOUNCE_RESPONSE_MAX_SIZE];
|
||||||
const int len = encrypt_data_symmetric(onion_a->mem, shared_key, nonce, response, offset,
|
const int len = encrypt_data_symmetric(shared_key, nonce, response, offset,
|
||||||
data + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH + CRYPTO_NONCE_SIZE);
|
data + 1 + ONION_ANNOUNCE_SENDBACK_DATA_LENGTH + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
if (len != offset + CRYPTO_MAC_SIZE) {
|
if (len != offset + CRYPTO_MAC_SIZE) {
|
||||||
|
@ -65,7 +65,7 @@ void onion_announce_entry_set_time(Onion_Announce *onion_a, uint32_t entry, uint
|
|||||||
* return packet length on success.
|
* return packet length on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int create_announce_request(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
int create_announce_request(const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *dest_client_id,
|
||||||
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id, const uint8_t *client_id,
|
const uint8_t *public_key, const uint8_t *secret_key, const uint8_t *ping_id, const uint8_t *client_id,
|
||||||
const uint8_t *data_public_key, uint64_t sendback_data);
|
const uint8_t *data_public_key, uint64_t sendback_data);
|
||||||
|
|
||||||
@ -82,7 +82,7 @@ int create_announce_request(const Memory *mem, const Random *rng, uint8_t *packe
|
|||||||
* return 0 on success.
|
* return 0 on success.
|
||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int create_data_request(const Memory *mem, const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *public_key,
|
int create_data_request(const Random *rng, uint8_t *packet, uint16_t max_packet_length, const uint8_t *public_key,
|
||||||
const uint8_t *encrypt_public_key, const uint8_t *nonce, const uint8_t *data, uint16_t length);
|
const uint8_t *encrypt_public_key, const uint8_t *nonce, const uint8_t *data, uint16_t length);
|
||||||
|
|
||||||
/** @brief Create and send an onion announce request packet.
|
/** @brief Create and send an onion announce request packet.
|
||||||
@ -101,7 +101,7 @@ int create_data_request(const Memory *mem, const Random *rng, uint8_t *packet, u
|
|||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int send_announce_request(
|
int send_announce_request(
|
||||||
const Logger *log, const Memory *mem, const Networking_Core *net, const Random *rng,
|
const Logger *log, const Networking_Core *net, const Random *rng,
|
||||||
const Onion_Path *path, const Node_format *dest,
|
const Onion_Path *path, const Node_format *dest,
|
||||||
const uint8_t *public_key, const uint8_t *secret_key,
|
const uint8_t *public_key, const uint8_t *secret_key,
|
||||||
const uint8_t *ping_id, const uint8_t *client_id,
|
const uint8_t *ping_id, const uint8_t *client_id,
|
||||||
@ -125,7 +125,7 @@ int send_announce_request(
|
|||||||
*/
|
*/
|
||||||
non_null()
|
non_null()
|
||||||
int send_data_request(
|
int send_data_request(
|
||||||
const Logger *log, const Memory *mem, const Networking_Core *net, const Random *rng, const Onion_Path *path, const IP_Port *dest,
|
const Logger *log, const Networking_Core *net, const Random *rng, const Onion_Path *path, const IP_Port *dest,
|
||||||
const uint8_t *public_key, const uint8_t *encrypt_public_key, const uint8_t *nonce,
|
const uint8_t *public_key, const uint8_t *encrypt_public_key, const uint8_t *nonce,
|
||||||
const uint8_t *data, uint16_t length);
|
const uint8_t *data, uint16_t length);
|
||||||
|
|
||||||
|
143
external/toxcore/c-toxcore/toxcore/onion_client.c
vendored
143
external/toxcore/c-toxcore/toxcore/onion_client.c
vendored
@ -10,6 +10,7 @@
|
|||||||
#include "onion_client.h"
|
#include "onion_client.h"
|
||||||
|
|
||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#include "DHT.h"
|
#include "DHT.h"
|
||||||
@ -28,7 +29,6 @@
|
|||||||
#include "onion.h"
|
#include "onion.h"
|
||||||
#include "onion_announce.h"
|
#include "onion_announce.h"
|
||||||
#include "ping_array.h"
|
#include "ping_array.h"
|
||||||
#include "sort.h"
|
|
||||||
#include "timed_auth.h"
|
#include "timed_auth.h"
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
@ -528,7 +528,7 @@ static int send_onion_packet_tcp_udp(const Onion_Client *onion_c, const Onion_Pa
|
|||||||
{
|
{
|
||||||
if (net_family_is_ipv4(path->ip_port1.ip.family) || net_family_is_ipv6(path->ip_port1.ip.family)) {
|
if (net_family_is_ipv4(path->ip_port1.ip.family) || net_family_is_ipv6(path->ip_port1.ip.family)) {
|
||||||
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
||||||
const int len = create_onion_packet(onion_c->mem, onion_c->rng, packet, sizeof(packet), path, dest, data, length);
|
const int len = create_onion_packet(onion_c->rng, packet, sizeof(packet), path, dest, data, length);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -545,7 +545,7 @@ static int send_onion_packet_tcp_udp(const Onion_Client *onion_c, const Onion_Pa
|
|||||||
|
|
||||||
if (ip_port_to_tcp_connections_number(&path->ip_port1, &tcp_connections_number)) {
|
if (ip_port_to_tcp_connections_number(&path->ip_port1, &tcp_connections_number)) {
|
||||||
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
uint8_t packet[ONION_MAX_PACKET_SIZE];
|
||||||
const int len = create_onion_packet_tcp(onion_c->mem, onion_c->rng, packet, sizeof(packet), path, dest, data, length);
|
const int len = create_onion_packet_tcp(onion_c->rng, packet, sizeof(packet), path, dest, data, length);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
return -1;
|
return -1;
|
||||||
@ -661,7 +661,7 @@ static int client_send_announce_request(Onion_Client *onion_c, uint32_t num, con
|
|||||||
|
|
||||||
if (num == 0) {
|
if (num == 0) {
|
||||||
len = create_announce_request(
|
len = create_announce_request(
|
||||||
onion_c->mem, onion_c->rng, request, sizeof(request), dest_pubkey, nc_get_self_public_key(onion_c->c),
|
onion_c->rng, request, sizeof(request), dest_pubkey, nc_get_self_public_key(onion_c->c),
|
||||||
nc_get_self_secret_key(onion_c->c), ping_id, nc_get_self_public_key(onion_c->c),
|
nc_get_self_secret_key(onion_c->c), ping_id, nc_get_self_public_key(onion_c->c),
|
||||||
onion_c->temp_public_key, sendback);
|
onion_c->temp_public_key, sendback);
|
||||||
} else {
|
} else {
|
||||||
@ -669,14 +669,14 @@ static int client_send_announce_request(Onion_Client *onion_c, uint32_t num, con
|
|||||||
|
|
||||||
if (onion_friend->gc_data_length == 0) { // contact is a friend
|
if (onion_friend->gc_data_length == 0) { // contact is a friend
|
||||||
len = create_announce_request(
|
len = create_announce_request(
|
||||||
onion_c->mem, onion_c->rng, request, sizeof(request), dest_pubkey, onion_friend->temp_public_key,
|
onion_c->rng, request, sizeof(request), dest_pubkey, onion_friend->temp_public_key,
|
||||||
onion_friend->temp_secret_key, ping_id, onion_friend->real_public_key,
|
onion_friend->temp_secret_key, ping_id, onion_friend->real_public_key,
|
||||||
zero_ping_id, sendback);
|
zero_ping_id, sendback);
|
||||||
} else { // contact is a gc
|
} else { // contact is a gc
|
||||||
onion_friend->is_groupchat = true;
|
onion_friend->is_groupchat = true;
|
||||||
|
|
||||||
len = create_gca_announce_request(
|
len = create_gca_announce_request(
|
||||||
onion_c->mem, onion_c->rng, request, sizeof(request), dest_pubkey, onion_friend->temp_public_key,
|
onion_c->rng, request, sizeof(request), dest_pubkey, onion_friend->temp_public_key,
|
||||||
onion_friend->temp_secret_key, ping_id, onion_friend->real_public_key,
|
onion_friend->temp_secret_key, ping_id, onion_friend->real_public_key,
|
||||||
zero_ping_id, sendback, onion_friend->gc_data,
|
zero_ping_id, sendback, onion_friend->gc_data,
|
||||||
onion_friend->gc_data_length);
|
onion_friend->gc_data_length);
|
||||||
@ -694,17 +694,23 @@ static int client_send_announce_request(Onion_Client *onion_c, uint32_t num, con
|
|||||||
return send_onion_packet_tcp_udp(onion_c, &path, dest, request, len);
|
return send_onion_packet_tcp_udp(onion_c, &path, dest, request, len);
|
||||||
}
|
}
|
||||||
|
|
||||||
typedef struct Onion_Node_Cmp {
|
typedef struct Onion_Client_Cmp_Data {
|
||||||
const Memory *mem;
|
|
||||||
const Mono_Time *mono_time;
|
const Mono_Time *mono_time;
|
||||||
const uint8_t *comp_public_key;
|
const uint8_t *base_public_key;
|
||||||
} Onion_Node_Cmp;
|
Onion_Node entry;
|
||||||
|
} Onion_Client_Cmp_Data;
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
static int onion_node_cmp(const Onion_Node_Cmp *cmp, const Onion_Node *entry1, const Onion_Node *entry2)
|
static int onion_client_cmp_entry(const void *a, const void *b)
|
||||||
{
|
{
|
||||||
const bool t1 = onion_node_timed_out(entry1, cmp->mono_time);
|
const Onion_Client_Cmp_Data *cmp1 = (const Onion_Client_Cmp_Data *)a;
|
||||||
const bool t2 = onion_node_timed_out(entry2, cmp->mono_time);
|
const Onion_Client_Cmp_Data *cmp2 = (const Onion_Client_Cmp_Data *)b;
|
||||||
|
const Onion_Node entry1 = cmp1->entry;
|
||||||
|
const Onion_Node entry2 = cmp2->entry;
|
||||||
|
const uint8_t *cmp_public_key = cmp1->base_public_key;
|
||||||
|
|
||||||
|
const bool t1 = onion_node_timed_out(&entry1, cmp1->mono_time);
|
||||||
|
const bool t2 = onion_node_timed_out(&entry2, cmp2->mono_time);
|
||||||
|
|
||||||
if (t1 && t2) {
|
if (t1 && t2) {
|
||||||
return 0;
|
return 0;
|
||||||
@ -718,7 +724,7 @@ static int onion_node_cmp(const Onion_Node_Cmp *cmp, const Onion_Node *entry1, c
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
const int closest = id_closest(cmp->comp_public_key, entry1->public_key, entry2->public_key);
|
const int closest = id_closest(cmp_public_key, entry1.public_key, entry2.public_key);
|
||||||
|
|
||||||
if (closest == 1) {
|
if (closest == 1) {
|
||||||
return 1;
|
return 1;
|
||||||
@ -731,80 +737,31 @@ static int onion_node_cmp(const Onion_Node_Cmp *cmp, const Onion_Node *entry1, c
|
|||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
non_null()
|
|
||||||
static bool onion_node_less_handler(const void *object, const void *a, const void *b)
|
|
||||||
{
|
|
||||||
const Onion_Node_Cmp *cmp = (const Onion_Node_Cmp *)object;
|
|
||||||
const Onion_Node *entry1 = (const Onion_Node *)a;
|
|
||||||
const Onion_Node *entry2 = (const Onion_Node *)b;
|
|
||||||
|
|
||||||
return onion_node_cmp(cmp, entry1, entry2) < 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static const void *onion_node_get_handler(const void *arr, uint32_t index)
|
|
||||||
{
|
|
||||||
const Onion_Node *entries = (const Onion_Node *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void onion_node_set_handler(void *arr, uint32_t index, const void *val)
|
|
||||||
{
|
|
||||||
Onion_Node *entries = (Onion_Node *)arr;
|
|
||||||
const Onion_Node *entry = (const Onion_Node *)val;
|
|
||||||
entries[index] = *entry;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *onion_node_subarr_handler(void *arr, uint32_t index, uint32_t size)
|
|
||||||
{
|
|
||||||
Onion_Node *entries = (Onion_Node *)arr;
|
|
||||||
return &entries[index];
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void *onion_node_alloc_handler(const void *object, uint32_t size)
|
|
||||||
{
|
|
||||||
const Onion_Node_Cmp *cmp = (const Onion_Node_Cmp *)object;
|
|
||||||
Onion_Node *tmp = (Onion_Node *)mem_valloc(cmp->mem, size, sizeof(Onion_Node));
|
|
||||||
|
|
||||||
if (tmp == nullptr) {
|
|
||||||
return nullptr;
|
|
||||||
}
|
|
||||||
|
|
||||||
return tmp;
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void onion_node_delete_handler(const void *object, void *arr, uint32_t size)
|
|
||||||
{
|
|
||||||
const Onion_Node_Cmp *cmp = (const Onion_Node_Cmp *)object;
|
|
||||||
mem_delete(cmp->mem, arr);
|
|
||||||
}
|
|
||||||
|
|
||||||
static const Sort_Funcs onion_node_cmp_funcs = {
|
|
||||||
onion_node_less_handler,
|
|
||||||
onion_node_get_handler,
|
|
||||||
onion_node_set_handler,
|
|
||||||
onion_node_subarr_handler,
|
|
||||||
onion_node_alloc_handler,
|
|
||||||
onion_node_delete_handler,
|
|
||||||
};
|
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
static void sort_onion_node_list(const Memory *mem, const Mono_Time *mono_time,
|
static void sort_onion_node_list(const Memory *mem, const Mono_Time *mono_time,
|
||||||
Onion_Node *list, unsigned int length, const uint8_t *comp_public_key)
|
Onion_Node *list, unsigned int length, const uint8_t *comp_public_key)
|
||||||
{
|
{
|
||||||
// Pass comp_public_key to sort with each Onion_Node entry, so the
|
// Pass comp_public_key to qsort with each Client_data entry, so the
|
||||||
// comparison function can use it as the base of comparison.
|
// comparison function can use it as the base of comparison.
|
||||||
const Onion_Node_Cmp cmp = {
|
Onion_Client_Cmp_Data *cmp_list = (Onion_Client_Cmp_Data *)mem_valloc(mem, length, sizeof(Onion_Client_Cmp_Data));
|
||||||
mem,
|
|
||||||
mono_time,
|
|
||||||
comp_public_key,
|
|
||||||
};
|
|
||||||
|
|
||||||
merge_sort(list, length, &cmp, &onion_node_cmp_funcs);
|
if (cmp_list == nullptr) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < length; ++i) {
|
||||||
|
cmp_list[i].mono_time = mono_time;
|
||||||
|
cmp_list[i].base_public_key = comp_public_key;
|
||||||
|
cmp_list[i].entry = list[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
qsort(cmp_list, length, sizeof(Onion_Client_Cmp_Data), onion_client_cmp_entry);
|
||||||
|
|
||||||
|
for (uint32_t i = 0; i < length; ++i) {
|
||||||
|
list[i] = cmp_list[i].entry;
|
||||||
|
}
|
||||||
|
|
||||||
|
mem_delete(mem, cmp_list);
|
||||||
}
|
}
|
||||||
|
|
||||||
non_null()
|
non_null()
|
||||||
@ -1005,7 +962,7 @@ static int handle_announce_response(void *object, const IP_Port *source, const u
|
|||||||
const uint16_t ciphertext_size = length - ciphertext_start;
|
const uint16_t ciphertext_size = length - ciphertext_start;
|
||||||
|
|
||||||
if (num == 0) {
|
if (num == 0) {
|
||||||
len = decrypt_data(onion_c->mem, public_key, nc_get_self_secret_key(onion_c->c),
|
len = decrypt_data(public_key, nc_get_self_secret_key(onion_c->c),
|
||||||
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
||||||
} else {
|
} else {
|
||||||
if (!onion_c->friends_list[num - 1].is_valid) {
|
if (!onion_c->friends_list[num - 1].is_valid) {
|
||||||
@ -1013,7 +970,7 @@ static int handle_announce_response(void *object, const IP_Port *source, const u
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
len = decrypt_data(onion_c->mem, public_key, onion_c->friends_list[num - 1].temp_secret_key,
|
len = decrypt_data(public_key, onion_c->friends_list[num - 1].temp_secret_key,
|
||||||
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1109,7 +1066,7 @@ static int handle_announce_response_old(void *object, const IP_Port *source, con
|
|||||||
const uint16_t ciphertext_size = length - ciphertext_start;
|
const uint16_t ciphertext_size = length - ciphertext_start;
|
||||||
|
|
||||||
if (num == 0) {
|
if (num == 0) {
|
||||||
len = decrypt_data(onion_c->mem, public_key, nc_get_self_secret_key(onion_c->c),
|
len = decrypt_data(public_key, nc_get_self_secret_key(onion_c->c),
|
||||||
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
||||||
} else {
|
} else {
|
||||||
if (!onion_c->friends_list[num - 1].is_valid) {
|
if (!onion_c->friends_list[num - 1].is_valid) {
|
||||||
@ -1117,7 +1074,7 @@ static int handle_announce_response_old(void *object, const IP_Port *source, con
|
|||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
len = decrypt_data(onion_c->mem, public_key, onion_c->friends_list[num - 1].temp_secret_key,
|
len = decrypt_data(public_key, onion_c->friends_list[num - 1].temp_secret_key,
|
||||||
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
&packet[nonce_start], &packet[ciphertext_start], ciphertext_size, plain);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1179,7 +1136,7 @@ static int handle_data_response(void *object, const IP_Port *source, const uint8
|
|||||||
|
|
||||||
const uint16_t temp_plain_size = length - ONION_DATA_RESPONSE_MIN_SIZE;
|
const uint16_t temp_plain_size = length - ONION_DATA_RESPONSE_MIN_SIZE;
|
||||||
VLA(uint8_t, temp_plain, temp_plain_size);
|
VLA(uint8_t, temp_plain, temp_plain_size);
|
||||||
int len = decrypt_data(onion_c->mem, packet + 1 + CRYPTO_NONCE_SIZE, onion_c->temp_secret_key, packet + 1,
|
int len = decrypt_data(packet + 1 + CRYPTO_NONCE_SIZE, onion_c->temp_secret_key, packet + 1,
|
||||||
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE), temp_plain);
|
length - (1 + CRYPTO_NONCE_SIZE + CRYPTO_PUBLIC_KEY_SIZE), temp_plain);
|
||||||
|
|
||||||
@ -1189,7 +1146,7 @@ static int handle_data_response(void *object, const IP_Port *source, const uint8
|
|||||||
|
|
||||||
const uint16_t plain_size = temp_plain_size - DATA_IN_RESPONSE_MIN_SIZE;
|
const uint16_t plain_size = temp_plain_size - DATA_IN_RESPONSE_MIN_SIZE;
|
||||||
VLA(uint8_t, plain, plain_size);
|
VLA(uint8_t, plain, plain_size);
|
||||||
len = decrypt_data(onion_c->mem, temp_plain, nc_get_self_secret_key(onion_c->c),
|
len = decrypt_data(temp_plain, nc_get_self_secret_key(onion_c->c),
|
||||||
packet + 1, temp_plain + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1, temp_plain + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
temp_plain_size - CRYPTO_PUBLIC_KEY_SIZE, plain);
|
temp_plain_size - CRYPTO_PUBLIC_KEY_SIZE, plain);
|
||||||
|
|
||||||
@ -1348,7 +1305,7 @@ int send_onion_data(Onion_Client *onion_c, int friend_num, const uint8_t *data,
|
|||||||
const uint16_t packet_size = DATA_IN_RESPONSE_MIN_SIZE + length;
|
const uint16_t packet_size = DATA_IN_RESPONSE_MIN_SIZE + length;
|
||||||
VLA(uint8_t, packet, packet_size);
|
VLA(uint8_t, packet, packet_size);
|
||||||
memcpy(packet, nc_get_self_public_key(onion_c->c), CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(packet, nc_get_self_public_key(onion_c->c), CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
int len = encrypt_data(onion_c->mem, onion_c->friends_list[friend_num].real_public_key,
|
int len = encrypt_data(onion_c->friends_list[friend_num].real_public_key,
|
||||||
nc_get_self_secret_key(onion_c->c), nonce, data,
|
nc_get_self_secret_key(onion_c->c), nonce, data,
|
||||||
length, packet + CRYPTO_PUBLIC_KEY_SIZE);
|
length, packet + CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
|
|
||||||
@ -1367,7 +1324,7 @@ int send_onion_data(Onion_Client *onion_c, int friend_num, const uint8_t *data,
|
|||||||
|
|
||||||
uint8_t o_packet[ONION_MAX_PACKET_SIZE];
|
uint8_t o_packet[ONION_MAX_PACKET_SIZE];
|
||||||
len = create_data_request(
|
len = create_data_request(
|
||||||
onion_c->mem, onion_c->rng, o_packet, sizeof(o_packet), onion_c->friends_list[friend_num].real_public_key,
|
onion_c->rng, o_packet, sizeof(o_packet), onion_c->friends_list[friend_num].real_public_key,
|
||||||
node_list[good_nodes[i]].data_public_key, nonce, packet, packet_size);
|
node_list[good_nodes[i]].data_public_key, nonce, packet, packet_size);
|
||||||
|
|
||||||
if (len == -1) {
|
if (len == -1) {
|
||||||
@ -1407,7 +1364,7 @@ static int send_dht_dhtpk(const Onion_Client *onion_c, int friend_num, const uin
|
|||||||
VLA(uint8_t, temp, temp_size);
|
VLA(uint8_t, temp, temp_size);
|
||||||
memcpy(temp, nc_get_self_public_key(onion_c->c), CRYPTO_PUBLIC_KEY_SIZE);
|
memcpy(temp, nc_get_self_public_key(onion_c->c), CRYPTO_PUBLIC_KEY_SIZE);
|
||||||
memcpy(temp + CRYPTO_PUBLIC_KEY_SIZE, nonce, CRYPTO_NONCE_SIZE);
|
memcpy(temp + CRYPTO_PUBLIC_KEY_SIZE, nonce, CRYPTO_NONCE_SIZE);
|
||||||
int len = encrypt_data(onion_c->mem, onion_c->friends_list[friend_num].real_public_key,
|
int len = encrypt_data(onion_c->friends_list[friend_num].real_public_key,
|
||||||
nc_get_self_secret_key(onion_c->c), nonce, data,
|
nc_get_self_secret_key(onion_c->c), nonce, data,
|
||||||
length, temp + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
length, temp + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
|
|
||||||
@ -1417,7 +1374,7 @@ static int send_dht_dhtpk(const Onion_Client *onion_c, int friend_num, const uin
|
|||||||
|
|
||||||
uint8_t packet_data[MAX_CRYPTO_REQUEST_SIZE];
|
uint8_t packet_data[MAX_CRYPTO_REQUEST_SIZE];
|
||||||
len = create_request(
|
len = create_request(
|
||||||
onion_c->mem, onion_c->rng, dht_get_self_public_key(onion_c->dht), dht_get_self_secret_key(onion_c->dht), packet_data,
|
onion_c->rng, dht_get_self_public_key(onion_c->dht), dht_get_self_secret_key(onion_c->dht), packet_data,
|
||||||
onion_c->friends_list[friend_num].dht_public_key, temp, temp_size, CRYPTO_PACKET_DHTPK);
|
onion_c->friends_list[friend_num].dht_public_key, temp, temp_size, CRYPTO_PACKET_DHTPK);
|
||||||
assert(len <= UINT16_MAX);
|
assert(len <= UINT16_MAX);
|
||||||
const Packet packet = {packet_data, (uint16_t)len};
|
const Packet packet = {packet_data, (uint16_t)len};
|
||||||
@ -1444,7 +1401,7 @@ static int handle_dht_dhtpk(void *object, const IP_Port *source, const uint8_t *
|
|||||||
}
|
}
|
||||||
|
|
||||||
uint8_t plain[DHTPK_DATA_MAX_LENGTH];
|
uint8_t plain[DHTPK_DATA_MAX_LENGTH];
|
||||||
const int len = decrypt_data(onion_c->mem, packet, nc_get_self_secret_key(onion_c->c),
|
const int len = decrypt_data(packet, nc_get_self_secret_key(onion_c->c),
|
||||||
packet + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
packet + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
length - (CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE), plain);
|
length - (CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE), plain);
|
||||||
|
10
external/toxcore/c-toxcore/toxcore/ping.c
vendored
10
external/toxcore/c-toxcore/toxcore/ping.c
vendored
@ -31,7 +31,6 @@
|
|||||||
struct Ping {
|
struct Ping {
|
||||||
const Mono_Time *mono_time;
|
const Mono_Time *mono_time;
|
||||||
const Random *rng;
|
const Random *rng;
|
||||||
const Memory *mem;
|
|
||||||
DHT *dht;
|
DHT *dht;
|
||||||
|
|
||||||
Ping_Array *ping_array;
|
Ping_Array *ping_array;
|
||||||
@ -73,7 +72,7 @@ void ping_send_request(Ping *ping, const IP_Port *ipp, const uint8_t *public_key
|
|||||||
pk_copy(pk + 1, dht_get_self_public_key(ping->dht)); // Our pubkey
|
pk_copy(pk + 1, dht_get_self_public_key(ping->dht)); // Our pubkey
|
||||||
random_nonce(ping->rng, pk + 1 + CRYPTO_PUBLIC_KEY_SIZE); // Generate new nonce
|
random_nonce(ping->rng, pk + 1 + CRYPTO_PUBLIC_KEY_SIZE); // Generate new nonce
|
||||||
|
|
||||||
rc = encrypt_data_symmetric(ping->mem, shared_key,
|
rc = encrypt_data_symmetric(shared_key,
|
||||||
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
ping_plain, sizeof(ping_plain),
|
ping_plain, sizeof(ping_plain),
|
||||||
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
@ -105,7 +104,7 @@ static int ping_send_response(const Ping *ping, const IP_Port *ipp, const uint8_
|
|||||||
random_nonce(ping->rng, pk + 1 + CRYPTO_PUBLIC_KEY_SIZE); // Generate new nonce
|
random_nonce(ping->rng, pk + 1 + CRYPTO_PUBLIC_KEY_SIZE); // Generate new nonce
|
||||||
|
|
||||||
// Encrypt ping_id using recipient privkey
|
// Encrypt ping_id using recipient privkey
|
||||||
const int rc = encrypt_data_symmetric(ping->mem, shared_encryption_key,
|
const int rc = encrypt_data_symmetric(shared_encryption_key,
|
||||||
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
ping_plain, sizeof(ping_plain),
|
ping_plain, sizeof(ping_plain),
|
||||||
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
pk + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE);
|
||||||
@ -138,7 +137,7 @@ static int handle_ping_request(void *object, const IP_Port *source, const uint8_
|
|||||||
uint8_t ping_plain[PING_PLAIN_SIZE];
|
uint8_t ping_plain[PING_PLAIN_SIZE];
|
||||||
|
|
||||||
// Decrypt ping_id
|
// Decrypt ping_id
|
||||||
const int rc = decrypt_data_symmetric(ping->mem, shared_key,
|
const int rc = decrypt_data_symmetric(shared_key,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
PING_PLAIN_SIZE + CRYPTO_MAC_SIZE,
|
PING_PLAIN_SIZE + CRYPTO_MAC_SIZE,
|
||||||
@ -183,7 +182,7 @@ static int handle_ping_response(void *object, const IP_Port *source, const uint8
|
|||||||
|
|
||||||
uint8_t ping_plain[PING_PLAIN_SIZE];
|
uint8_t ping_plain[PING_PLAIN_SIZE];
|
||||||
// Decrypt ping_id
|
// Decrypt ping_id
|
||||||
rc = decrypt_data_symmetric(ping->mem, shared_key,
|
rc = decrypt_data_symmetric(shared_key,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE,
|
||||||
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
packet + 1 + CRYPTO_PUBLIC_KEY_SIZE + CRYPTO_NONCE_SIZE,
|
||||||
PING_PLAIN_SIZE + CRYPTO_MAC_SIZE,
|
PING_PLAIN_SIZE + CRYPTO_MAC_SIZE,
|
||||||
@ -349,7 +348,6 @@ Ping *ping_new(const Memory *mem, const Mono_Time *mono_time, const Random *rng,
|
|||||||
|
|
||||||
ping->mono_time = mono_time;
|
ping->mono_time = mono_time;
|
||||||
ping->rng = rng;
|
ping->rng = rng;
|
||||||
ping->mem = mem;
|
|
||||||
ping->dht = dht;
|
ping->dht = dht;
|
||||||
networking_registerhandler(dht_get_net(ping->dht), NET_PACKET_PING_REQUEST, &handle_ping_request, dht);
|
networking_registerhandler(dht_get_net(ping->dht), NET_PACKET_PING_REQUEST, &handle_ping_request, dht);
|
||||||
networking_registerhandler(dht_get_net(ping->dht), NET_PACKET_PING_RESPONSE, &handle_ping_response, dht);
|
networking_registerhandler(dht_get_net(ping->dht), NET_PACKET_PING_RESPONSE, &handle_ping_response, dht);
|
||||||
|
182
external/toxcore/c-toxcore/toxcore/sort.c
vendored
182
external/toxcore/c-toxcore/toxcore/sort.c
vendored
@ -1,182 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
* Copyright © 2023-2024 The TokTok team.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "sort.h"
|
|
||||||
|
|
||||||
#include <assert.h>
|
|
||||||
|
|
||||||
#include "attributes.h"
|
|
||||||
#include "ccompat.h"
|
|
||||||
#include "util.h"
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Threshold for when to switch to insertion sort.
|
|
||||||
*
|
|
||||||
* This is a trade-off between the complexity of insertion sort and the
|
|
||||||
* overhead of merge sort. The threshold is chosen to be the smallest value
|
|
||||||
* that gives a measurable speedup for insertion sort over merge sort. This is
|
|
||||||
* based on measurements done in sort_bench.cc. Starting from 32 elements,
|
|
||||||
* merge sort is faster than insertion sort in all our tests (both unsorted
|
|
||||||
* and mostly-sorted).
|
|
||||||
*
|
|
||||||
* Toxcore has a lot of small arrays it wants to sort, so this optimisation
|
|
||||||
* makes sense.
|
|
||||||
*/
|
|
||||||
#define SMALL_ARRAY_THRESHOLD 16
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void merge_sort_merge_back(
|
|
||||||
void *arr,
|
|
||||||
const void *l_arr, uint32_t l_arr_size,
|
|
||||||
const void *r_arr, uint32_t r_arr_size,
|
|
||||||
uint32_t left_start,
|
|
||||||
const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
uint32_t li = 0;
|
|
||||||
uint32_t ri = 0;
|
|
||||||
uint32_t k = left_start;
|
|
||||||
|
|
||||||
while (li < l_arr_size && ri < r_arr_size) {
|
|
||||||
const void *l = funcs->get_callback(l_arr, li);
|
|
||||||
const void *r = funcs->get_callback(r_arr, ri);
|
|
||||||
// !(r < l) <=> (r >= l) <=> (l <= r)
|
|
||||||
if (!funcs->less_callback(object, r, l)) {
|
|
||||||
funcs->set_callback(arr, k, l);
|
|
||||||
++li;
|
|
||||||
} else {
|
|
||||||
funcs->set_callback(arr, k, r);
|
|
||||||
++ri;
|
|
||||||
}
|
|
||||||
++k;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Copy the remaining elements of `l_arr[]`, if there are any. */
|
|
||||||
while (li < l_arr_size) {
|
|
||||||
funcs->set_callback(arr, k, funcs->get_callback(l_arr, li));
|
|
||||||
++li;
|
|
||||||
++k;
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Copy the remaining elements of `r_arr[]`, if there are any. */
|
|
||||||
while (ri < r_arr_size) {
|
|
||||||
funcs->set_callback(arr, k, funcs->get_callback(r_arr, ri));
|
|
||||||
++ri;
|
|
||||||
++k;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Function to merge the two haves `arr[left_start..mid]` and `arr[mid+1..right_end]` of array `arr[]`. */
|
|
||||||
non_null()
|
|
||||||
static void merge_sort_merge(
|
|
||||||
void *arr, uint32_t left_start, uint32_t mid, uint32_t right_end, void *tmp,
|
|
||||||
const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
const uint32_t l_arr_size = mid - left_start + 1;
|
|
||||||
const uint32_t r_arr_size = right_end - mid;
|
|
||||||
|
|
||||||
/* Temporary arrays, using the tmp buffer created in `merge_sort` below. */
|
|
||||||
void *l_arr = funcs->subarr_callback(tmp, 0, l_arr_size);
|
|
||||||
void *r_arr = funcs->subarr_callback(tmp, l_arr_size, r_arr_size);
|
|
||||||
|
|
||||||
/* Copy data to temp arrays `l_arr[]` and `r_arr[]`.
|
|
||||||
*
|
|
||||||
* This is iterating and repeatedly calling `get` and `set`, which sounds
|
|
||||||
* slow, but is only marginally slower than having a `copy` callback. With
|
|
||||||
* a `copy` callback, we'd save 3-4% in time.
|
|
||||||
*/
|
|
||||||
for (uint32_t i = 0; i < l_arr_size; ++i) {
|
|
||||||
funcs->set_callback(l_arr, i, funcs->get_callback(arr, left_start + i));
|
|
||||||
}
|
|
||||||
for (uint32_t i = 0; i < r_arr_size; ++i) {
|
|
||||||
funcs->set_callback(r_arr, i, funcs->get_callback(arr, mid + 1 + i));
|
|
||||||
}
|
|
||||||
|
|
||||||
/* Merge the temp arrays back into `arr[left_start..right_end]`. */
|
|
||||||
merge_sort_merge_back(arr, l_arr, l_arr_size, r_arr, r_arr_size, left_start, object, funcs);
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void insertion_sort_step(void *arr, void *tmp, uint32_t i, const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
funcs->set_callback(tmp, 0, funcs->get_callback(arr, i));
|
|
||||||
uint32_t j = i;
|
|
||||||
|
|
||||||
while (j > 0) {
|
|
||||||
if (!funcs->less_callback(object, tmp, funcs->get_callback(arr, j - 1))) {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
funcs->set_callback(arr, j, funcs->get_callback(arr, j - 1));
|
|
||||||
--j;
|
|
||||||
}
|
|
||||||
|
|
||||||
funcs->set_callback(arr, j, tmp);
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static void insertion_sort_with_buf(void *arr, uint32_t arr_size, void *tmp, uint32_t tmp_size, const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
for (uint32_t i = 1; i < arr_size; ++i) {
|
|
||||||
insertion_sort_step(arr, tmp, i, object, funcs);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
non_null()
|
|
||||||
static bool insertion_sort(void *arr, uint32_t arr_size, const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
void *tmp = funcs->alloc_callback(object, 1);
|
|
||||||
|
|
||||||
if (tmp == nullptr) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
insertion_sort_with_buf(arr, arr_size, tmp, 1, object, funcs);
|
|
||||||
|
|
||||||
funcs->delete_callback(object, tmp, 1);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
void merge_sort_with_buf(void *arr, uint32_t arr_size, void *tmp, uint32_t tmp_size, const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
assert(tmp_size >= arr_size);
|
|
||||||
|
|
||||||
if (arr_size <= SMALL_ARRAY_THRESHOLD) {
|
|
||||||
assert(tmp_size >= 1);
|
|
||||||
insertion_sort_with_buf(arr, arr_size, tmp, tmp_size, object, funcs);
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Merge subarrays in bottom up manner. First merge subarrays of
|
|
||||||
// size 1 to create sorted subarrays of size 2, then merge subarrays
|
|
||||||
// of size 2 to create sorted subarrays of size 4, and so on.
|
|
||||||
for (uint32_t curr_size = 1; curr_size <= arr_size - 1; curr_size = 2 * curr_size) {
|
|
||||||
// Pick starting point of different subarrays of current size
|
|
||||||
for (uint32_t left_start = 0; left_start < arr_size - 1; left_start += 2 * curr_size) {
|
|
||||||
// Find ending point of left subarray. mid+1 is starting
|
|
||||||
// point of right
|
|
||||||
const uint32_t mid = min_u32(left_start + curr_size - 1, arr_size - 1);
|
|
||||||
const uint32_t right_end = min_u32(left_start + 2 * curr_size - 1, arr_size - 1);
|
|
||||||
|
|
||||||
// Merge Subarrays arr[left_start...mid] & arr[mid+1...right_end]
|
|
||||||
merge_sort_merge(arr, left_start, mid, right_end, tmp, object, funcs);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
bool merge_sort(void *arr, uint32_t arr_size, const void *object, const Sort_Funcs *funcs)
|
|
||||||
{
|
|
||||||
if (arr_size <= SMALL_ARRAY_THRESHOLD) {
|
|
||||||
return insertion_sort(arr, arr_size, object, funcs);
|
|
||||||
}
|
|
||||||
|
|
||||||
void *tmp = funcs->alloc_callback(object, arr_size);
|
|
||||||
|
|
||||||
if (tmp == nullptr) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
merge_sort_with_buf(arr, arr_size, tmp, arr_size, object, funcs);
|
|
||||||
|
|
||||||
funcs->delete_callback(object, tmp, arr_size);
|
|
||||||
return true;
|
|
||||||
}
|
|
116
external/toxcore/c-toxcore/toxcore/sort.h
vendored
116
external/toxcore/c-toxcore/toxcore/sort.h
vendored
@ -1,116 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
* Copyright © 2023-2024 The TokTok team.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef C_TOXCORE_TOXCORE_SORT_H
|
|
||||||
#define C_TOXCORE_TOXCORE_SORT_H
|
|
||||||
|
|
||||||
#include <stdbool.h>
|
|
||||||
#include <stdint.h>
|
|
||||||
|
|
||||||
#include "attributes.h"
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
extern "C" {
|
|
||||||
#endif
|
|
||||||
|
|
||||||
/** @brief Compare elements with a less-than ordering: `a < b`. */
|
|
||||||
typedef bool sort_less_cb(const void *object, const void *a, const void *b);
|
|
||||||
/** @brief Get element from array at index. */
|
|
||||||
typedef const void *sort_get_cb(const void *arr, uint32_t index);
|
|
||||||
/** @brief Set element in array at index to new value (perform copy). */
|
|
||||||
typedef void sort_set_cb(void *arr, uint32_t index, const void *val);
|
|
||||||
/** @brief Get a sub-array at an index of a given size (mutable pointer).
|
|
||||||
*
|
|
||||||
* Used to index in the temporary array allocated by `sort_alloc_cb` and get
|
|
||||||
* a sub-array for working memory.
|
|
||||||
*/
|
|
||||||
typedef void *sort_subarr_cb(void *arr, uint32_t index, uint32_t size);
|
|
||||||
/** @brief Allocate a new array of the element type.
|
|
||||||
*
|
|
||||||
* @param size The array size in elements of type T (not byte size). This value
|
|
||||||
* is always exactly the input array size as passed to `merge_sort`.
|
|
||||||
*/
|
|
||||||
typedef void *sort_alloc_cb(const void *object, uint32_t size);
|
|
||||||
/** @brief Free the element type array. */
|
|
||||||
typedef void sort_delete_cb(const void *object, void *arr, uint32_t size);
|
|
||||||
|
|
||||||
/** @brief Virtual function table for getting/setting elements in an array and
|
|
||||||
* comparing them.
|
|
||||||
*
|
|
||||||
* Only the `less`, `alloc`, and `delete` functions get a `this`-pointer. We
|
|
||||||
* assume that indexing in an array doesn't need any other information than the
|
|
||||||
* array itself.
|
|
||||||
*
|
|
||||||
* For now, the `this`-pointer is const, because we assume sorting doesn't need
|
|
||||||
* to mutate any state, but if necessary that can be changed in the future.
|
|
||||||
*/
|
|
||||||
typedef struct Sort_Funcs {
|
|
||||||
sort_less_cb *less_callback;
|
|
||||||
sort_get_cb *get_callback;
|
|
||||||
sort_set_cb *set_callback;
|
|
||||||
sort_subarr_cb *subarr_callback;
|
|
||||||
sort_alloc_cb *alloc_callback;
|
|
||||||
sort_delete_cb *delete_callback;
|
|
||||||
} Sort_Funcs;
|
|
||||||
|
|
||||||
/** @brief Non-recursive merge sort function to sort `arr[0...arr_size-1]`.
|
|
||||||
*
|
|
||||||
* Avoids `memcpy` and avoids treating elements as byte arrays. Instead, uses
|
|
||||||
* callbacks to index in arrays and copy elements. This makes it quite a bit
|
|
||||||
* slower than `qsort`, but works with elements that require special care when
|
|
||||||
* being copied (e.g. if they are part of a graph or other data structure that
|
|
||||||
* with pointers or other invariants).
|
|
||||||
*
|
|
||||||
* This function actually uses insertion sort for small arrays (up to 16
|
|
||||||
* elements), which is faster than merge sort for small arrays, especially
|
|
||||||
* when mostly sorted (a common use case in toxcore).
|
|
||||||
*
|
|
||||||
* Allocates a single temporary array with the provided alloc callback, and
|
|
||||||
* frees it at the end. This is significantly faster than an in-place
|
|
||||||
* implementation.
|
|
||||||
*
|
|
||||||
* Complexity:
|
|
||||||
* - Space: `O(n) where n = array_size`.
|
|
||||||
* - Time: `O(n * log n) where n = array_size`.
|
|
||||||
*
|
|
||||||
* Compared to `qsort`, this is about 60-70% slower for large arrays. For small
|
|
||||||
* arrays (up to 16 elements), it's about 50% faster than `qsort`.
|
|
||||||
*
|
|
||||||
* @param[in,out] arr An array of type T.
|
|
||||||
* @param arr_size Number of elements in @p arr (count, not byte size).
|
|
||||||
* @param[in] object Comparator object.
|
|
||||||
* @param[in] funcs Callback struct for elements of type T.
|
|
||||||
*/
|
|
||||||
non_null()
|
|
||||||
bool merge_sort(void *arr, uint32_t arr_size, const void *object, const Sort_Funcs *funcs);
|
|
||||||
|
|
||||||
/**
|
|
||||||
* @brief Merge sort like above but with a pre-allocated buffer.
|
|
||||||
*
|
|
||||||
* This function is the same as `merge_sort` but uses a pre-allocated buffer
|
|
||||||
* for temporary storage. This can be useful if the caller wants to avoid
|
|
||||||
* dynamic memory allocation.
|
|
||||||
*
|
|
||||||
* This function is 1-2% faster than `merge_sort` for small arrays up to 1000
|
|
||||||
* elements, and about 5-10% faster for large arrays (2000+ elements).
|
|
||||||
*
|
|
||||||
* The main upside is that `alloc` and `delete` callbacks don't need to be
|
|
||||||
* implemented, and the caller can use a stack-allocated buffer.
|
|
||||||
*
|
|
||||||
* @param[in,out] arr An array of type T.
|
|
||||||
* @param arr_size Number of elements in @p arr (count, not byte size).
|
|
||||||
* @param[in,out] tmp A buffer of size `tmp_size` for temporary storage.
|
|
||||||
* @param tmp_size Number of elements in @p tmp (count, not byte size). Must be
|
|
||||||
* at least as large as `arr_size`.
|
|
||||||
* @param[in] object Comparator object.
|
|
||||||
* @param[in] funcs Callback struct for elements of type T.
|
|
||||||
*/
|
|
||||||
non_null()
|
|
||||||
void merge_sort_with_buf(void *arr, uint32_t arr_size, void *tmp, uint32_t tmp_size, const void *object, const Sort_Funcs *funcs);
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
|
||||||
} /* extern "C" */
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif /* C_TOXCORE_TOXCORE_SORT_H */
|
|
140
external/toxcore/c-toxcore/toxcore/sort_bench.cc
vendored
140
external/toxcore/c-toxcore/toxcore/sort_bench.cc
vendored
@ -1,140 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
* Copyright © 2023-2024 The TokTok team.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include <benchmark/benchmark.h>
|
|
||||||
|
|
||||||
#include <algorithm>
|
|
||||||
#include <array>
|
|
||||||
#include <cstdint>
|
|
||||||
#include <random>
|
|
||||||
|
|
||||||
#include "mem.h"
|
|
||||||
#include "sort.h"
|
|
||||||
#include "sort_test_util.hh"
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
std::pair<std::vector<Some_Type>, std::mt19937> random_vec(benchmark::State &state)
|
|
||||||
{
|
|
||||||
std::mt19937 rng;
|
|
||||||
// INT_MAX-1 so later we have room to add 1 larger element if needed.
|
|
||||||
std::uniform_int_distribution<uint32_t> dist{
|
|
||||||
std::numeric_limits<uint32_t>::min(), std::numeric_limits<uint32_t>::max() - 1};
|
|
||||||
|
|
||||||
std::vector<Some_Type> vec(state.range(0));
|
|
||||||
std::generate(std::begin(vec), std::end(vec), [&]() {
|
|
||||||
std::array<uint32_t, 8> compare_value;
|
|
||||||
std::generate(
|
|
||||||
std::begin(compare_value), std::end(compare_value), [&]() { return dist(rng); });
|
|
||||||
return Some_Type{nullptr, compare_value, "hello there"};
|
|
||||||
});
|
|
||||||
|
|
||||||
return {vec, rng};
|
|
||||||
}
|
|
||||||
|
|
||||||
std::vector<Some_Type> mostly_sorted_vec(benchmark::State &state)
|
|
||||||
{
|
|
||||||
auto [vec, rng] = random_vec(state);
|
|
||||||
std::sort(vec.begin(), vec.end());
|
|
||||||
|
|
||||||
// Randomly swap 5% of the vector.
|
|
||||||
std::uniform_int_distribution<std::size_t> dist{0, vec.size() - 1};
|
|
||||||
for (std::size_t i = 0; i < vec.size() / 20; ++i) {
|
|
||||||
const auto a = dist(rng);
|
|
||||||
const auto b = dist(rng);
|
|
||||||
std::swap(vec[a], vec[b]);
|
|
||||||
}
|
|
||||||
|
|
||||||
return vec;
|
|
||||||
}
|
|
||||||
|
|
||||||
void BM_merge_sort(benchmark::State &state)
|
|
||||||
{
|
|
||||||
const auto vec = random_vec(state).first;
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
merge_sort(unsorted.data(), unsorted.size(), &state, &Some_Type::funcs);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_merge_sort)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
void BM_merge_sort_with_buf(benchmark::State &state)
|
|
||||||
{
|
|
||||||
const auto vec = random_vec(state).first;
|
|
||||||
std::vector<Some_Type> buf(vec.size());
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
merge_sort_with_buf(
|
|
||||||
unsorted.data(), unsorted.size(), buf.data(), buf.size(), &state, &Some_Type::funcs);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_merge_sort_with_buf)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
void BM_merge_sort_mostly_sorted(benchmark::State &state)
|
|
||||||
{
|
|
||||||
auto vec = mostly_sorted_vec(state);
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
merge_sort(unsorted.data(), unsorted.size(), &state, &Some_Type::funcs);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_merge_sort_mostly_sorted)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
void BM_qsort(benchmark::State &state)
|
|
||||||
{
|
|
||||||
const auto vec = random_vec(state).first;
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
qsort(unsorted.data(), unsorted.size(), sizeof(unsorted[0]), my_type_cmp);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_qsort)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
void BM_qsort_mostly_sorted(benchmark::State &state)
|
|
||||||
{
|
|
||||||
auto vec = mostly_sorted_vec(state);
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
qsort(unsorted.data(), unsorted.size(), sizeof(unsorted[0]), my_type_cmp);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_qsort_mostly_sorted)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
void BM_std_sort(benchmark::State &state)
|
|
||||||
{
|
|
||||||
const auto vec = random_vec(state).first;
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
std::sort(unsorted.begin(), unsorted.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_std_sort)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
void BM_std_sort_mostly_sorted(benchmark::State &state)
|
|
||||||
{
|
|
||||||
auto vec = mostly_sorted_vec(state);
|
|
||||||
|
|
||||||
for (auto _ : state) {
|
|
||||||
auto unsorted = vec;
|
|
||||||
std::sort(unsorted.begin(), unsorted.end());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK(BM_std_sort_mostly_sorted)->RangeMultiplier(2)->Range(8, 8 << 8);
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
BENCHMARK_MAIN();
|
|
79
external/toxcore/c-toxcore/toxcore/sort_test.cc
vendored
79
external/toxcore/c-toxcore/toxcore/sort_test.cc
vendored
@ -1,79 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
* Copyright © 2023-2024 The TokTok team.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#include "sort.h"
|
|
||||||
|
|
||||||
#include <gtest/gtest.h>
|
|
||||||
|
|
||||||
#include <limits>
|
|
||||||
#include <random>
|
|
||||||
|
|
||||||
#include "sort_test_util.hh"
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
TEST(MergeSort, BehavesLikeStdSort)
|
|
||||||
{
|
|
||||||
std::mt19937 rng;
|
|
||||||
// INT_MAX-1 so later we have room to add 1 larger element if needed.
|
|
||||||
std::uniform_int_distribution<int> dist{
|
|
||||||
std::numeric_limits<int>::min(), std::numeric_limits<int>::max() - 1};
|
|
||||||
|
|
||||||
constexpr auto int_funcs = sort_funcs<int>();
|
|
||||||
|
|
||||||
// Test with int arrays.
|
|
||||||
for (uint32_t i = 1; i < 500; ++i) {
|
|
||||||
std::vector<int> vec(i);
|
|
||||||
std::generate(std::begin(vec), std::end(vec), [&]() { return dist(rng); });
|
|
||||||
|
|
||||||
auto sorted = vec;
|
|
||||||
std::sort(sorted.begin(), sorted.end(), std::less<int>());
|
|
||||||
|
|
||||||
// If vec was accidentally sorted, add another larger element that almost definitely makes
|
|
||||||
// it not sorted.
|
|
||||||
if (vec == sorted) {
|
|
||||||
int const largest = *std::prev(sorted.end()) + 1;
|
|
||||||
sorted.push_back(largest);
|
|
||||||
vec.insert(vec.begin(), largest);
|
|
||||||
}
|
|
||||||
ASSERT_NE(vec, sorted);
|
|
||||||
|
|
||||||
// Just pass some arbitrary "self" to make sure the callbacks pass it through.
|
|
||||||
ASSERT_TRUE(merge_sort(vec.data(), vec.size(), &i, &int_funcs));
|
|
||||||
ASSERT_EQ(vec, sorted);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
TEST(MergeSort, WorksWithNonTrivialTypes)
|
|
||||||
{
|
|
||||||
std::mt19937 rng;
|
|
||||||
std::uniform_int_distribution<int> dist{
|
|
||||||
std::numeric_limits<int>::min(), std::numeric_limits<int>::max()};
|
|
||||||
|
|
||||||
constexpr auto string_funcs = sort_funcs<std::string>();
|
|
||||||
|
|
||||||
// Test with std::string arrays.
|
|
||||||
for (uint32_t i = 1; i < 500; ++i) {
|
|
||||||
std::vector<std::string> vec(i);
|
|
||||||
std::generate(std::begin(vec), std::end(vec), [&]() { return std::to_string(dist(rng)); });
|
|
||||||
|
|
||||||
auto sorted = vec;
|
|
||||||
std::sort(sorted.begin(), sorted.end(), std::less<std::string>());
|
|
||||||
|
|
||||||
// If vec was accidentally sorted, add another larger element that almost definitely makes
|
|
||||||
// it not sorted.
|
|
||||||
if (vec == sorted) {
|
|
||||||
std::string const largest = "larger than largest int";
|
|
||||||
sorted.push_back(largest);
|
|
||||||
vec.insert(vec.begin(), largest);
|
|
||||||
}
|
|
||||||
ASSERT_NE(vec, sorted);
|
|
||||||
|
|
||||||
// Just pass some arbitrary "self" to make sure the callbacks pass it through.
|
|
||||||
ASSERT_TRUE(merge_sort(vec.data(), vec.size(), &i, &string_funcs));
|
|
||||||
ASSERT_EQ(vec, sorted);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
} // namespace
|
|
@ -1,32 +0,0 @@
|
|||||||
#include "sort_test_util.hh"
|
|
||||||
|
|
||||||
#include <array>
|
|
||||||
#include <cstddef>
|
|
||||||
|
|
||||||
#include "sort.h"
|
|
||||||
#include "util.h"
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
template <typename T, std::size_t N>
|
|
||||||
int cmp_uint_array(const std::array<T, N> &a, const std::array<T, N> &b)
|
|
||||||
{
|
|
||||||
for (std::size_t i = 0; i < a.size(); ++i) {
|
|
||||||
const int cmp = cmp_uint(a[i], b[i]);
|
|
||||||
if (cmp != 0) {
|
|
||||||
return cmp;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
const Sort_Funcs Some_Type::funcs = sort_funcs<Some_Type>();
|
|
||||||
|
|
||||||
int my_type_cmp(const void *va, const void *vb)
|
|
||||||
{
|
|
||||||
const auto *a = static_cast<const Some_Type *>(va);
|
|
||||||
const auto *b = static_cast<const Some_Type *>(vb);
|
|
||||||
return cmp_uint_array(a->compare_value, b->compare_value);
|
|
||||||
}
|
|
||||||
|
|
||||||
bool operator<(const Some_Type &a, const Some_Type &b) { return a.compare_value < b.compare_value; }
|
|
@ -1,54 +0,0 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
* Copyright © 2023-2024 The TokTok team.
|
|
||||||
*/
|
|
||||||
|
|
||||||
#ifndef C_TOXCORE_TOXCORE_SORT_TEST_UTIL_H
|
|
||||||
#define C_TOXCORE_TOXCORE_SORT_TEST_UTIL_H
|
|
||||||
|
|
||||||
#include <array>
|
|
||||||
|
|
||||||
#include "sort.h"
|
|
||||||
|
|
||||||
struct Memory;
|
|
||||||
|
|
||||||
template <typename T>
|
|
||||||
constexpr Sort_Funcs sort_funcs()
|
|
||||||
{
|
|
||||||
return {
|
|
||||||
[](const void *object, const void *va, const void *vb) {
|
|
||||||
const T *a = static_cast<const T *>(va);
|
|
||||||
const T *b = static_cast<const T *>(vb);
|
|
||||||
|
|
||||||
return *a < *b;
|
|
||||||
},
|
|
||||||
[](const void *arr, uint32_t index) -> const void * {
|
|
||||||
const T *vec = static_cast<const T *>(arr);
|
|
||||||
return &vec[index];
|
|
||||||
},
|
|
||||||
[](void *arr, uint32_t index, const void *val) {
|
|
||||||
T *vec = static_cast<T *>(arr);
|
|
||||||
const T *value = static_cast<const T *>(val);
|
|
||||||
vec[index] = *value;
|
|
||||||
},
|
|
||||||
[](void *arr, uint32_t index, uint32_t size) -> void * {
|
|
||||||
T *vec = static_cast<T *>(arr);
|
|
||||||
return &vec[index];
|
|
||||||
},
|
|
||||||
[](const void *object, uint32_t size) -> void * { return new T[size]; },
|
|
||||||
[](const void *object, void *arr, uint32_t size) { delete[] static_cast<T *>(arr); },
|
|
||||||
};
|
|
||||||
}
|
|
||||||
|
|
||||||
// A realistic test case where we have a struct with some stuff and an expensive value we compare.
|
|
||||||
struct Some_Type {
|
|
||||||
const Memory *mem;
|
|
||||||
std::array<uint32_t, 8> compare_value;
|
|
||||||
const char *name;
|
|
||||||
|
|
||||||
static const Sort_Funcs funcs;
|
|
||||||
};
|
|
||||||
|
|
||||||
int my_type_cmp(const void *va, const void *vb);
|
|
||||||
bool operator<(const Some_Type &a, const Some_Type &b);
|
|
||||||
|
|
||||||
#endif // C_TOXCORE_TOXCORE_SORT_TEST_UTIL_H
|
|
2
external/toxcore/c-toxcore/toxcore/tox.c
vendored
2
external/toxcore/c-toxcore/toxcore/tox.c
vendored
@ -1,5 +1,5 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
/* SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
* Copyright © 2016-2018 The TokTok team.
|
* Copyright © 2016-2024 The TokTok team.
|
||||||
* Copyright © 2013 Tox project.
|
* Copyright © 2013 Tox project.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
117
external/toxcore/c-toxcore/toxcore/tox_api.c
vendored
117
external/toxcore/c-toxcore/toxcore/tox_api.c
vendored
@ -1722,3 +1722,120 @@ const char *tox_group_mod_event_to_string(Tox_Group_Mod_Event value)
|
|||||||
|
|
||||||
return "<invalid Tox_Group_Mod_Event>";
|
return "<invalid Tox_Group_Mod_Event>";
|
||||||
}
|
}
|
||||||
|
const char *tox_netprof_packet_id_to_string(Tox_Netprof_Packet_Id value)
|
||||||
|
{
|
||||||
|
switch (value) {
|
||||||
|
case TOX_NETPROF_PACKET_ID_ZERO:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ZERO";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TWO:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TWO";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_DISCONNECT:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_DISCONNECT";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FOUR:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_FOUR";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_PONG:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_PONG";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_OOB_SEND:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_OOB_SEND";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_OOB_RECV:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_OOB_RECV";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_ONION_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_ONION_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_ONION_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_ONION_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_DATA:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_TCP_DATA";
|
||||||
|
case TOX_NETPROF_PACKET_ID_COOKIE_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_COOKIE_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_COOKIE_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_COOKIE_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_CRYPTO_HS:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_CRYPTO_HS";
|
||||||
|
case TOX_NETPROF_PACKET_ID_CRYPTO_DATA:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_CRYPTO_DATA";
|
||||||
|
case TOX_NETPROF_PACKET_ID_CRYPTO:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_CRYPTO";
|
||||||
|
case TOX_NETPROF_PACKET_ID_LAN_DISCOVERY:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_LAN_DISCOVERY";
|
||||||
|
case TOX_NETPROF_PACKET_ID_GC_HANDSHAKE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_GC_HANDSHAKE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_GC_LOSSLESS:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_GC_LOSSLESS";
|
||||||
|
case TOX_NETPROF_PACKET_ID_GC_LOSSY:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_GC_LOSSY";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_SEND_INITIAL:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_SEND_INITIAL";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_SEND_1:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_SEND_1";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_SEND_2:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_SEND_2";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST_OLD:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST_OLD";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE_OLD:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE_OLD";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_DATA_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_DATA_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_DATA_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_DATA_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_RECV_3:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_RECV_3";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_RECV_2:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_RECV_2";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_RECV_1:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_ONION_RECV_1";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FORWARD_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_FORWARD_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FORWARDING:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_FORWARDING";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FORWARD_REPLY:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_FORWARD_REPLY";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_SEARCH_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_DATA_SEARCH_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_SEARCH_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_DATA_SEARCH_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_REQUEST:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_REQUEST";
|
||||||
|
case TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_RESPONSE:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_RESPONSE";
|
||||||
|
case TOX_NETPROF_PACKET_ID_BOOTSTRAP_INFO:
|
||||||
|
return "TOX_NETPROF_PACKET_ID_BOOTSTRAP_INFO";
|
||||||
|
}
|
||||||
|
|
||||||
|
return "<invalid Tox_Netprof_Packet_Id>";
|
||||||
|
}
|
||||||
|
const char *tox_netprof_packet_type_to_string(Tox_Netprof_Packet_Type value)
|
||||||
|
{
|
||||||
|
switch (value) {
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_CLIENT:
|
||||||
|
return "TOX_NETPROF_PACKET_TYPE_TCP_CLIENT";
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_SERVER:
|
||||||
|
return "TOX_NETPROF_PACKET_TYPE_TCP_SERVER";
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP:
|
||||||
|
return "TOX_NETPROF_PACKET_TYPE_TCP";
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_UDP:
|
||||||
|
return "TOX_NETPROF_PACKET_TYPE_UDP";
|
||||||
|
}
|
||||||
|
|
||||||
|
return "<invalid Tox_Netprof_Packet_Type>";
|
||||||
|
}
|
||||||
|
const char *tox_netprof_direction_to_string(Tox_Netprof_Direction value)
|
||||||
|
{
|
||||||
|
switch (value) {
|
||||||
|
case TOX_NETPROF_DIRECTION_SENT:
|
||||||
|
return "TOX_NETPROF_DIRECTION_SENT";
|
||||||
|
case TOX_NETPROF_DIRECTION_RECV:
|
||||||
|
return "TOX_NETPROF_DIRECTION_RECV";
|
||||||
|
}
|
||||||
|
|
||||||
|
return "<invalid Tox_Netprof_Direction>";
|
||||||
|
}
|
||||||
|
201
external/toxcore/c-toxcore/toxcore/tox_private.c
vendored
201
external/toxcore/c-toxcore/toxcore/tox_private.c
vendored
@ -1,5 +1,5 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
/* SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
* Copyright © 2016-2022 The TokTok team.
|
* Copyright © 2016-2024 The TokTok team.
|
||||||
* Copyright © 2013 Tox project.
|
* Copyright © 2013 Tox project.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
@ -11,13 +11,16 @@
|
|||||||
#include <assert.h>
|
#include <assert.h>
|
||||||
|
|
||||||
#include "DHT.h"
|
#include "DHT.h"
|
||||||
|
#include "TCP_server.h"
|
||||||
#include "attributes.h"
|
#include "attributes.h"
|
||||||
#include "ccompat.h"
|
#include "ccompat.h"
|
||||||
#include "crypto_core.h"
|
#include "crypto_core.h"
|
||||||
#include "group_chats.h"
|
#include "group_chats.h"
|
||||||
#include "group_common.h"
|
#include "group_common.h"
|
||||||
|
#include "logger.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
#include "net_crypto.h"
|
#include "net_crypto.h"
|
||||||
|
#include "net_profile.h"
|
||||||
#include "network.h"
|
#include "network.h"
|
||||||
#include "tox.h"
|
#include "tox.h"
|
||||||
#include "tox_struct.h"
|
#include "tox_struct.h"
|
||||||
@ -226,3 +229,199 @@ bool tox_group_peer_get_ip_address(const Tox *tox, uint32_t group_number, uint32
|
|||||||
SET_ERROR_PARAMETER(error, TOX_ERR_GROUP_PEER_QUERY_OK);
|
SET_ERROR_PARAMETER(error, TOX_ERR_GROUP_PEER_QUERY_OK);
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
uint64_t tox_netprof_get_packet_id_count(const Tox *tox, Tox_Netprof_Packet_Type type, uint8_t id,
|
||||||
|
Tox_Netprof_Direction direction)
|
||||||
|
{
|
||||||
|
assert(tox != nullptr);
|
||||||
|
|
||||||
|
tox_lock(tox);
|
||||||
|
|
||||||
|
const Net_Profile *tcp_c_profile = nc_get_tcp_client_net_profile(tox->m->net_crypto);
|
||||||
|
const Net_Profile *tcp_s_profile = tcp_server_get_net_profile(tox->m->tcp_server);
|
||||||
|
|
||||||
|
const Packet_Direction dir = (Packet_Direction) direction;
|
||||||
|
|
||||||
|
uint64_t count = 0;
|
||||||
|
|
||||||
|
switch (type) {
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_CLIENT: {
|
||||||
|
count = netprof_get_packet_count_id(tcp_c_profile, id, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_SERVER: {
|
||||||
|
count = netprof_get_packet_count_id(tcp_s_profile, id, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP: {
|
||||||
|
const uint64_t tcp_c_count = netprof_get_packet_count_id(tcp_c_profile, id, dir);
|
||||||
|
const uint64_t tcp_s_count = netprof_get_packet_count_id(tcp_s_profile, id, dir);
|
||||||
|
count = tcp_c_count + tcp_s_count;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_UDP: {
|
||||||
|
const Net_Profile *udp_profile = net_get_net_profile(tox->m->net);
|
||||||
|
count = netprof_get_packet_count_id(udp_profile, id, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
default: {
|
||||||
|
LOGGER_ERROR(tox->m->log, "invalid packet type: %d", type);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tox_unlock(tox);
|
||||||
|
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t tox_netprof_get_packet_total_count(const Tox *tox, Tox_Netprof_Packet_Type type,
|
||||||
|
Tox_Netprof_Direction direction)
|
||||||
|
{
|
||||||
|
assert(tox != nullptr);
|
||||||
|
|
||||||
|
tox_lock(tox);
|
||||||
|
|
||||||
|
const Net_Profile *tcp_c_profile = nc_get_tcp_client_net_profile(tox->m->net_crypto);
|
||||||
|
const Net_Profile *tcp_s_profile = tcp_server_get_net_profile(tox->m->tcp_server);
|
||||||
|
|
||||||
|
const Packet_Direction dir = (Packet_Direction) direction;
|
||||||
|
|
||||||
|
uint64_t count = 0;
|
||||||
|
|
||||||
|
switch (type) {
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_CLIENT: {
|
||||||
|
count = netprof_get_packet_count_total(tcp_c_profile, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_SERVER: {
|
||||||
|
count = netprof_get_packet_count_total(tcp_s_profile, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP: {
|
||||||
|
const uint64_t tcp_c_count = netprof_get_packet_count_total(tcp_c_profile, dir);
|
||||||
|
const uint64_t tcp_s_count = netprof_get_packet_count_total(tcp_s_profile, dir);
|
||||||
|
count = tcp_c_count + tcp_s_count;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_UDP: {
|
||||||
|
const Net_Profile *udp_profile = net_get_net_profile(tox->m->net);
|
||||||
|
count = netprof_get_packet_count_total(udp_profile, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
default: {
|
||||||
|
LOGGER_ERROR(tox->m->log, "invalid packet type: %d", type);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tox_unlock(tox);
|
||||||
|
|
||||||
|
return count;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t tox_netprof_get_packet_id_bytes(const Tox *tox, Tox_Netprof_Packet_Type type, uint8_t id,
|
||||||
|
Tox_Netprof_Direction direction)
|
||||||
|
{
|
||||||
|
assert(tox != nullptr);
|
||||||
|
|
||||||
|
tox_lock(tox);
|
||||||
|
|
||||||
|
const Net_Profile *tcp_c_profile = nc_get_tcp_client_net_profile(tox->m->net_crypto);
|
||||||
|
const Net_Profile *tcp_s_profile = tcp_server_get_net_profile(tox->m->tcp_server);
|
||||||
|
|
||||||
|
const Packet_Direction dir = (Packet_Direction) direction;
|
||||||
|
|
||||||
|
uint64_t bytes = 0;
|
||||||
|
|
||||||
|
switch (type) {
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_CLIENT: {
|
||||||
|
bytes = netprof_get_bytes_id(tcp_c_profile, id, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_SERVER: {
|
||||||
|
bytes = netprof_get_bytes_id(tcp_s_profile, id, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP: {
|
||||||
|
const uint64_t tcp_c_bytes = netprof_get_bytes_id(tcp_c_profile, id, dir);
|
||||||
|
const uint64_t tcp_s_bytes = netprof_get_bytes_id(tcp_s_profile, id, dir);
|
||||||
|
bytes = tcp_c_bytes + tcp_s_bytes;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_UDP: {
|
||||||
|
const Net_Profile *udp_profile = net_get_net_profile(tox->m->net);
|
||||||
|
bytes = netprof_get_bytes_id(udp_profile, id, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
default: {
|
||||||
|
LOGGER_ERROR(tox->m->log, "invalid packet type: %d", type);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tox_unlock(tox);
|
||||||
|
|
||||||
|
return bytes;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t tox_netprof_get_packet_total_bytes(const Tox *tox, Tox_Netprof_Packet_Type type,
|
||||||
|
Tox_Netprof_Direction direction)
|
||||||
|
{
|
||||||
|
assert(tox != nullptr);
|
||||||
|
|
||||||
|
tox_lock(tox);
|
||||||
|
|
||||||
|
const Net_Profile *tcp_c_profile = nc_get_tcp_client_net_profile(tox->m->net_crypto);
|
||||||
|
const Net_Profile *tcp_s_profile = tcp_server_get_net_profile(tox->m->tcp_server);
|
||||||
|
|
||||||
|
const Packet_Direction dir = (Packet_Direction) direction;
|
||||||
|
|
||||||
|
uint64_t bytes = 0;
|
||||||
|
|
||||||
|
switch (type) {
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_CLIENT: {
|
||||||
|
bytes = netprof_get_bytes_total(tcp_c_profile, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP_SERVER: {
|
||||||
|
bytes = netprof_get_bytes_total(tcp_s_profile, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_TCP: {
|
||||||
|
const uint64_t tcp_c_bytes = netprof_get_bytes_total(tcp_c_profile, dir);
|
||||||
|
const uint64_t tcp_s_bytes = netprof_get_bytes_total(tcp_s_profile, dir);
|
||||||
|
bytes = tcp_c_bytes + tcp_s_bytes;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
case TOX_NETPROF_PACKET_TYPE_UDP: {
|
||||||
|
const Net_Profile *udp_profile = net_get_net_profile(tox->m->net);
|
||||||
|
bytes = netprof_get_bytes_total(udp_profile, dir);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
default: {
|
||||||
|
LOGGER_ERROR(tox->m->log, "invalid packet type: %d", type);
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
tox_unlock(tox);
|
||||||
|
|
||||||
|
return bytes;
|
||||||
|
}
|
||||||
|
261
external/toxcore/c-toxcore/toxcore/tox_private.h
vendored
261
external/toxcore/c-toxcore/toxcore/tox_private.h
vendored
@ -169,6 +169,267 @@ uint16_t tox_dht_get_num_closelist(const Tox *tox);
|
|||||||
*/
|
*/
|
||||||
uint16_t tox_dht_get_num_closelist_announce_capable(const Tox *tox);
|
uint16_t tox_dht_get_num_closelist_announce_capable(const Tox *tox);
|
||||||
|
|
||||||
|
/*******************************************************************************
|
||||||
|
*
|
||||||
|
* :: Network profiler
|
||||||
|
*
|
||||||
|
******************************************************************************/
|
||||||
|
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Represents all of the network packet identifiers that Toxcore uses.
|
||||||
|
*
|
||||||
|
* Notes:
|
||||||
|
* - Some packet ID's have different purposes depending on the
|
||||||
|
* packet type. These ID's are given numeral names.
|
||||||
|
*
|
||||||
|
* - Queries for invalid packet ID's return undefined results. For example,
|
||||||
|
* querying a TCP-exclusive packet ID for UDP, or querying an ID that
|
||||||
|
* doesn't exist in this enum.
|
||||||
|
*/
|
||||||
|
typedef enum Tox_Netprof_Packet_Id {
|
||||||
|
/**
|
||||||
|
* Ping request packet (UDP).
|
||||||
|
* Routing request (TCP).
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ZERO = 0x00,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Ping response packet (UDP).
|
||||||
|
* Routing response (TCP).
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ONE = 0x01,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Get nodes request packet (UDP).
|
||||||
|
* Connection notification (TCP).
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TWO = 0x02,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP disconnect notification.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_DISCONNECT = 0x03,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Send nodes response packet (UDP).
|
||||||
|
* Ping packet (TCP).
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_FOUR = 0x04,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP pong packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_PONG = 0x05,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP out-of-band send packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_OOB_SEND = 0x06,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP out-of-band receive packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_OOB_RECV = 0x07,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP onion request packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_ONION_REQUEST = 0x08,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP onion response packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_ONION_RESPONSE = 0x09,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP data packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_TCP_DATA = 0x10,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Cookie request packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_COOKIE_REQUEST = 0x18,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Cookie response packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_COOKIE_RESPONSE = 0x19,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Crypto handshake packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_CRYPTO_HS = 0x1a,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Crypto data packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_CRYPTO_DATA = 0x1b,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Encrypted data packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_CRYPTO = 0x20,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* LAN discovery packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_LAN_DISCOVERY = 0x21,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* DHT groupchat packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_GC_HANDSHAKE = 0x5a,
|
||||||
|
TOX_NETPROF_PACKET_ID_GC_LOSSLESS = 0x5b,
|
||||||
|
TOX_NETPROF_PACKET_ID_GC_LOSSY = 0x5c,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Onion send packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_SEND_INITIAL = 0x80,
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_SEND_1 = 0x81,
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_SEND_2 = 0x82,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* DHT announce request packet (deprecated).
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST_OLD = 0x83,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* DHT announce response packet (deprecated).
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE_OLD = 0x84,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Onion data request packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_DATA_REQUEST = 0x85,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Onion data response packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_DATA_RESPONSE = 0x86,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* DHT announce request packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST = 0x87,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* DHT announce response packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE = 0x88,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Onion receive packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_RECV_3 = 0x8c,
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_RECV_2 = 0x8d,
|
||||||
|
TOX_NETPROF_PACKET_ID_ONION_RECV_1 = 0x8e,
|
||||||
|
|
||||||
|
TOX_NETPROF_PACKET_ID_FORWARD_REQUEST = 0x90,
|
||||||
|
TOX_NETPROF_PACKET_ID_FORWARDING = 0x91,
|
||||||
|
TOX_NETPROF_PACKET_ID_FORWARD_REPLY = 0x92,
|
||||||
|
|
||||||
|
TOX_NETPROF_PACKET_ID_DATA_SEARCH_REQUEST = 0x93,
|
||||||
|
TOX_NETPROF_PACKET_ID_DATA_SEARCH_RESPONSE = 0x94,
|
||||||
|
TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_REQUEST = 0x95,
|
||||||
|
TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_RESPONSE = 0x96,
|
||||||
|
TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_REQUEST = 0x97,
|
||||||
|
TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_RESPONSE = 0x98,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Bootstrap info packet.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_ID_BOOTSTRAP_INFO = 0xf0,
|
||||||
|
} Tox_Netprof_Packet_Id;
|
||||||
|
|
||||||
|
const char *tox_netprof_packet_id_to_string(Tox_Netprof_Packet_Id value);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Specifies the packet type for a given query.
|
||||||
|
*/
|
||||||
|
typedef enum Tox_Netprof_Packet_Type {
|
||||||
|
/**
|
||||||
|
* TCP client packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_TYPE_TCP_CLIENT,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* TCP server packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_TYPE_TCP_SERVER,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Combined TCP server and TCP client packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_TYPE_TCP,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* UDP packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_PACKET_TYPE_UDP,
|
||||||
|
} Tox_Netprof_Packet_Type;
|
||||||
|
|
||||||
|
const char *tox_netprof_packet_type_to_string(Tox_Netprof_Packet_Type value);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Specifies the packet direction for a given query.
|
||||||
|
*/
|
||||||
|
typedef enum Tox_Netprof_Direction {
|
||||||
|
/**
|
||||||
|
* Outbound packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_DIRECTION_SENT,
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Inbound packets.
|
||||||
|
*/
|
||||||
|
TOX_NETPROF_DIRECTION_RECV,
|
||||||
|
} Tox_Netprof_Direction;
|
||||||
|
|
||||||
|
const char *tox_netprof_direction_to_string(Tox_Netprof_Direction value);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the number of packets sent or received for a specific packet ID.
|
||||||
|
*
|
||||||
|
* @param type The types of packets being queried.
|
||||||
|
* @param id The packet ID being queried.
|
||||||
|
* @param direction The packet direction.
|
||||||
|
*/
|
||||||
|
uint64_t tox_netprof_get_packet_id_count(const Tox *tox, Tox_Netprof_Packet_Type type, uint8_t id,
|
||||||
|
Tox_Netprof_Direction direction);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the total number of packets sent or received.
|
||||||
|
*
|
||||||
|
* @param type The types of packets being queried.
|
||||||
|
* @param direction The packet direction.
|
||||||
|
*/
|
||||||
|
uint64_t tox_netprof_get_packet_total_count(const Tox *tox, Tox_Netprof_Packet_Type type,
|
||||||
|
Tox_Netprof_Direction direction);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the number of bytes sent or received for a specific packet ID.
|
||||||
|
*
|
||||||
|
* @param type The types of packets being queried.
|
||||||
|
* @param id The packet ID being queried.
|
||||||
|
* @param direction The packet direction.
|
||||||
|
*/
|
||||||
|
uint64_t tox_netprof_get_packet_id_bytes(const Tox *tox, Tox_Netprof_Packet_Type type, uint8_t id,
|
||||||
|
Tox_Netprof_Direction direction);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Return the total number of bytes sent or received.
|
||||||
|
*
|
||||||
|
* @param type The types of packets being queried.
|
||||||
|
* @param direction The packet direction.
|
||||||
|
*/
|
||||||
|
uint64_t tox_netprof_get_packet_total_bytes(const Tox *tox, Tox_Netprof_Packet_Type type,
|
||||||
|
Tox_Netprof_Direction direction);
|
||||||
|
|
||||||
|
|
||||||
/*******************************************************************************
|
/*******************************************************************************
|
||||||
*
|
*
|
||||||
* :: DHT groupchat queries.
|
* :: DHT groupchat queries.
|
||||||
|
3
external/toxcore/c-toxcore/toxcore/util.c
vendored
3
external/toxcore/c-toxcore/toxcore/util.c
vendored
@ -1,5 +1,5 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
/* SPDX-License-Identifier: GPL-3.0-or-later
|
||||||
* Copyright © 2016-2024 The TokTok team.
|
* Copyright © 2016-2018 The TokTok team.
|
||||||
* Copyright © 2013 Tox project.
|
* Copyright © 2013 Tox project.
|
||||||
* Copyright © 2013 plutooo
|
* Copyright © 2013 plutooo
|
||||||
*/
|
*/
|
||||||
@ -16,7 +16,6 @@
|
|||||||
#include <stdlib.h>
|
#include <stdlib.h>
|
||||||
#include <string.h>
|
#include <string.h>
|
||||||
|
|
||||||
#include "attributes.h"
|
|
||||||
#include "ccompat.h"
|
#include "ccompat.h"
|
||||||
#include "mem.h"
|
#include "mem.h"
|
||||||
|
|
||||||
|
33
external/toxcore/c-toxcore/toxcore/util_test.cc
vendored
33
external/toxcore/c-toxcore/toxcore/util_test.cc
vendored
@ -1,14 +1,39 @@
|
|||||||
/* SPDX-License-Identifier: GPL-3.0-or-later
|
|
||||||
* Copyright © 2016-2024 The TokTok team.
|
|
||||||
*/
|
|
||||||
#include "util.h"
|
#include "util.h"
|
||||||
|
|
||||||
#include <gtest/gtest.h>
|
#include <gtest/gtest.h>
|
||||||
|
|
||||||
#include <climits>
|
#include "crypto_core.h"
|
||||||
|
#include "crypto_core_test_util.hh"
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
|
TEST(Util, TwoRandomIdsAreNotEqual)
|
||||||
|
{
|
||||||
|
Test_Random rng;
|
||||||
|
uint8_t pk1[CRYPTO_PUBLIC_KEY_SIZE];
|
||||||
|
uint8_t sk1[CRYPTO_SECRET_KEY_SIZE];
|
||||||
|
uint8_t pk2[CRYPTO_PUBLIC_KEY_SIZE];
|
||||||
|
uint8_t sk2[CRYPTO_SECRET_KEY_SIZE];
|
||||||
|
|
||||||
|
crypto_new_keypair(rng, pk1, sk1);
|
||||||
|
crypto_new_keypair(rng, pk2, sk2);
|
||||||
|
|
||||||
|
EXPECT_FALSE(pk_equal(pk1, pk2));
|
||||||
|
}
|
||||||
|
|
||||||
|
TEST(Util, IdCopyMakesKeysEqual)
|
||||||
|
{
|
||||||
|
Test_Random rng;
|
||||||
|
uint8_t pk1[CRYPTO_PUBLIC_KEY_SIZE];
|
||||||
|
uint8_t sk1[CRYPTO_SECRET_KEY_SIZE];
|
||||||
|
uint8_t pk2[CRYPTO_PUBLIC_KEY_SIZE] = {0};
|
||||||
|
|
||||||
|
crypto_new_keypair(rng, pk1, sk1);
|
||||||
|
pk_copy(pk2, pk1);
|
||||||
|
|
||||||
|
EXPECT_TRUE(pk_equal(pk1, pk2));
|
||||||
|
}
|
||||||
|
|
||||||
TEST(Cmp, OrdersNumbersCorrectly)
|
TEST(Cmp, OrdersNumbersCorrectly)
|
||||||
{
|
{
|
||||||
EXPECT_EQ(cmp_uint(1, 2), -1);
|
EXPECT_EQ(cmp_uint(1, 2), -1);
|
||||||
|
@ -231,7 +231,7 @@ bool tox_pass_key_encrypt(const Tox_Pass_Key *key, const uint8_t plaintext[], si
|
|||||||
ciphertext += crypto_box_NONCEBYTES;
|
ciphertext += crypto_box_NONCEBYTES;
|
||||||
|
|
||||||
/* now encrypt */
|
/* now encrypt */
|
||||||
const int32_t encrypted_len = encrypt_data_symmetric(os_memory(), key->key, nonce, plaintext, plaintext_len, ciphertext);
|
const int32_t encrypted_len = encrypt_data_symmetric(key->key, nonce, plaintext, plaintext_len, ciphertext);
|
||||||
if (encrypted_len < 0 || (size_t)encrypted_len != plaintext_len + crypto_box_MACBYTES) {
|
if (encrypted_len < 0 || (size_t)encrypted_len != plaintext_len + crypto_box_MACBYTES) {
|
||||||
SET_ERROR_PARAMETER(error, TOX_ERR_ENCRYPTION_FAILED);
|
SET_ERROR_PARAMETER(error, TOX_ERR_ENCRYPTION_FAILED);
|
||||||
return false;
|
return false;
|
||||||
@ -316,7 +316,7 @@ bool tox_pass_key_decrypt(const Tox_Pass_Key *key, const uint8_t ciphertext[], s
|
|||||||
ciphertext += crypto_box_NONCEBYTES;
|
ciphertext += crypto_box_NONCEBYTES;
|
||||||
|
|
||||||
/* decrypt the ciphertext */
|
/* decrypt the ciphertext */
|
||||||
const int32_t decrypted_len = decrypt_data_symmetric(os_memory(), key->key, nonce, ciphertext, decrypt_length + crypto_box_MACBYTES, plaintext);
|
const int32_t decrypted_len = decrypt_data_symmetric(key->key, nonce, ciphertext, decrypt_length + crypto_box_MACBYTES, plaintext);
|
||||||
if (decrypted_len < 0 || (size_t)decrypted_len != decrypt_length) {
|
if (decrypted_len < 0 || (size_t)decrypted_len != decrypt_length) {
|
||||||
SET_ERROR_PARAMETER(error, TOX_ERR_DECRYPTION_FAILED);
|
SET_ERROR_PARAMETER(error, TOX_ERR_DECRYPTION_FAILED);
|
||||||
return false;
|
return false;
|
||||||
|
@ -56,8 +56,6 @@ target_sources(tomato PUBLIC
|
|||||||
./tox_avatar_loader.cpp
|
./tox_avatar_loader.cpp
|
||||||
./message_image_loader.hpp
|
./message_image_loader.hpp
|
||||||
./message_image_loader.cpp
|
./message_image_loader.cpp
|
||||||
./bitset_image_loader.hpp
|
|
||||||
./bitset_image_loader.cpp
|
|
||||||
|
|
||||||
./tox_avatar_manager.hpp
|
./tox_avatar_manager.hpp
|
||||||
./tox_avatar_manager.cpp
|
./tox_avatar_manager.cpp
|
||||||
@ -99,6 +97,9 @@ target_sources(tomato PUBLIC
|
|||||||
./tox_dht_cap_histo.hpp
|
./tox_dht_cap_histo.hpp
|
||||||
./tox_dht_cap_histo.cpp
|
./tox_dht_cap_histo.cpp
|
||||||
|
|
||||||
|
./tox_netprof_ui.hpp
|
||||||
|
./tox_netprof_ui.cpp
|
||||||
|
|
||||||
./tox_friend_faux_offline_messaging.hpp
|
./tox_friend_faux_offline_messaging.hpp
|
||||||
./tox_friend_faux_offline_messaging.cpp
|
./tox_friend_faux_offline_messaging.cpp
|
||||||
|
|
||||||
|
@ -1,153 +0,0 @@
|
|||||||
#include "./bitset_image_loader.hpp"
|
|
||||||
|
|
||||||
#include <solanaceae/object_store/object_store.hpp>
|
|
||||||
#include <solanaceae/object_store/meta_components_file.hpp>
|
|
||||||
|
|
||||||
#include "./os_comps.hpp"
|
|
||||||
|
|
||||||
#include <entt/entity/entity.hpp>
|
|
||||||
|
|
||||||
#include <SDL3/SDL.h>
|
|
||||||
|
|
||||||
#include <iostream>
|
|
||||||
|
|
||||||
// fwd
|
|
||||||
namespace Message {
|
|
||||||
uint64_t getTimeMS(void);
|
|
||||||
}
|
|
||||||
|
|
||||||
std::optional<TextureEntry> BitsetImageLoader::haveToTexture(TextureUploaderI& tu, BitSet& have, ObjectHandle o) {
|
|
||||||
assert(have.size_bits() > 0);
|
|
||||||
|
|
||||||
auto* surf = SDL_CreateSurfaceFrom(
|
|
||||||
have.size_bits(), 1,
|
|
||||||
SDL_PIXELFORMAT_INDEX1MSB, // LSB ?
|
|
||||||
have.data(), have.size_bytes()
|
|
||||||
);
|
|
||||||
if (surf == nullptr) {
|
|
||||||
std::cerr << "BIL error: bitset to 1bit surface creationg failed o:" << entt::to_integral(o.entity()) << "\n";
|
|
||||||
return std::nullopt;
|
|
||||||
}
|
|
||||||
|
|
||||||
SDL_Color colors[] {
|
|
||||||
{0, 0, 0, 0},
|
|
||||||
{255, 255, 255, 255},
|
|
||||||
};
|
|
||||||
|
|
||||||
SDL_Palette* palette = SDL_CreatePalette(2);
|
|
||||||
SDL_SetPaletteColors(palette, colors, 0, 2);
|
|
||||||
SDL_SetSurfacePalette(surf, palette);
|
|
||||||
auto* conv_surf = SDL_ConvertSurface(surf, SDL_PIXELFORMAT_RGBA32);
|
|
||||||
|
|
||||||
SDL_DestroySurface(surf);
|
|
||||||
SDL_DestroyPalette(palette);
|
|
||||||
|
|
||||||
if (conv_surf == nullptr) {
|
|
||||||
std::cerr << "BIL error: surface conversion failed o:" << entt::to_integral(o.entity()) << " : " << SDL_GetError() << "\n";
|
|
||||||
return std::nullopt;
|
|
||||||
}
|
|
||||||
|
|
||||||
SDL_LockSurface(conv_surf);
|
|
||||||
|
|
||||||
TextureEntry new_entry;
|
|
||||||
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
|
||||||
new_entry.width = have.size_bits();
|
|
||||||
new_entry.height = 1;
|
|
||||||
|
|
||||||
const auto n_t = tu.upload(static_cast<uint8_t*>(conv_surf->pixels), conv_surf->w, conv_surf->h, TextureUploaderI::RGBA);
|
|
||||||
assert(n_t != 0);
|
|
||||||
new_entry.textures.push_back(n_t);
|
|
||||||
new_entry.frame_duration.push_back(1);
|
|
||||||
|
|
||||||
std::cout << "BIL: genereated bitset image o:" << entt::to_integral(o.entity()) << "\n";
|
|
||||||
|
|
||||||
SDL_UnlockSurface(conv_surf);
|
|
||||||
SDL_DestroySurface(conv_surf);
|
|
||||||
|
|
||||||
return new_entry;
|
|
||||||
}
|
|
||||||
|
|
||||||
BitsetImageLoader::BitsetImageLoader(void) {
|
|
||||||
}
|
|
||||||
|
|
||||||
TextureLoaderResult BitsetImageLoader::load(TextureUploaderI& tu, ObjectHandle o) {
|
|
||||||
if (!static_cast<bool>(o)) {
|
|
||||||
std::cerr << "BIL error: trying to load invalid object\n";
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!o.any_of<ObjComp::F::LocalHaveBitset, ObjComp::F::RemoteHaveBitset>()) {
|
|
||||||
// after completion, this is called until the texture times out
|
|
||||||
//std::cout << "BIL: no have bitset\n";
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
if (o.all_of<ObjComp::F::LocalHaveBitset>()) {
|
|
||||||
auto& have = o.get<ObjComp::F::LocalHaveBitset>().have;
|
|
||||||
assert(have.size_bits() > 0);
|
|
||||||
return {haveToTexture(tu, have, o)};
|
|
||||||
} else if (o.all_of<ObjComp::F::RemoteHaveBitset>()) {
|
|
||||||
auto& list = o.get<ObjComp::F::RemoteHaveBitset>().others;
|
|
||||||
if (list.empty()) {
|
|
||||||
std::cout << "BIL: remote set list empty\n";
|
|
||||||
_tmp_bitset = {8};
|
|
||||||
return {haveToTexture(tu, _tmp_bitset, o)};
|
|
||||||
}
|
|
||||||
const auto& first_entry = list.begin()->second;
|
|
||||||
|
|
||||||
if (first_entry.have_all) {
|
|
||||||
_tmp_bitset = {8};
|
|
||||||
_tmp_bitset.invert();
|
|
||||||
std::cout << "BIL: remote first have all\n";
|
|
||||||
} else {
|
|
||||||
_tmp_bitset = first_entry.have;
|
|
||||||
assert(_tmp_bitset.size_bits() == first_entry.have.size_bits());
|
|
||||||
|
|
||||||
for (auto it = list.begin()+1; it != list.end(); it++) {
|
|
||||||
if (it->second.have_all) {
|
|
||||||
_tmp_bitset = {8};
|
|
||||||
_tmp_bitset.invert();
|
|
||||||
std::cout << "BIL: remote have all\n";
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
_tmp_bitset.merge(it->second.have);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return {haveToTexture(tu, _tmp_bitset, o)};
|
|
||||||
}
|
|
||||||
|
|
||||||
return {};
|
|
||||||
}
|
|
||||||
|
|
||||||
std::optional<TextureEntry> BitsetImageLoader::load(TextureUploaderI& tu, ObjectContactSub ocs) {
|
|
||||||
if (!static_cast<bool>(ocs.o)) {
|
|
||||||
std::cerr << "BIL error: trying to load invalid object\n";
|
|
||||||
return std::nullopt;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (!ocs.o.all_of<ObjComp::F::RemoteHaveBitset>()) {
|
|
||||||
// after completion, this is called until the texture times out
|
|
||||||
return std::nullopt;
|
|
||||||
}
|
|
||||||
|
|
||||||
auto& map = ocs.o.get<ObjComp::F::RemoteHaveBitset>().others;
|
|
||||||
auto it = map.find(ocs.c);
|
|
||||||
if (it == map.end()) {
|
|
||||||
// contact not found
|
|
||||||
return std::nullopt;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (it->second.have_all) {
|
|
||||||
BitSet tmp{8}; // or 1?
|
|
||||||
tmp.invert();
|
|
||||||
return haveToTexture(tu, tmp, ocs.o);
|
|
||||||
} else if (it->second.have.size_bits() == 0) {
|
|
||||||
BitSet tmp{8}; // or 1?
|
|
||||||
return haveToTexture(tu, tmp, ocs.o);
|
|
||||||
} else {
|
|
||||||
return haveToTexture(tu, it->second.have, ocs.o);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,36 +0,0 @@
|
|||||||
#pragma once
|
|
||||||
|
|
||||||
#include <solanaceae/object_store/fwd.hpp>
|
|
||||||
#include <solanaceae/contact/contact_model3.hpp>
|
|
||||||
#include <solanaceae/util/bitset.hpp>
|
|
||||||
|
|
||||||
#include "./texture_cache.hpp"
|
|
||||||
|
|
||||||
#include <optional>
|
|
||||||
|
|
||||||
struct ObjectContactSub final {
|
|
||||||
ObjectHandle o;
|
|
||||||
Contact3 c{entt::null};
|
|
||||||
};
|
|
||||||
|
|
||||||
template<>
|
|
||||||
struct std::hash<ObjectContactSub> {
|
|
||||||
std::size_t operator()(ObjectContactSub const& ocs) const noexcept {
|
|
||||||
const std::size_t h1 = reinterpret_cast<std::size_t>(ocs.o.registry());
|
|
||||||
const std::size_t h2 = entt::to_integral(ocs.o.entity());
|
|
||||||
const std::size_t h3 = entt::to_integral(ocs.c);
|
|
||||||
return (h1 << 3) ^ (h3 << 7) ^ (h2 * 11400714819323198485llu);
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
class BitsetImageLoader {
|
|
||||||
BitSet _tmp_bitset;
|
|
||||||
|
|
||||||
std::optional<TextureEntry> haveToTexture(TextureUploaderI& tu, BitSet& have, ObjectHandle o);
|
|
||||||
|
|
||||||
public:
|
|
||||||
BitsetImageLoader(void);
|
|
||||||
TextureLoaderResult load(TextureUploaderI& tu, ObjectHandle o);
|
|
||||||
std::optional<TextureEntry> load(TextureUploaderI& tu, ObjectContactSub ocs);
|
|
||||||
};
|
|
||||||
|
|
@ -11,8 +11,6 @@
|
|||||||
|
|
||||||
#include <imgui/imgui.h>
|
#include <imgui/imgui.h>
|
||||||
|
|
||||||
#include <cmath>
|
|
||||||
|
|
||||||
// fwd
|
// fwd
|
||||||
namespace Message {
|
namespace Message {
|
||||||
uint64_t getTimeMS(void);
|
uint64_t getTimeMS(void);
|
||||||
@ -93,7 +91,7 @@ bool SendImagePopup::load(void) {
|
|||||||
preview_image.timestamp_last_rendered = Message::getTimeMS();
|
preview_image.timestamp_last_rendered = Message::getTimeMS();
|
||||||
preview_image.current_texture = 0;
|
preview_image.current_texture = 0;
|
||||||
for (const auto& [ms, data] : original_image.frames) {
|
for (const auto& [ms, data] : original_image.frames) {
|
||||||
const auto n_t = _tu.upload(data.data(), original_image.width, original_image.height);
|
const auto n_t = _tu.uploadRGBA(data.data(), original_image.width, original_image.height);
|
||||||
preview_image.textures.push_back(n_t);
|
preview_image.textures.push_back(n_t);
|
||||||
preview_image.frame_duration.push_back(ms);
|
preview_image.frame_duration.push_back(ms);
|
||||||
}
|
}
|
||||||
|
@ -1,5 +1,7 @@
|
|||||||
#include "./chat_gui4.hpp"
|
#include "./chat_gui4.hpp"
|
||||||
|
|
||||||
|
#include <solanaceae/object_store/object_store.hpp>
|
||||||
|
|
||||||
#include <solanaceae/message3/components.hpp>
|
#include <solanaceae/message3/components.hpp>
|
||||||
#include <solanaceae/tox_messages/msg_components.hpp>
|
#include <solanaceae/tox_messages/msg_components.hpp>
|
||||||
#include <solanaceae/tox_messages/obj_components.hpp>
|
#include <solanaceae/tox_messages/obj_components.hpp>
|
||||||
@ -16,7 +18,6 @@
|
|||||||
|
|
||||||
#include <imgui/imgui.h>
|
#include <imgui/imgui.h>
|
||||||
#include <imgui/misc/cpp/imgui_stdlib.h>
|
#include <imgui/misc/cpp/imgui_stdlib.h>
|
||||||
#include <imgui/imgui_internal.h>
|
|
||||||
|
|
||||||
#include <SDL3/SDL.h>
|
#include <SDL3/SDL.h>
|
||||||
|
|
||||||
@ -24,7 +25,6 @@
|
|||||||
|
|
||||||
#include "./media_meta_info_loader.hpp"
|
#include "./media_meta_info_loader.hpp"
|
||||||
#include "./sdl_clipboard_utils.hpp"
|
#include "./sdl_clipboard_utils.hpp"
|
||||||
#include "os_comps.hpp"
|
|
||||||
|
|
||||||
#include <cctype>
|
#include <cctype>
|
||||||
#include <ctime>
|
#include <ctime>
|
||||||
@ -246,19 +246,7 @@ ChatGui4::ChatGui4(
|
|||||||
ContactTextureCache& contact_tc,
|
ContactTextureCache& contact_tc,
|
||||||
MessageTextureCache& msg_tc,
|
MessageTextureCache& msg_tc,
|
||||||
Theme& theme
|
Theme& theme
|
||||||
) :
|
) : _conf(conf), _os(os), _rmm(rmm), _cr(cr), _contact_tc(contact_tc), _msg_tc(msg_tc), _theme(theme), _sip(tu) {
|
||||||
_conf(conf),
|
|
||||||
_os(os),
|
|
||||||
_os_sr(_os.newSubRef(this)),
|
|
||||||
_rmm(rmm),
|
|
||||||
_cr(cr),
|
|
||||||
_contact_tc(contact_tc),
|
|
||||||
_msg_tc(msg_tc),
|
|
||||||
_b_tc(_bil, tu),
|
|
||||||
_theme(theme),
|
|
||||||
_sip(tu)
|
|
||||||
{
|
|
||||||
_os_sr.subscribe(ObjectStore_Event::object_update);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
ChatGui4::~ChatGui4(void) {
|
ChatGui4::~ChatGui4(void) {
|
||||||
@ -275,8 +263,6 @@ ChatGui4::~ChatGui4(void) {
|
|||||||
float ChatGui4::render(float time_delta) {
|
float ChatGui4::render(float time_delta) {
|
||||||
_fss.render();
|
_fss.render();
|
||||||
_sip.render(time_delta);
|
_sip.render(time_delta);
|
||||||
_b_tc.update();
|
|
||||||
_b_tc.workLoadQueue();
|
|
||||||
|
|
||||||
const ImGuiViewport* viewport = ImGui::GetMainViewport();
|
const ImGuiViewport* viewport = ImGui::GetMainViewport();
|
||||||
ImGui::SetNextWindowPos(viewport->WorkPos);
|
ImGui::SetNextWindowPos(viewport->WorkPos);
|
||||||
@ -1044,7 +1030,7 @@ void ChatGui4::renderMessageBodyText(Message3Registry& reg, const Message3 e) {
|
|||||||
ImGui::BeginGroup();
|
ImGui::BeginGroup();
|
||||||
do {
|
do {
|
||||||
const auto current_line = msgtext_sv.substr(pos_prev, pos_next - pos_prev);
|
const auto current_line = msgtext_sv.substr(pos_prev, pos_next - pos_prev);
|
||||||
if (!current_line.empty() && current_line.front() == '>') {
|
if (current_line.front() == '>') {
|
||||||
// TODO: theming
|
// TODO: theming
|
||||||
ImGui::PushStyleColor(ImGuiCol_Text, {0.3f, 0.9f, 0.1f, 1.f});
|
ImGui::PushStyleColor(ImGuiCol_Text, {0.3f, 0.9f, 0.1f, 1.f});
|
||||||
ImGui::TextUnformatted(current_line.data(), current_line.data()+current_line.size());
|
ImGui::TextUnformatted(current_line.data(), current_line.data()+current_line.size());
|
||||||
@ -1176,8 +1162,6 @@ void ChatGui4::renderMessageBodyFile(Message3Registry& reg, const Message3 e) {
|
|||||||
// hacky
|
// hacky
|
||||||
const auto* fts = o.try_get<ObjComp::Ephemeral::File::TransferStats>();
|
const auto* fts = o.try_get<ObjComp::Ephemeral::File::TransferStats>();
|
||||||
if (fts != nullptr && o.any_of<ObjComp::F::SingleInfo, ObjComp::F::CollectionInfo>()) {
|
if (fts != nullptr && o.any_of<ObjComp::F::SingleInfo, ObjComp::F::CollectionInfo>()) {
|
||||||
const bool upload = o.all_of<ObjComp::F::TagLocalHaveAll>() && fts->total_down <= 0;
|
|
||||||
|
|
||||||
const int64_t total_size =
|
const int64_t total_size =
|
||||||
o.all_of<ObjComp::F::SingleInfo>() ?
|
o.all_of<ObjComp::F::SingleInfo>() ?
|
||||||
o.get<ObjComp::F::SingleInfo>().file_size :
|
o.get<ObjComp::F::SingleInfo>().file_size :
|
||||||
@ -1186,7 +1170,7 @@ void ChatGui4::renderMessageBodyFile(Message3Registry& reg, const Message3 e) {
|
|||||||
|
|
||||||
int64_t transfer_total {0u};
|
int64_t transfer_total {0u};
|
||||||
float transfer_rate {0.f};
|
float transfer_rate {0.f};
|
||||||
if (upload) {
|
if (o.all_of<ObjComp::F::TagLocalHaveAll>() && fts->total_down <= 0) {
|
||||||
// if have all AND no dl -> show upload progress
|
// if have all AND no dl -> show upload progress
|
||||||
ImGui::TextUnformatted(" up");
|
ImGui::TextUnformatted(" up");
|
||||||
transfer_total = fts->total_up;
|
transfer_total = fts->total_up;
|
||||||
@ -1199,12 +1183,7 @@ void ChatGui4::renderMessageBodyFile(Message3Registry& reg, const Message3 e) {
|
|||||||
}
|
}
|
||||||
ImGui::SameLine();
|
ImGui::SameLine();
|
||||||
|
|
||||||
float fraction{0.f};
|
float fraction = float(transfer_total) / total_size;
|
||||||
if (total_size > 0) {
|
|
||||||
fraction = float(transfer_total) / total_size;
|
|
||||||
} else if (o.all_of<ObjComp::F::TagLocalHaveAll>()) {
|
|
||||||
fraction = 1.f;
|
|
||||||
}
|
|
||||||
|
|
||||||
char overlay_buf[128];
|
char overlay_buf[128];
|
||||||
if (transfer_rate > 0.000001f) {
|
if (transfer_rate > 0.000001f) {
|
||||||
@ -1232,57 +1211,11 @@ void ChatGui4::renderMessageBodyFile(Message3Registry& reg, const Message3 e) {
|
|||||||
std::snprintf(overlay_buf, sizeof(overlay_buf), "%.1f%%", fraction * 100 + 0.01f);
|
std::snprintf(overlay_buf, sizeof(overlay_buf), "%.1f%%", fraction * 100 + 0.01f);
|
||||||
}
|
}
|
||||||
|
|
||||||
if (
|
|
||||||
(!upload && !o.all_of<ObjComp::F::TagLocalHaveAll>() && o.all_of<ObjComp::F::LocalHaveBitset>()) ||
|
|
||||||
(upload && o.all_of<ObjComp::F::RemoteHaveBitset>())
|
|
||||||
) {
|
|
||||||
ImGui::BeginGroup();
|
|
||||||
|
|
||||||
// TODO: hights are all off
|
|
||||||
|
|
||||||
ImGui::ProgressBar(
|
|
||||||
fraction,
|
|
||||||
{-FLT_MIN, TEXT_BASE_HEIGHT*0.66f},
|
|
||||||
overlay_buf
|
|
||||||
);
|
|
||||||
|
|
||||||
ImVec2 orig_curser_pos = ImGui::GetCursorPos();
|
|
||||||
const ImVec2 bar_size{ImGui::GetContentRegionAvail().x, TEXT_BASE_HEIGHT*0.15f};
|
|
||||||
// deploy dummy and check visibility
|
|
||||||
ImGui::Dummy(bar_size);
|
|
||||||
if (ImGui::IsItemVisible()) {
|
|
||||||
ImGui::SetCursorPos(orig_curser_pos); // reset before dummy
|
|
||||||
|
|
||||||
auto const cursor_start_vec = ImGui::GetCursorScreenPos();
|
|
||||||
// TODO: replace with own version, so we dont have to internal
|
|
||||||
ImGui::RenderFrame(
|
|
||||||
cursor_start_vec,
|
|
||||||
{
|
|
||||||
cursor_start_vec.x + bar_size.x,
|
|
||||||
cursor_start_vec.y + bar_size.y
|
|
||||||
},
|
|
||||||
ImGui::GetColorU32(ImGuiCol_FrameBg),
|
|
||||||
false
|
|
||||||
);
|
|
||||||
|
|
||||||
auto [id, img_width, img_height] = _b_tc.get(o);
|
|
||||||
ImGui::Image(
|
|
||||||
id,
|
|
||||||
bar_size,
|
|
||||||
{0.f, 0.f}, // default
|
|
||||||
{1.f, 1.f}, // default
|
|
||||||
ImGui::GetStyleColorVec4(ImGuiCol_PlotHistogram)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
ImGui::EndGroup();
|
|
||||||
} else {
|
|
||||||
ImGui::ProgressBar(
|
ImGui::ProgressBar(
|
||||||
fraction,
|
fraction,
|
||||||
{-FLT_MIN, TEXT_BASE_HEIGHT},
|
{-FLT_MIN, TEXT_BASE_HEIGHT},
|
||||||
overlay_buf
|
overlay_buf
|
||||||
);
|
);
|
||||||
}
|
|
||||||
} else {
|
} else {
|
||||||
// infinite scrolling progressbar fallback
|
// infinite scrolling progressbar fallback
|
||||||
ImGui::TextUnformatted(" ??");
|
ImGui::TextUnformatted(" ??");
|
||||||
@ -1294,6 +1227,16 @@ void ChatGui4::renderMessageBodyFile(Message3Registry& reg, const Message3 e) {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (!o.all_of<ObjComp::F::TagLocalHaveAll>() && o.all_of<ObjComp::F::LocalHaveBitset>()) {
|
||||||
|
// texture based on have bitset
|
||||||
|
// TODO: missing have chunks/chunksize to get the correct size
|
||||||
|
|
||||||
|
//const auto& bitest = o.get<ObjComp::F::LocalHaveBitset>().have;
|
||||||
|
// generate 1bit sdlsurface zerocopy using bitset.data() and bitset.size_bytes()
|
||||||
|
// optionally scale down filtered (would copy)
|
||||||
|
// update texture? in cache?
|
||||||
|
}
|
||||||
|
|
||||||
if (o.all_of<ObjComp::F::FrameDims>()) {
|
if (o.all_of<ObjComp::F::FrameDims>()) {
|
||||||
const auto& frame_dims = o.get<ObjComp::F::FrameDims>();
|
const auto& frame_dims = o.get<ObjComp::F::FrameDims>();
|
||||||
|
|
||||||
@ -1778,11 +1721,3 @@ void ChatGui4::sendFileList(const std::vector<std::string_view>& list) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
bool ChatGui4::onEvent(const ObjectStore::Events::ObjectUpdate& e) {
|
|
||||||
if (e.e.any_of<ObjComp::F::LocalHaveBitset, ObjComp::F::RemoteHaveBitset>()) {
|
|
||||||
_b_tc.stale(e.e);
|
|
||||||
}
|
|
||||||
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
@ -1,6 +1,6 @@
|
|||||||
#pragma once
|
#pragma once
|
||||||
|
|
||||||
#include <solanaceae/object_store/object_store.hpp>
|
#include <solanaceae/object_store/fwd.hpp>
|
||||||
#include <solanaceae/message3/registry_message_model.hpp>
|
#include <solanaceae/message3/registry_message_model.hpp>
|
||||||
#include <solanaceae/util/config_model.hpp>
|
#include <solanaceae/util/config_model.hpp>
|
||||||
|
|
||||||
@ -10,7 +10,6 @@
|
|||||||
#include "./texture_cache.hpp"
|
#include "./texture_cache.hpp"
|
||||||
#include "./tox_avatar_loader.hpp"
|
#include "./tox_avatar_loader.hpp"
|
||||||
#include "./message_image_loader.hpp"
|
#include "./message_image_loader.hpp"
|
||||||
#include "./bitset_image_loader.hpp"
|
|
||||||
#include "./chat_gui/file_selector.hpp"
|
#include "./chat_gui/file_selector.hpp"
|
||||||
#include "./chat_gui/send_image_popup.hpp"
|
#include "./chat_gui/send_image_popup.hpp"
|
||||||
|
|
||||||
@ -24,19 +23,15 @@
|
|||||||
|
|
||||||
using ContactTextureCache = TextureCache<void*, Contact3, ToxAvatarLoader>;
|
using ContactTextureCache = TextureCache<void*, Contact3, ToxAvatarLoader>;
|
||||||
using MessageTextureCache = TextureCache<void*, Message3Handle, MessageImageLoader>;
|
using MessageTextureCache = TextureCache<void*, Message3Handle, MessageImageLoader>;
|
||||||
using BitsetTextureCache = TextureCache<void*, ObjectHandle, BitsetImageLoader>;
|
|
||||||
|
|
||||||
class ChatGui4 : public ObjectStoreEventI {
|
class ChatGui4 {
|
||||||
ConfigModelI& _conf;
|
ConfigModelI& _conf;
|
||||||
ObjectStore2& _os;
|
ObjectStore2& _os;
|
||||||
ObjectStoreEventProviderI::SubscriptionReference _os_sr;
|
|
||||||
RegistryMessageModelI& _rmm;
|
RegistryMessageModelI& _rmm;
|
||||||
Contact3Registry& _cr;
|
Contact3Registry& _cr;
|
||||||
|
|
||||||
ContactTextureCache& _contact_tc;
|
ContactTextureCache& _contact_tc;
|
||||||
MessageTextureCache& _msg_tc;
|
MessageTextureCache& _msg_tc;
|
||||||
BitsetImageLoader _bil;
|
|
||||||
BitsetTextureCache _b_tc;
|
|
||||||
|
|
||||||
Theme& _theme;
|
Theme& _theme;
|
||||||
|
|
||||||
@ -91,9 +86,6 @@ class ChatGui4 : public ObjectStoreEventI {
|
|||||||
//bool renderSubContactListContact(const Contact3 c, const bool selected) const;
|
//bool renderSubContactListContact(const Contact3 c, const bool selected) const;
|
||||||
|
|
||||||
void pasteFile(const char* mime_type);
|
void pasteFile(const char* mime_type);
|
||||||
|
|
||||||
protected:
|
|
||||||
bool onEvent(const ObjectStore::Events::ObjectUpdate&) override;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
||||||
|
@ -55,7 +55,6 @@ ImageLoaderSDLImage::ImageInfo ImageLoaderSDLImage::loadInfoFromMemory(const uin
|
|||||||
// we ignore tga
|
// we ignore tga
|
||||||
auto ext_opt = getExt(ios);
|
auto ext_opt = getExt(ios);
|
||||||
if (!ext_opt.has_value()) {
|
if (!ext_opt.has_value()) {
|
||||||
SDL_CloseIO(ios);
|
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -81,7 +80,6 @@ ImageLoaderSDLImage::ImageResult ImageLoaderSDLImage::loadFromMemoryRGBA(const u
|
|||||||
// we ignore tga
|
// we ignore tga
|
||||||
auto ext_opt = getExt(ios);
|
auto ext_opt = getExt(ios);
|
||||||
if (!ext_opt.has_value()) {
|
if (!ext_opt.has_value()) {
|
||||||
SDL_CloseIO(ios);
|
|
||||||
return res;
|
return res;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -48,6 +48,7 @@ MainScreen::MainScreen(SimpleConfigModel&& conf_, SDL_Renderer* renderer_, Theme
|
|||||||
osui(os),
|
osui(os),
|
||||||
tuiu(tc, conf, &tpi),
|
tuiu(tc, conf, &tpi),
|
||||||
tdch(tpi),
|
tdch(tpi),
|
||||||
|
tnui(tpi),
|
||||||
smui(os, sm),
|
smui(os, sm),
|
||||||
dvt(os, sm, sdlrtu)
|
dvt(os, sm, sdlrtu)
|
||||||
{
|
{
|
||||||
@ -362,6 +363,7 @@ Screen* MainScreen::render(float time_delta, bool&) {
|
|||||||
osui.render();
|
osui.render();
|
||||||
tuiu.render(); // render
|
tuiu.render(); // render
|
||||||
tdch.render(); // render
|
tdch.render(); // render
|
||||||
|
const float tnui_interval = tnui.render(time_delta);
|
||||||
smui.render();
|
smui.render();
|
||||||
const float dvt_interval = dvt.render();
|
const float dvt_interval = dvt.render();
|
||||||
|
|
||||||
@ -552,6 +554,7 @@ Screen* MainScreen::render(float time_delta, bool&) {
|
|||||||
if (!_window_hidden && _time_since_event < curr_profile.low_delay_window) {
|
if (!_window_hidden && _time_since_event < curr_profile.low_delay_window) {
|
||||||
_render_interval = std::min<float>(_render_interval, ctc_interval);
|
_render_interval = std::min<float>(_render_interval, ctc_interval);
|
||||||
_render_interval = std::min<float>(_render_interval, msgtc_interval);
|
_render_interval = std::min<float>(_render_interval, msgtc_interval);
|
||||||
|
_render_interval = std::min<float>(_render_interval, tnui_interval);
|
||||||
|
|
||||||
_render_interval = std::clamp(
|
_render_interval = std::clamp(
|
||||||
_render_interval,
|
_render_interval,
|
||||||
@ -562,6 +565,7 @@ Screen* MainScreen::render(float time_delta, bool&) {
|
|||||||
} else if (!_window_hidden && _time_since_event < curr_profile.mid_delay_window) {
|
} else if (!_window_hidden && _time_since_event < curr_profile.mid_delay_window) {
|
||||||
_render_interval = std::min<float>(_render_interval, ctc_interval);
|
_render_interval = std::min<float>(_render_interval, ctc_interval);
|
||||||
_render_interval = std::min<float>(_render_interval, msgtc_interval);
|
_render_interval = std::min<float>(_render_interval, msgtc_interval);
|
||||||
|
_render_interval = std::min<float>(_render_interval, tnui_interval);
|
||||||
|
|
||||||
_render_interval = std::clamp(
|
_render_interval = std::clamp(
|
||||||
_render_interval,
|
_render_interval,
|
||||||
@ -607,6 +611,7 @@ Screen* MainScreen::tick(float time_delta, bool& quit) {
|
|||||||
const float pm_interval = pm.tick(time_delta); // compute
|
const float pm_interval = pm.tick(time_delta); // compute
|
||||||
|
|
||||||
tdch.tick(time_delta); // compute
|
tdch.tick(time_delta); // compute
|
||||||
|
tnui.tick(time_delta); // compute
|
||||||
|
|
||||||
mts.iterate(); // compute (after mfs)
|
mts.iterate(); // compute (after mfs)
|
||||||
|
|
||||||
|
@ -34,6 +34,7 @@
|
|||||||
#include "./object_store_ui.hpp"
|
#include "./object_store_ui.hpp"
|
||||||
#include "./tox_ui_utils.hpp"
|
#include "./tox_ui_utils.hpp"
|
||||||
#include "./tox_dht_cap_histo.hpp"
|
#include "./tox_dht_cap_histo.hpp"
|
||||||
|
#include "./tox_netprof_ui.hpp"
|
||||||
#include "./tox_friend_faux_offline_messaging.hpp"
|
#include "./tox_friend_faux_offline_messaging.hpp"
|
||||||
#include "./stream_manager_ui.hpp"
|
#include "./stream_manager_ui.hpp"
|
||||||
#include "./debug_video_tap.hpp"
|
#include "./debug_video_tap.hpp"
|
||||||
@ -96,6 +97,7 @@ struct MainScreen final : public Screen {
|
|||||||
ObjectStoreUI osui;
|
ObjectStoreUI osui;
|
||||||
ToxUIUtils tuiu;
|
ToxUIUtils tuiu;
|
||||||
ToxDHTCapHisto tdch;
|
ToxDHTCapHisto tdch;
|
||||||
|
ToxNetprofUI tnui;
|
||||||
StreamManagerUI smui;
|
StreamManagerUI smui;
|
||||||
DebugVideoTap dvt;
|
DebugVideoTap dvt;
|
||||||
|
|
||||||
|
@ -30,29 +30,29 @@ MessageImageLoader::MessageImageLoader(void) {
|
|||||||
_image_loaders.push_back(std::make_unique<ImageLoaderSDLImage>());
|
_image_loaders.push_back(std::make_unique<ImageLoaderSDLImage>());
|
||||||
}
|
}
|
||||||
|
|
||||||
TextureLoaderResult MessageImageLoader::load(TextureUploaderI& tu, Message3Handle m) {
|
std::optional<TextureEntry> MessageImageLoader::load(TextureUploaderI& tu, Message3Handle m) {
|
||||||
if (!static_cast<bool>(m)) {
|
if (!static_cast<bool>(m)) {
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (m.all_of<Message::Components::TagNotImage>()) {
|
if (m.all_of<Message::Components::TagNotImage>()) {
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!m.all_of<Message::Components::MessageFileObject>()) {
|
if (!m.all_of<Message::Components::MessageFileObject>()) {
|
||||||
// not a file message
|
// not a file message
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
const auto& o = m.get<Message::Components::MessageFileObject>().o;
|
const auto& o = m.get<Message::Components::MessageFileObject>().o;
|
||||||
|
|
||||||
if (!static_cast<bool>(o)) {
|
if (!static_cast<bool>(o)) {
|
||||||
std::cerr << "MIL error: invalid object in file message\n";
|
std::cerr << "MIL error: invalid object in file message\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!o.all_of<ObjComp::Ephemeral::Backend, ObjComp::F::SingleInfo>()) {
|
if (!o.all_of<ObjComp::Ephemeral::Backend, ObjComp::F::SingleInfo>()) {
|
||||||
std::cerr << "MIL error: object missing backend (?)\n";
|
std::cerr << "MIL error: object missing backend (?)\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
// TODO: handle collections
|
// TODO: handle collections
|
||||||
@ -60,40 +60,40 @@ TextureLoaderResult MessageImageLoader::load(TextureUploaderI& tu, Message3Handl
|
|||||||
|
|
||||||
if (file_size > 50*1024*1024) {
|
if (file_size > 50*1024*1024) {
|
||||||
std::cerr << "MIL error: image file too large\n";
|
std::cerr << "MIL error: image file too large\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (file_size == 0) {
|
if (file_size == 0) {
|
||||||
std::cerr << "MIL warning: empty file\n";
|
std::cerr << "MIL warning: empty file\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (!o.all_of<ObjComp::F::TagLocalHaveAll>()) {
|
if (!o.all_of<ObjComp::F::TagLocalHaveAll>()) {
|
||||||
// not ready yet
|
// not ready yet
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto* file_backend = o.get<ObjComp::Ephemeral::Backend>().ptr;
|
auto* file_backend = o.get<ObjComp::Ephemeral::Backend>().ptr;
|
||||||
if (file_backend == nullptr) {
|
if (file_backend == nullptr) {
|
||||||
std::cerr << "MIL error: object backend nullptr\n";
|
std::cerr << "MIL error: object backend nullptr\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto file2 = file_backend->file2(o, StorageBackendI::FILE2_READ);
|
auto file2 = file_backend->file2(o, StorageBackendI::FILE2_READ);
|
||||||
if (!file2 || !file2->isGood() || !file2->can_read) {
|
if (!file2 || !file2->isGood() || !file2->can_read) {
|
||||||
std::cerr << "MIL error: creating file2 from object via backendI\n";
|
std::cerr << "MIL error: creating file2 from object via backendI\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
auto read_data = file2->read(file_size, 0);
|
auto read_data = file2->read(file_size, 0);
|
||||||
if (read_data.ptr == nullptr) {
|
if (read_data.ptr == nullptr) {
|
||||||
std::cerr << "MMIL error: reading from file2 returned nullptr\n";
|
std::cerr << "MMIL error: reading from file2 returned nullptr\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (read_data.size != file_size) {
|
if (read_data.size != file_size) {
|
||||||
std::cerr << "MIL error: reading from file2 size missmatch, should be " << file_size << ", is " << read_data.size << "\n";
|
std::cerr << "MIL error: reading from file2 size missmatch, should be " << file_size << ", is " << read_data.size << "\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
// try all loaders after another
|
// try all loaders after another
|
||||||
@ -107,7 +107,7 @@ TextureLoaderResult MessageImageLoader::load(TextureUploaderI& tu, Message3Handl
|
|||||||
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
||||||
new_entry.current_texture = 0;
|
new_entry.current_texture = 0;
|
||||||
for (const auto& [ms, data] : res.frames) {
|
for (const auto& [ms, data] : res.frames) {
|
||||||
const auto n_t = tu.upload(data.data(), res.width, res.height);
|
const auto n_t = tu.uploadRGBA(data.data(), res.width, res.height);
|
||||||
new_entry.textures.push_back(n_t);
|
new_entry.textures.push_back(n_t);
|
||||||
new_entry.frame_duration.push_back(ms);
|
new_entry.frame_duration.push_back(ms);
|
||||||
}
|
}
|
||||||
@ -117,10 +117,10 @@ TextureLoaderResult MessageImageLoader::load(TextureUploaderI& tu, Message3Handl
|
|||||||
|
|
||||||
std::cout << "MIL: loaded image file o:" << /*file_path*/ entt::to_integral(o.entity()) << "\n";
|
std::cout << "MIL: loaded image file o:" << /*file_path*/ entt::to_integral(o.entity()) << "\n";
|
||||||
|
|
||||||
return {new_entry};
|
return new_entry;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::cerr << "MIL error: failed to load message (unhandled format)\n";
|
std::cerr << "MIL error: failed to load message (unhandled format)\n";
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -12,6 +12,6 @@ class MessageImageLoader {
|
|||||||
|
|
||||||
public:
|
public:
|
||||||
MessageImageLoader(void);
|
MessageImageLoader(void);
|
||||||
TextureLoaderResult load(TextureUploaderI& tu, Message3Handle m);
|
std::optional<TextureEntry> load(TextureUploaderI& tu, Message3Handle c);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -8,6 +8,21 @@
|
|||||||
|
|
||||||
namespace ObjectStore::Components {
|
namespace ObjectStore::Components {
|
||||||
|
|
||||||
|
// until i find a better name
|
||||||
|
namespace File {
|
||||||
|
|
||||||
|
// ephemeral?, not sure saving this to disk makes sense
|
||||||
|
// tag remove have all?
|
||||||
|
struct RemoteHaveBitset {
|
||||||
|
struct Entry {
|
||||||
|
bool have_all {false};
|
||||||
|
BitSet have;
|
||||||
|
};
|
||||||
|
entt::dense_map<Contact3, Entry> others;
|
||||||
|
};
|
||||||
|
|
||||||
|
} // File
|
||||||
|
|
||||||
namespace Ephemeral {
|
namespace Ephemeral {
|
||||||
|
|
||||||
namespace File {
|
namespace File {
|
||||||
|
@ -18,6 +18,8 @@ constexpr std::string_view entt::type_name<x>::value() noexcept { \
|
|||||||
|
|
||||||
// cross compile(r) stable ids
|
// cross compile(r) stable ids
|
||||||
|
|
||||||
|
DEFINE_COMP_ID(ObjComp::F::RemoteHaveBitset)
|
||||||
|
|
||||||
DEFINE_COMP_ID(ObjComp::Ephemeral::File::TransferStatsSeparated)
|
DEFINE_COMP_ID(ObjComp::Ephemeral::File::TransferStatsSeparated)
|
||||||
|
|
||||||
#undef DEFINE_COMP_ID
|
#undef DEFINE_COMP_ID
|
||||||
|
@ -46,7 +46,7 @@ TextureEntry generateTestAnim(TextureUploaderI& tu) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
const auto n_t = tu.upload(pixels.data(), width, height);
|
const auto n_t = tu.uploadRGBA(pixels.data(), width, height);
|
||||||
|
|
||||||
// this is so ugly
|
// this is so ugly
|
||||||
new_entry.textures.emplace_back(n_t);
|
new_entry.textures.emplace_back(n_t);
|
||||||
|
@ -5,10 +5,6 @@
|
|||||||
#include <entt/container/dense_map.hpp>
|
#include <entt/container/dense_map.hpp>
|
||||||
#include <entt/container/dense_set.hpp>
|
#include <entt/container/dense_set.hpp>
|
||||||
|
|
||||||
#include <optional>
|
|
||||||
#include <vector>
|
|
||||||
#include <cassert>
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
struct TextureEntry {
|
struct TextureEntry {
|
||||||
@ -74,11 +70,6 @@ struct TextureEntry {
|
|||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
struct TextureLoaderResult {
|
|
||||||
std::optional<TextureEntry> texture;
|
|
||||||
bool keep_trying{false}; // if set, cant be cleared, as some async might be going on
|
|
||||||
};
|
|
||||||
|
|
||||||
TextureEntry generateTestAnim(TextureUploaderI& tu);
|
TextureEntry generateTestAnim(TextureUploaderI& tu);
|
||||||
|
|
||||||
// fwd
|
// fwd
|
||||||
@ -100,7 +91,6 @@ struct TextureCache {
|
|||||||
|
|
||||||
entt::dense_map<KeyType, TextureEntry> _cache;
|
entt::dense_map<KeyType, TextureEntry> _cache;
|
||||||
entt::dense_set<KeyType> _to_load;
|
entt::dense_set<KeyType> _to_load;
|
||||||
// to_reload // to_update? _marked_stale?
|
|
||||||
|
|
||||||
const uint64_t ms_before_purge {60 * 1000ull};
|
const uint64_t ms_before_purge {60 * 1000ull};
|
||||||
const size_t min_count_before_purge {0}; // starts purging after that
|
const size_t min_count_before_purge {0}; // starts purging after that
|
||||||
@ -144,18 +134,6 @@ struct TextureCache {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// markes a texture as stale and will reload it
|
|
||||||
// only if it already is loaded, does not update ts
|
|
||||||
bool stale(const KeyType& key) {
|
|
||||||
auto it = _cache.find(key);
|
|
||||||
|
|
||||||
if (it == _cache.end()) {
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
_to_load.insert(key);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
float update(void) {
|
float update(void) {
|
||||||
const uint64_t ts_now = Message::getTimeMS();
|
const uint64_t ts_now = Message::getTimeMS();
|
||||||
uint64_t ts_min_next = ts_now + ms_before_purge;
|
uint64_t ts_min_next = ts_now + ms_before_purge;
|
||||||
@ -166,10 +144,7 @@ struct TextureCache {
|
|||||||
const uint64_t ts_next = te.doAnimation(ts_now);
|
const uint64_t ts_next = te.doAnimation(ts_now);
|
||||||
te.rendered_this_frame = false;
|
te.rendered_this_frame = false;
|
||||||
ts_min_next = std::min(ts_min_next, ts_next);
|
ts_min_next = std::min(ts_min_next, ts_next);
|
||||||
} else if (
|
} else if (_cache.size() > min_count_before_purge && ts_now - te.timestamp_last_rendered >= ms_before_purge) {
|
||||||
_cache.size() > min_count_before_purge &&
|
|
||||||
ts_now - te.timestamp_last_rendered >= ms_before_purge
|
|
||||||
) {
|
|
||||||
to_purge.push_back(key);
|
to_purge.push_back(key);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -184,64 +159,31 @@ struct TextureCache {
|
|||||||
|
|
||||||
void invalidate(const std::vector<KeyType>& to_purge) {
|
void invalidate(const std::vector<KeyType>& to_purge) {
|
||||||
for (const auto& key : to_purge) {
|
for (const auto& key : to_purge) {
|
||||||
if (_to_load.count(key)) {
|
|
||||||
// TODO: only remove if not keep trying
|
|
||||||
_to_load.erase(key);
|
|
||||||
}
|
|
||||||
if (_cache.count(key)) {
|
if (_cache.count(key)) {
|
||||||
for (const auto& tex_id : _cache.at(key).textures) {
|
for (const auto& tex_id : _cache.at(key).textures) {
|
||||||
_tu.destroy(tex_id);
|
_tu.destroy(tex_id);
|
||||||
}
|
}
|
||||||
_cache.erase(key);
|
|
||||||
}
|
}
|
||||||
|
_cache.erase(key);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// returns true if there is still work queued up
|
// returns true if there is still work queued up
|
||||||
bool workLoadQueue(void) {
|
bool workLoadQueue(void) {
|
||||||
auto it = _to_load.cbegin();
|
auto it = _to_load.begin();
|
||||||
for (; it != _to_load.cend(); it++) {
|
for (; it != _to_load.end(); it++) {
|
||||||
auto new_entry_opt = _l.load(_tu, *it);
|
auto new_entry_opt = _l.load(_tu, *it);
|
||||||
if (_cache.count(*it)) {
|
if (new_entry_opt.has_value()) {
|
||||||
if (new_entry_opt.texture.has_value()) {
|
_cache.emplace(*it, new_entry_opt.value());
|
||||||
auto old_entry = _cache.at(*it); // copy
|
|
||||||
assert(!old_entry.textures.empty());
|
|
||||||
for (const auto& tex_id : old_entry.textures) {
|
|
||||||
_tu.destroy(tex_id);
|
|
||||||
}
|
|
||||||
|
|
||||||
_cache.erase(*it);
|
|
||||||
auto& new_entry = _cache[*it] = new_entry_opt.texture.value();
|
|
||||||
// TODO: make update interface and let loader handle this
|
|
||||||
//new_entry.current_texture = old_entry.current_texture; // ??
|
|
||||||
new_entry.rendered_this_frame = old_entry.rendered_this_frame;
|
|
||||||
new_entry.timestamp_last_rendered = old_entry.timestamp_last_rendered;
|
|
||||||
|
|
||||||
it = _to_load.erase(it);
|
it = _to_load.erase(it);
|
||||||
|
|
||||||
// TODO: not a good idea?
|
// TODO: not a good idea?
|
||||||
break; // end load from queue/onlyload 1 per update
|
break; // end load from queue/onlyload 1 per update
|
||||||
} else if (!new_entry_opt.keep_trying) {
|
|
||||||
// failed to load and the loader is done
|
|
||||||
it = _to_load.erase(it);
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
if (new_entry_opt.texture.has_value()) {
|
|
||||||
_cache.emplace(*it, new_entry_opt.texture.value());
|
|
||||||
_cache.at(*it).rendered_this_frame = true; // ?
|
|
||||||
it = _to_load.erase(it);
|
|
||||||
|
|
||||||
// TODO: not a good idea?
|
|
||||||
break; // end load from queue/onlyload 1 per update
|
|
||||||
} else if (!new_entry_opt.keep_trying) {
|
|
||||||
// failed to load and the loader is done
|
|
||||||
it = _to_load.erase(it);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// peek
|
// peak
|
||||||
return it != _to_load.cend();
|
return it != _to_load.end();
|
||||||
}
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -4,7 +4,6 @@
|
|||||||
#include "./image_loader_qoi.hpp"
|
#include "./image_loader_qoi.hpp"
|
||||||
#include "./image_loader_webp.hpp"
|
#include "./image_loader_webp.hpp"
|
||||||
#include "./image_loader_sdl_image.hpp"
|
#include "./image_loader_sdl_image.hpp"
|
||||||
#include "texture_uploader.hpp"
|
|
||||||
|
|
||||||
#include <solanaceae/contact/components.hpp>
|
#include <solanaceae/contact/components.hpp>
|
||||||
#include <solanaceae/tox_contacts/components.hpp>
|
#include <solanaceae/tox_contacts/components.hpp>
|
||||||
@ -120,9 +119,9 @@ static std::vector<uint8_t> generateToxIdenticon(const ToxKey& key) {
|
|||||||
return pixels;
|
return pixels;
|
||||||
}
|
}
|
||||||
|
|
||||||
TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
std::optional<TextureEntry> ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
||||||
if (!_cr.valid(c)) {
|
if (!_cr.valid(c)) {
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (_cr.all_of<Contact::Components::AvatarMemory>(c)) {
|
if (_cr.all_of<Contact::Components::AvatarMemory>(c)) {
|
||||||
@ -135,13 +134,13 @@ TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
|||||||
new_entry.width = a_m.width;
|
new_entry.width = a_m.width;
|
||||||
new_entry.height = a_m.height;
|
new_entry.height = a_m.height;
|
||||||
|
|
||||||
const auto n_t = tu.upload(a_m.data.data(), a_m.width, a_m.height);
|
const auto n_t = tu.uploadRGBA(a_m.data.data(), a_m.width, a_m.height);
|
||||||
new_entry.textures.push_back(n_t);
|
new_entry.textures.push_back(n_t);
|
||||||
new_entry.frame_duration.push_back(250);
|
new_entry.frame_duration.push_back(250);
|
||||||
|
|
||||||
std::cout << "TAL: loaded memory buffer\n";
|
std::cout << "TAL: loaded memory buffer\n";
|
||||||
|
|
||||||
return {new_entry};
|
return new_entry;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (_cr.all_of<Contact::Components::AvatarFile>(c)) {
|
if (_cr.all_of<Contact::Components::AvatarFile>(c)) {
|
||||||
@ -170,7 +169,7 @@ TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
|||||||
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
||||||
new_entry.current_texture = 0;
|
new_entry.current_texture = 0;
|
||||||
for (const auto& [ms, data] : res.frames) {
|
for (const auto& [ms, data] : res.frames) {
|
||||||
const auto n_t = tu.upload(data.data(), res.width, res.height);
|
const auto n_t = tu.uploadRGBA(data.data(), res.width, res.height);
|
||||||
new_entry.textures.push_back(n_t);
|
new_entry.textures.push_back(n_t);
|
||||||
new_entry.frame_duration.push_back(ms);
|
new_entry.frame_duration.push_back(ms);
|
||||||
}
|
}
|
||||||
@ -180,7 +179,7 @@ TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
|||||||
|
|
||||||
std::cout << "TAL: loaded image file " << a_f.file_path << "\n";
|
std::cout << "TAL: loaded image file " << a_f.file_path << "\n";
|
||||||
|
|
||||||
return {new_entry};
|
return new_entry;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} // continues if loading img fails
|
} // continues if loading img fails
|
||||||
@ -191,7 +190,7 @@ TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
|||||||
Contact::Components::ToxGroupPeerPersistent,
|
Contact::Components::ToxGroupPeerPersistent,
|
||||||
Contact::Components::ID
|
Contact::Components::ID
|
||||||
>(c)) {
|
>(c)) {
|
||||||
return {std::nullopt};
|
return std::nullopt;
|
||||||
}
|
}
|
||||||
|
|
||||||
std::vector<uint8_t> pixels;
|
std::vector<uint8_t> pixels;
|
||||||
@ -213,7 +212,7 @@ TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
|||||||
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
new_entry.timestamp_last_rendered = Message::getTimeMS();
|
||||||
new_entry.current_texture = 0;
|
new_entry.current_texture = 0;
|
||||||
|
|
||||||
const auto n_t = tu.upload(pixels.data(), 5, 5, TextureUploaderI::RGBA, TextureUploaderI::NEAREST);
|
const auto n_t = tu.uploadRGBA(pixels.data(), 5, 5, TextureUploaderI::NEAREST);
|
||||||
new_entry.textures.push_back(n_t);
|
new_entry.textures.push_back(n_t);
|
||||||
new_entry.frame_duration.push_back(250);
|
new_entry.frame_duration.push_back(250);
|
||||||
|
|
||||||
@ -222,6 +221,6 @@ TextureLoaderResult ToxAvatarLoader::load(TextureUploaderI& tu, Contact3 c) {
|
|||||||
|
|
||||||
std::cout << "TAL: generated ToxIdenticon\n";
|
std::cout << "TAL: generated ToxIdenticon\n";
|
||||||
|
|
||||||
return {new_entry};
|
return new_entry;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -14,6 +14,6 @@ class ToxAvatarLoader {
|
|||||||
|
|
||||||
public:
|
public:
|
||||||
ToxAvatarLoader(Contact3Registry& cr);
|
ToxAvatarLoader(Contact3Registry& cr);
|
||||||
TextureLoaderResult load(TextureUploaderI& tu, Contact3 c);
|
std::optional<TextureEntry> load(TextureUploaderI& tu, Contact3 c);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
310
src/tox_netprof_ui.cpp
Normal file
310
src/tox_netprof_ui.cpp
Normal file
@ -0,0 +1,310 @@
|
|||||||
|
#include "./tox_netprof_ui.hpp"
|
||||||
|
|
||||||
|
#include <imgui/imgui.h>
|
||||||
|
|
||||||
|
static const char* typedPkgIDToString(Tox_Netprof_Packet_Type type, uint8_t id) {
|
||||||
|
// pain
|
||||||
|
if (type == TOX_NETPROF_PACKET_TYPE_UDP) {
|
||||||
|
switch (id) {
|
||||||
|
case TOX_NETPROF_PACKET_ID_ZERO: return "Ping request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONE: return "Ping response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TWO: return "Get nodes request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FOUR: return "Send nodes response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_COOKIE_REQUEST: return "Cookie request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_COOKIE_RESPONSE: return "Cookie response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_CRYPTO_HS: return "Crypto handshake";
|
||||||
|
case TOX_NETPROF_PACKET_ID_CRYPTO_DATA: return "Crypto data";
|
||||||
|
case TOX_NETPROF_PACKET_ID_CRYPTO: return "Encrypted data";
|
||||||
|
case TOX_NETPROF_PACKET_ID_LAN_DISCOVERY: return "LAN discovery";
|
||||||
|
case TOX_NETPROF_PACKET_ID_GC_HANDSHAKE: return "DHT groupchat handshake";
|
||||||
|
case TOX_NETPROF_PACKET_ID_GC_LOSSLESS: return "DHT groupchat lossless";
|
||||||
|
case TOX_NETPROF_PACKET_ID_GC_LOSSY: return "DHT groupchat lossy";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_SEND_INITIAL: return "Onion send init";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_SEND_1: return "Onion send 1";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_SEND_2: return "Onion send 2";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST_OLD: return "DHT announce request (old)";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE_OLD: return "DHT announce response (old)";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_DATA_REQUEST: return "Onion data request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_DATA_RESPONSE: return "Onion data response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_REQUEST: return "DHT announce request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ANNOUNCE_RESPONSE: return "DHT announce response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_RECV_3: return "Onion receive 3";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_RECV_2: return "Onion receive 2";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONION_RECV_1: return "Onion receive 1";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FORWARD_REQUEST: return "DHT forwarding request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FORWARDING: return "DHT forwarding";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FORWARD_REPLY: return "DHT forward reply";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_SEARCH_REQUEST: return "DHT data search request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_SEARCH_RESPONSE: return "DHT data search response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_REQUEST: return "DHT data retrieve request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_DATA_RETRIEVE_RESPONSE: return "DHT data retrieve response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_REQUEST: return "DHT store announce request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_STORE_ANNOUNCE_RESPONSE: return "DHT store announce response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_BOOTSTRAP_INFO: return "Bootstrap info";
|
||||||
|
}
|
||||||
|
} else if (type == TOX_NETPROF_PACKET_TYPE_TCP) { // TODO: or client/server
|
||||||
|
switch (id) {
|
||||||
|
case TOX_NETPROF_PACKET_ID_ZERO: return "Routing request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_ONE: return "Routing response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TWO: return "Connection notification";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_DISCONNECT: return "disconnect notification";
|
||||||
|
case TOX_NETPROF_PACKET_ID_FOUR: return "Ping packet";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_PONG: return "pong packet";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_OOB_SEND: return "out-of-band send";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_OOB_RECV: return "out-of-band receive";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_ONION_REQUEST: return "onion request";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_ONION_RESPONSE: return "onion response";
|
||||||
|
case TOX_NETPROF_PACKET_ID_TCP_DATA: return "data";
|
||||||
|
//case TOX_NETPROF_PACKET_ID_BOOTSTRAP_INFO: return "Bootstrap info";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return "UNK";
|
||||||
|
}
|
||||||
|
|
||||||
|
void ToxNetprofUI::tick(float time_delta) {
|
||||||
|
if (!_enabled) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
_time_since_last_add += time_delta;
|
||||||
|
if (_time_since_last_add >= _value_add_interval) {
|
||||||
|
_time_since_last_add = 0.f; // very loose
|
||||||
|
|
||||||
|
if (_udp_tctx.empty()) {
|
||||||
|
_udp_tctx.push_back(0.f);
|
||||||
|
_udp_tctx_prev = _tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
} else {
|
||||||
|
const auto new_value = _tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
_udp_tctx.push_back(new_value - _udp_tctx_prev);
|
||||||
|
_udp_tctx_prev = new_value;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_udp_tcrx.empty()) {
|
||||||
|
_udp_tcrx.push_back(0.f);
|
||||||
|
_udp_tcrx_prev = _tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
} else {
|
||||||
|
const auto new_value = _tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
_udp_tcrx.push_back(new_value - _udp_tcrx_prev);
|
||||||
|
_udp_tcrx_prev = new_value;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_udp_tbtx.empty()) {
|
||||||
|
_udp_tbtx.push_back(0.f);
|
||||||
|
_udp_tbtx_prev = _tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
} else {
|
||||||
|
const auto new_value = _tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
_udp_tbtx.push_back(new_value - _udp_tbtx_prev);
|
||||||
|
_udp_tbtx_prev = new_value;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_udp_tbrx.empty()) {
|
||||||
|
_udp_tbrx.push_back(0.f);
|
||||||
|
_udp_tbrx_prev = _tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
} else {
|
||||||
|
const auto new_value = _tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
_udp_tbrx.push_back(new_value - _udp_tbrx_prev);
|
||||||
|
_udp_tbrx_prev = new_value;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_udp_tbrx.empty()) {
|
||||||
|
}
|
||||||
|
|
||||||
|
// TODO: limit
|
||||||
|
while (_udp_tctx.size() > 5*60) {
|
||||||
|
_udp_tctx.erase(_udp_tctx.begin());
|
||||||
|
}
|
||||||
|
|
||||||
|
while (_udp_tcrx.size() > 5*60) {
|
||||||
|
_udp_tcrx.erase(_udp_tcrx.begin());
|
||||||
|
}
|
||||||
|
|
||||||
|
while (_udp_tbtx.size() > 5*60) {
|
||||||
|
_udp_tbtx.erase(_udp_tbtx.begin());
|
||||||
|
}
|
||||||
|
|
||||||
|
while (_udp_tbrx.size() > 5*60) {
|
||||||
|
_udp_tbrx.erase(_udp_tbrx.begin());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
float ToxNetprofUI::render(float time_delta) {
|
||||||
|
{ // main window menubar injection
|
||||||
|
// assumes the window "tomato" was rendered already by cg
|
||||||
|
if (ImGui::Begin("tomato")) {
|
||||||
|
if (ImGui::BeginMenuBar()) {
|
||||||
|
if (ImGui::BeginMenu("Tox")) {
|
||||||
|
ImGui::SeparatorText("Net diagnostics");
|
||||||
|
|
||||||
|
if (ImGui::MenuItem("Breakdown table", nullptr, _show_window_table)) {
|
||||||
|
_show_window_table = !_show_window_table;
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::Checkbox("histogram logging", &_enabled);
|
||||||
|
|
||||||
|
if (ImGui::MenuItem("Netprof histograms", nullptr, _show_window_histo)) {
|
||||||
|
_show_window_histo = !_show_window_histo;
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::EndMenu();
|
||||||
|
}
|
||||||
|
ImGui::EndMenuBar();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ImGui::End();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_show_window_table) {
|
||||||
|
if (ImGui::Begin("Tox Netprof table", &_show_window_table)) {
|
||||||
|
ImGui::Text("UDP total Count tx/rx: %zu/%zu",
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_RECV)
|
||||||
|
);
|
||||||
|
ImGui::Text("UDP total Bytes tx/rx: %zu/%zu",
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_UDP, TOX_NETPROF_DIRECTION_RECV)
|
||||||
|
);
|
||||||
|
ImGui::Text("TCP total Count tx/rx: %zu/%zu (client: %zu/%zu; server: %zu/%zu)",
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_TCP, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_TCP, TOX_NETPROF_DIRECTION_RECV),
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_TCP_CLIENT, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_TCP_CLIENT, TOX_NETPROF_DIRECTION_RECV),
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_TCP_SERVER, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalCount(TOX_NETPROF_PACKET_TYPE_TCP_SERVER, TOX_NETPROF_DIRECTION_RECV)
|
||||||
|
);
|
||||||
|
ImGui::Text("TCP total Bytes tx/rx: %zu/%zu (client: %zu/%zu; server: %zu/%zu)",
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_TCP, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_TCP, TOX_NETPROF_DIRECTION_RECV),
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_TCP_CLIENT, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_TCP_CLIENT, TOX_NETPROF_DIRECTION_RECV),
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_TCP_SERVER, TOX_NETPROF_DIRECTION_SENT),
|
||||||
|
_tpi.toxNetprofGetPacketTotalBytes(TOX_NETPROF_PACKET_TYPE_TCP_SERVER, TOX_NETPROF_DIRECTION_RECV)
|
||||||
|
);
|
||||||
|
|
||||||
|
// TODO: color types (tcp/udp and maybe dht)
|
||||||
|
|
||||||
|
static float decay_rate = 3.f;
|
||||||
|
ImGui::SliderFloat("heat decay (/s)", &decay_rate, 0.f, 50.0f);
|
||||||
|
|
||||||
|
// type (udp/tcp), id/name, count tx, count rx, bytes tx, bytes rx
|
||||||
|
if (ImGui::BeginTable("per packet", 6, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_Borders | ImGuiTableFlags_RowBg | ImGuiTableFlags_ScrollY)) {
|
||||||
|
|
||||||
|
ImGui::TableSetupScrollFreeze(0, 1); // Make top row always visible
|
||||||
|
ImGui::TableSetupColumn("type");
|
||||||
|
ImGui::TableSetupColumn("pkt type");
|
||||||
|
ImGui::TableSetupColumn("count tx");
|
||||||
|
ImGui::TableSetupColumn("count rx");
|
||||||
|
ImGui::TableSetupColumn("bytes tx");
|
||||||
|
ImGui::TableSetupColumn("bytes rx");
|
||||||
|
|
||||||
|
ImGui::TableHeadersRow();
|
||||||
|
|
||||||
|
auto value_fn = [time_delta](size_t i, uint64_t value, auto& prev_map, auto& heat_map, const float scale = 0.2f) {
|
||||||
|
ImGui::TableNextColumn();
|
||||||
|
ImGui::Text("%zu", value);
|
||||||
|
if (prev_map.count(i)) {
|
||||||
|
const auto delta = value - prev_map[i];
|
||||||
|
float& heat = heat_map[i];
|
||||||
|
heat += delta * scale; // count 0.1, bytes 0.02?
|
||||||
|
|
||||||
|
// TODO: actual color function
|
||||||
|
float green = 0.7f;
|
||||||
|
if (heat > 1.f) {
|
||||||
|
green -= (heat - 1.f) * 0.1f;
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::TableSetBgColor(ImGuiTableBgTarget_CellBg, ImGui::GetColorU32(ImVec4(0.9f, green, 0.0f, heat)));
|
||||||
|
|
||||||
|
//ImGui::SameLine();
|
||||||
|
//ImGui::Text("%.2f", heat);
|
||||||
|
|
||||||
|
//heat *= 0.94f;
|
||||||
|
float decay = decay_rate * time_delta;
|
||||||
|
if (decay > 0.999f) {
|
||||||
|
decay = 0.999f;
|
||||||
|
}
|
||||||
|
heat *= (1.f - decay);
|
||||||
|
}
|
||||||
|
prev_map[i] = value;
|
||||||
|
};
|
||||||
|
|
||||||
|
for (size_t i = 0; i < 0xff; i++) {
|
||||||
|
if (i == 0x10) {
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
const auto count_sent = _tpi.toxNetprofGetPacketIdCount(TOX_NETPROF_PACKET_TYPE_UDP, i, TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
const auto count_received = _tpi.toxNetprofGetPacketIdCount(TOX_NETPROF_PACKET_TYPE_UDP, i, TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
|
||||||
|
if (count_sent == 0 && count_received == 0) {
|
||||||
|
continue; // skip empty
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::TableNextColumn();
|
||||||
|
ImGui::TextUnformatted("UDP");
|
||||||
|
|
||||||
|
ImGui::TableNextColumn();
|
||||||
|
ImGui::Text("%02zx(%s)", i, typedPkgIDToString(TOX_NETPROF_PACKET_TYPE_UDP, i));
|
||||||
|
|
||||||
|
value_fn(i, count_sent, _udp_ctx_prev, _udp_ctx_heat);
|
||||||
|
|
||||||
|
value_fn(i, count_received, _udp_crx_prev, _udp_crx_heat);
|
||||||
|
|
||||||
|
value_fn(i, _tpi.toxNetprofGetPacketIdBytes(TOX_NETPROF_PACKET_TYPE_UDP, i, TOX_NETPROF_DIRECTION_SENT), _udp_btx_prev, _udp_btx_heat, 0.005f);
|
||||||
|
|
||||||
|
value_fn(i, _tpi.toxNetprofGetPacketIdBytes(TOX_NETPROF_PACKET_TYPE_UDP, i, TOX_NETPROF_DIRECTION_RECV), _udp_brx_prev, _udp_brx_heat, 0.005f);
|
||||||
|
}
|
||||||
|
|
||||||
|
for (size_t i = 0; i <= 0x10; i++) {
|
||||||
|
const auto count_sent = _tpi.toxNetprofGetPacketIdCount(TOX_NETPROF_PACKET_TYPE_TCP, i, TOX_NETPROF_DIRECTION_SENT);
|
||||||
|
const auto count_received = _tpi.toxNetprofGetPacketIdCount(TOX_NETPROF_PACKET_TYPE_TCP, i, TOX_NETPROF_DIRECTION_RECV);
|
||||||
|
|
||||||
|
if (count_sent == 0 && count_received == 0) {
|
||||||
|
continue; // skip empty
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::TableNextColumn();
|
||||||
|
ImGui::TextUnformatted("TCP");
|
||||||
|
|
||||||
|
ImGui::TableNextColumn();
|
||||||
|
ImGui::Text("%02zx(%s)", i, typedPkgIDToString(TOX_NETPROF_PACKET_TYPE_TCP, i));
|
||||||
|
|
||||||
|
value_fn(i, count_sent, _tcp_ctx_prev, _tcp_ctx_heat);
|
||||||
|
|
||||||
|
value_fn(i, count_received, _tcp_crx_prev, _tcp_crx_heat);
|
||||||
|
|
||||||
|
value_fn(i, _tpi.toxNetprofGetPacketIdBytes(TOX_NETPROF_PACKET_TYPE_TCP, i, TOX_NETPROF_DIRECTION_SENT), _tcp_btx_prev, _tcp_btx_heat, 0.005f);
|
||||||
|
|
||||||
|
value_fn(i, _tpi.toxNetprofGetPacketIdBytes(TOX_NETPROF_PACKET_TYPE_TCP, i, TOX_NETPROF_DIRECTION_RECV), _tcp_brx_prev, _tcp_brx_heat, 0.005f);
|
||||||
|
}
|
||||||
|
|
||||||
|
ImGui::EndTable();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ImGui::End();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_show_window_histo) {
|
||||||
|
if (ImGui::Begin("Tox Netprof histograms", &_show_window_histo)) {
|
||||||
|
if (_enabled) {
|
||||||
|
const float line_height = ImGui::GetTextLineHeight();
|
||||||
|
ImGui::PlotHistogram("udp total count sent##histograms", _udp_tctx.data(), _udp_tctx.size(), 0, nullptr, 0.f, FLT_MAX, {0, 3*line_height});
|
||||||
|
ImGui::PlotHistogram("udp total count received##histograms", _udp_tcrx.data(), _udp_tcrx.size(), 0, nullptr, 0.f, FLT_MAX, {0, 3*line_height});
|
||||||
|
ImGui::PlotHistogram("udp total bytes sent##histograms", _udp_tbtx.data(), _udp_tbtx.size(), 0, nullptr, 0.f, FLT_MAX, {0, 3*line_height});
|
||||||
|
ImGui::PlotHistogram("udp total bytes received##histograms", _udp_tbrx.data(), _udp_tbrx.size(), 0, nullptr, 0.f, FLT_MAX, {0, 3*line_height});
|
||||||
|
} else {
|
||||||
|
ImGui::TextUnformatted("logging disabled!");
|
||||||
|
if (ImGui::Button("enable")) {
|
||||||
|
_enabled = true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
ImGui::End();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (_show_window_table) {
|
||||||
|
return 0.1f; // min 10fps
|
||||||
|
}
|
||||||
|
return 2.f;
|
||||||
|
}
|
||||||
|
|
54
src/tox_netprof_ui.hpp
Normal file
54
src/tox_netprof_ui.hpp
Normal file
@ -0,0 +1,54 @@
|
|||||||
|
#pragma once
|
||||||
|
|
||||||
|
#include "./tox_private_impl.hpp"
|
||||||
|
|
||||||
|
#include <cstdint>
|
||||||
|
#include <vector>
|
||||||
|
#include <map>
|
||||||
|
|
||||||
|
class ToxNetprofUI {
|
||||||
|
ToxPrivateImpl& _tpi;
|
||||||
|
|
||||||
|
bool _enabled {true};
|
||||||
|
bool _show_window_table {false};
|
||||||
|
bool _show_window_histo {false};
|
||||||
|
|
||||||
|
// table delta
|
||||||
|
std::map<uint8_t, uint64_t> _udp_ctx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _udp_crx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _udp_btx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _udp_brx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _tcp_ctx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _tcp_crx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _tcp_btx_prev;
|
||||||
|
std::map<uint8_t, uint64_t> _tcp_brx_prev;
|
||||||
|
|
||||||
|
// table heat
|
||||||
|
std::map<uint8_t, float> _udp_ctx_heat;
|
||||||
|
std::map<uint8_t, float> _udp_crx_heat;
|
||||||
|
std::map<uint8_t, float> _udp_btx_heat;
|
||||||
|
std::map<uint8_t, float> _udp_brx_heat;
|
||||||
|
std::map<uint8_t, float> _tcp_ctx_heat;
|
||||||
|
std::map<uint8_t, float> _tcp_crx_heat;
|
||||||
|
std::map<uint8_t, float> _tcp_btx_heat;
|
||||||
|
std::map<uint8_t, float> _tcp_brx_heat;
|
||||||
|
|
||||||
|
// histogram totals
|
||||||
|
uint64_t _udp_tctx_prev;
|
||||||
|
uint64_t _udp_tcrx_prev;
|
||||||
|
uint64_t _udp_tbtx_prev;
|
||||||
|
uint64_t _udp_tbrx_prev;
|
||||||
|
std::vector<float> _udp_tctx;
|
||||||
|
std::vector<float> _udp_tcrx;
|
||||||
|
std::vector<float> _udp_tbtx;
|
||||||
|
std::vector<float> _udp_tbrx;
|
||||||
|
|
||||||
|
const float _value_add_interval {1.f}; // every second
|
||||||
|
float _time_since_last_add {0.f};
|
||||||
|
|
||||||
|
public:
|
||||||
|
ToxNetprofUI(ToxPrivateImpl& tpi) : _tpi(tpi) {}
|
||||||
|
|
||||||
|
void tick(float time_delta);
|
||||||
|
float render(float time_delta);
|
||||||
|
};
|
@ -32,4 +32,22 @@ struct ToxPrivateImpl : public ToxPrivateI {
|
|||||||
return {std::nullopt, err};
|
return {std::nullopt, err};
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// TODO: add to interface
|
||||||
|
uint64_t toxNetprofGetPacketIdCount(Tox_Netprof_Packet_Type type, uint8_t id, Tox_Netprof_Direction direction) {
|
||||||
|
return tox_netprof_get_packet_id_count(_tox, type, id, direction);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t toxNetprofGetPacketTotalCount(Tox_Netprof_Packet_Type type, Tox_Netprof_Direction direction) {
|
||||||
|
return tox_netprof_get_packet_total_count(_tox, type, direction);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t toxNetprofGetPacketIdBytes(Tox_Netprof_Packet_Type type, uint8_t id, Tox_Netprof_Direction direction) {
|
||||||
|
return tox_netprof_get_packet_id_bytes(_tox, type, id, direction);
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t toxNetprofGetPacketTotalBytes(Tox_Netprof_Packet_Type type, Tox_Netprof_Direction direction) {
|
||||||
|
return tox_netprof_get_packet_total_bytes(_tox, type, direction);
|
||||||
|
}
|
||||||
|
|
||||||
};
|
};
|
||||||
|
Reference in New Issue
Block a user