ClearML/dropbear
1
0
Fork 0
mirror of https://github.com/clearml/dropbear synced 2025-03-13 07:08:26 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Use buf_burn_free() instead of two calls

Browse Source
This commit is contained in:
Matt Johnston 2022-03-29 23:47:30 +08:00
parent 81e3977360
commit cb9a00951f
9 changed files with 18 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
bignum.c
View File

@ -100,5 +100,5 @@ void hash_process_mp(const struct ltc_hash_descriptor *hash_desc,
plus header + some leeway*/
buf_putmpint(buf, mp);
hash_desc->process(hs, buf->data, buf->len);
buf_free(buf);
buf_burn_free(buf);
}

6
buffer.c
View File

@ -55,11 +55,13 @@ void buf_free(buffer* buf) {
m_free(buf);
}
/* overwrite the contents of the buffer to clear it */
void buf_burn(const buffer* buf) {
/* overwrite the contents of the buffer then free it */
void buf_burn_free(buffer* buf) {
m_burn(buf->data, buf->size);
m_free(buf);
}
/* resize a buffer, pos and len will be repositioned if required when
* downsizing */
buffer* buf_resize(buffer *buf, unsigned int newsize) {

2
buffer.h
View File

@ -44,7 +44,7 @@ buffer * buf_new(unsigned int size);
/* Possibly returns a new buffer*, like realloc() */
buffer * buf_resize(buffer *buf, unsigned int newsize);
void buf_free(buffer* buf);
void buf_burn(const buffer* buf);
void buf_burn_free(buffer* buf);
buffer* buf_newcopy(const buffer* buf);
void buf_setlen(buffer* buf, unsigned int len);
void buf_incrlen(buffer* buf, unsigned int incr);

6
common-kex.c
View File

@ -306,8 +306,7 @@ static void gen_new_keys() {
mp_clear(ses.dh_K);
m_free(ses.dh_K);
hash_desc->process(&hs, ses.hash->data, ses.hash->len);
buf_burn(ses.hash);
buf_free(ses.hash);
buf_burn_free(ses.hash);
ses.hash = NULL;
if (IS_DROPBEAR_CLIENT) {
@ -803,8 +802,7 @@ void finish_kexhashbuf(void) {
}
#endif
buf_burn(ses.kexhashbuf);
buf_free(ses.kexhashbuf);
buf_burn_free(ses.kexhashbuf);
m_burn(&hs, sizeof(hash_state));
ses.kexhashbuf = NULL;

3
common-runopts.c
View File

@ -57,8 +57,7 @@ int readhostkey(const char * filename, sign_key * hostkey,
ret = DROPBEAR_SUCCESS;
out:
buf_burn(buf);
buf_free(buf);
buf_burn_free(buf);
return ret;
}

3
common-session.c
View File

@ -285,8 +285,7 @@ static void cleanup_buf(buffer **buf) {
if (!*buf) {
return;
}
buf_burn(*buf);
buf_free(*buf);
buf_burn_free(*buf);
*buf = NULL;
}

3
dropbearkey.c
View File

@ -309,8 +309,7 @@ static int printpubfile(const char* filename) {
err = DROPBEAR_SUCCESS;
out:
buf_burn(buf);
buf_free(buf);
buf_burn_free(buf);
buf = NULL;
if (key) {
sign_key_free(key);

3
gensignkey.c
View File

@ -181,8 +181,7 @@ int signkey_generate(enum signkey_type keytype, int bits, const char* filename,
out:
if (buf) {
buf_burn(buf);
buf_free(buf);
buf_burn_free(buf);
}
if (fn_temp) {

18
keyimport.c
View File

@ -504,16 +504,14 @@ static struct openssh_key *load_openssh_key(const char *filename)
}
if (buf) {
buf_burn(buf);
buf_free(buf);
buf_burn_free(buf);
}
m_burn(buffer, sizeof(buffer));
return ret;
error:
if (buf) {
buf_burn(buf);
buf_free(buf);
buf_burn_free(buf);
}
m_burn(buffer, sizeof(buffer));
if (ret) {
@ -898,8 +896,7 @@ static sign_key *openssh_read(const char *filename, const char * UNUSED(passphra
error:
if (blobbuf) {
buf_burn(blobbuf);
buf_free(blobbuf);
buf_burn_free(blobbuf);
}
m_burn(key->keyblob, key->keyblob_size);
m_free(key->keyblob);
@ -1070,8 +1067,7 @@ static int openssh_write(const char *filename, sign_key *key,
outblob = (unsigned char*)m_malloc(outlen);
memcpy(outblob, buf->data, buf->len);
buf_burn(buf);
buf_free(buf);
buf_burn_free(buf);
buf = NULL;
header = "-----BEGIN OPENSSH PRIVATE KEY-----\n";
@ -1133,12 +1129,10 @@ static int openssh_write(const char *filename, sign_key *key,
m_free(outblob);
}
if (keyblob) {
buf_burn(keyblob);
buf_free(keyblob);
buf_burn_free(keyblob);
}
if (extrablob) {
buf_burn(extrablob);
buf_free(extrablob);
buf_burn_free(extrablob);
}
return ret;
}