stephenseo/UDPConnection
1
0
Fork 0
Code Issues Pull requests Actions Projects Releases Wiki Activity

Minor fixes/refactorings

Browse source
This commit is contained in:
Stephen Seo 2019-11-27 20:12:57 +09:00
parent 04fa5ca017
commit 69256839ac
2 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
src/UDPConnection.cpp
View file

@ -1879,6 +1879,7 @@ void UDPC_destroy(UDPC_HContext ctx) {
delete[] optE.value().v.pk;
}
}
UDPC_ctx->_contextIdentifier = 0;
delete UDPC_ctx;
}
}
@ -1977,7 +1978,8 @@ int UDPC_set_accept_new_connections(UDPC_HContext ctx, int isAccepting) {
if(!c) {
return 0;
}
return c->isAcceptNewConnections.exchange(isAccepting == 0 ? false : true);
return c->isAcceptNewConnections.exchange(isAccepting == 0 ? false : true)
? 1 : 0;
}
void UDPC_drop_connection(UDPC_HContext ctx, UDPC_ConnectionId connectionId, int dropAllWithAddr) {
@ -2131,14 +2133,17 @@ int UDPC_set_libsodium_key_easy(UDPC_HContext ctx, unsigned char *sk) {
return UDPC_set_libsodium_keys(ctx, sk, pk);
}
void UDPC_unset_libsodium_keys(UDPC_HContext ctx) {
int UDPC_unset_libsodium_keys(UDPC_HContext ctx) {
UDPC::Context *c = UDPC::verifyContext(ctx);
if(!c || !c->flags.test(2)) {
return;
return 0;
}
std::lock_guard<std::mutex> lock(c->mutex);
c->keysSet.store(false);
std::memset(c->pk, 0, crypto_sign_PUBLICKEYBYTES);
std::memset(c->sk, 0, crypto_sign_SECRETKEYBYTES);
return 1;
}
const char *UDPC_atostr_cid(UDPC_HContext ctx, UDPC_ConnectionId connectionId) {

2
src/UDPConnection.h
View file

@ -190,7 +190,7 @@ int UDPC_set_libsodium_keys(UDPC_HContext ctx, unsigned char *sk, unsigned char
int UDPC_set_libsodium_key_easy(UDPC_HContext ctx, unsigned char *sk);
void UDPC_unset_libsodium_keys(UDPC_HContext ctx);
int UDPC_unset_libsodium_keys(UDPC_HContext ctx);
const char *UDPC_atostr_cid(UDPC_HContext ctx, UDPC_ConnectionId connectionId);