mirror of
https://github.com/Tha14/toxic.git
synced 2024-11-22 20:33:02 +01:00
Dynamically allocate memory for save file instead of using stack
This commit is contained in:
parent
32442b6286
commit
cf16849b37
@ -133,17 +133,24 @@ void kill_friendlist(void)
|
|||||||
#define TEMP_BLOCKLIST_EXT ".tmp"
|
#define TEMP_BLOCKLIST_EXT ".tmp"
|
||||||
static int save_blocklist(char *path)
|
static int save_blocklist(char *path)
|
||||||
{
|
{
|
||||||
if (path == NULL)
|
if (path == NULL) {
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
int len = sizeof(BlockedFriend) * Blocked.num_blocked;
|
int len = sizeof(BlockedFriend) * Blocked.num_blocked;
|
||||||
char data[len];
|
char *data = malloc(len * sizeof(char));
|
||||||
|
|
||||||
|
if (data == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
int i, count = 0;
|
int i, count = 0;
|
||||||
|
|
||||||
for (i = 0; i < Blocked.max_idx; ++i) {
|
for (i = 0; i < Blocked.max_idx; ++i) {
|
||||||
if (count > Blocked.num_blocked)
|
if (count > Blocked.num_blocked) {
|
||||||
|
free(data);
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
if (Blocked.list[i].active) {
|
if (Blocked.list[i].active) {
|
||||||
BlockedFriend tmp;
|
BlockedFriend tmp;
|
||||||
@ -164,8 +171,11 @@ static int save_blocklist(char *path)
|
|||||||
|
|
||||||
/* Blocklist is empty, we can remove the empty file */
|
/* Blocklist is empty, we can remove the empty file */
|
||||||
if (count == 0) {
|
if (count == 0) {
|
||||||
if (remove(path) != 0)
|
free(data);
|
||||||
|
|
||||||
|
if (remove(path) != 0) {
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
@ -175,19 +185,24 @@ static int save_blocklist(char *path)
|
|||||||
|
|
||||||
FILE *fp = fopen(temp_path, "wb");
|
FILE *fp = fopen(temp_path, "wb");
|
||||||
|
|
||||||
if (fp == NULL)
|
if (fp == NULL) {
|
||||||
|
free(data);
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
if (fwrite(data, len, 1, fp) != 1) {
|
if (fwrite(data, len, 1, fp) != 1) {
|
||||||
fprintf(stderr, "Failed to write blocklist data.\n");
|
fprintf(stderr, "Failed to write blocklist data.\n");
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
|
free(data);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
|
free(data);
|
||||||
|
|
||||||
if (rename(temp_path, path) != 0)
|
if (rename(temp_path, path) != 0) {
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
28
src/toxic.c
28
src/toxic.c
@ -584,25 +584,36 @@ static void first_time_encrypt(const char *msg)
|
|||||||
#define TEMP_PROFILE_EXT ".tmp"
|
#define TEMP_PROFILE_EXT ".tmp"
|
||||||
int store_data(Tox *m, const char *path)
|
int store_data(Tox *m, const char *path)
|
||||||
{
|
{
|
||||||
if (path == NULL)
|
if (path == NULL) {
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
char temp_path[strlen(path) + strlen(TEMP_PROFILE_EXT) + 1];
|
char temp_path[strlen(path) + strlen(TEMP_PROFILE_EXT) + 1];
|
||||||
snprintf(temp_path, sizeof(temp_path), "%s%s", path, TEMP_PROFILE_EXT);
|
snprintf(temp_path, sizeof(temp_path), "%s%s", path, TEMP_PROFILE_EXT);
|
||||||
|
|
||||||
FILE *fp = fopen(temp_path, "wb");
|
FILE *fp = fopen(temp_path, "wb");
|
||||||
|
|
||||||
if (fp == NULL)
|
if (fp == NULL) {
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
size_t data_len = tox_get_savedata_size(m);
|
size_t data_len = tox_get_savedata_size(m);
|
||||||
char data[data_len];
|
char *data = malloc(data_len * sizeof(char));
|
||||||
|
|
||||||
|
if (data == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
tox_get_savedata(m, (uint8_t *) data);
|
tox_get_savedata(m, (uint8_t *) data);
|
||||||
|
|
||||||
if (user_password.data_is_encrypted && !arg_opts.unencrypt_data) {
|
if (user_password.data_is_encrypted && !arg_opts.unencrypt_data) {
|
||||||
size_t enc_len = data_len + TOX_PASS_ENCRYPTION_EXTRA_LENGTH;
|
size_t enc_len = data_len + TOX_PASS_ENCRYPTION_EXTRA_LENGTH;
|
||||||
char enc_data[enc_len];
|
char *enc_data = malloc(enc_len * sizeof(char));
|
||||||
|
|
||||||
|
if (enc_data == NULL) {
|
||||||
|
free(data);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
TOX_ERR_ENCRYPTION err;
|
TOX_ERR_ENCRYPTION err;
|
||||||
tox_pass_encrypt((uint8_t *) data, data_len, (uint8_t *) user_password.pass, user_password.len,
|
tox_pass_encrypt((uint8_t *) data, data_len, (uint8_t *) user_password.pass, user_password.len,
|
||||||
@ -611,26 +622,33 @@ int store_data(Tox *m, const char *path)
|
|||||||
if (err != TOX_ERR_ENCRYPTION_OK) {
|
if (err != TOX_ERR_ENCRYPTION_OK) {
|
||||||
fprintf(stderr, "tox_pass_encrypt() failed with error %d\n", err);
|
fprintf(stderr, "tox_pass_encrypt() failed with error %d\n", err);
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
|
free(data);
|
||||||
|
free(enc_data);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (fwrite(enc_data, enc_len, 1, fp) != 1) {
|
if (fwrite(enc_data, enc_len, 1, fp) != 1) {
|
||||||
fprintf(stderr, "Failed to write profile data.\n");
|
fprintf(stderr, "Failed to write profile data.\n");
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
|
free(data);
|
||||||
|
free(enc_data);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
} else { /* data will not be encrypted */
|
} else { /* data will not be encrypted */
|
||||||
if (fwrite(data, data_len, 1, fp) != 1) {
|
if (fwrite(data, data_len, 1, fp) != 1) {
|
||||||
fprintf(stderr, "Failed to write profile data.\n");
|
fprintf(stderr, "Failed to write profile data.\n");
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
|
free(data);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fclose(fp);
|
fclose(fp);
|
||||||
|
free(data);
|
||||||
|
|
||||||
if (rename(temp_path, path) != 0)
|
if (rename(temp_path, path) != 0) {
|
||||||
return -1;
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user