misc: Avoid the 4KB stack buffer in ssh_path_expand_escape
Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com> Reviewed-by: Jakub Jelen <jjelen@redhat.com> Change-Id: I908ef4dfa960bf89f8e42f99af2f8bcdbb006bc8
Этот коммит содержится в:
родитель
f2bd44969b
Коммит
9eba361ca2
28
src/misc.c
28
src/misc.c
@ -1105,8 +1105,9 @@ char *ssh_path_expand_tilde(const char *d) {
|
||||
*/
|
||||
char *ssh_path_expand_escape(ssh_session session, const char *s) {
|
||||
char host[NI_MAXHOST];
|
||||
char buf[MAX_BUF_SIZE];
|
||||
char *r, *x = NULL;
|
||||
char *buf = NULL;
|
||||
char *r = NULL;
|
||||
char *x = NULL;
|
||||
const char *p;
|
||||
size_t i, l;
|
||||
|
||||
@ -1122,6 +1123,13 @@ char *ssh_path_expand_escape(ssh_session session, const char *s) {
|
||||
return NULL;
|
||||
}
|
||||
|
||||
buf = malloc(MAX_BUF_SIZE);
|
||||
if (buf == NULL) {
|
||||
ssh_set_error_oom(session);
|
||||
free(r);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
p = r;
|
||||
buf[0] = '\0';
|
||||
|
||||
@ -1131,6 +1139,7 @@ char *ssh_path_expand_escape(ssh_session session, const char *s) {
|
||||
buf[i] = *p;
|
||||
i++;
|
||||
if (i >= MAX_BUF_SIZE) {
|
||||
free(buf);
|
||||
free(r);
|
||||
return NULL;
|
||||
}
|
||||
@ -1177,12 +1186,14 @@ char *ssh_path_expand_escape(ssh_session session, const char *s) {
|
||||
default:
|
||||
ssh_set_error(session, SSH_FATAL,
|
||||
"Wrong escape sequence detected");
|
||||
free(buf);
|
||||
free(r);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
if (x == NULL) {
|
||||
ssh_set_error_oom(session);
|
||||
free(buf);
|
||||
free(r);
|
||||
return NULL;
|
||||
}
|
||||
@ -1191,19 +1202,26 @@ char *ssh_path_expand_escape(ssh_session session, const char *s) {
|
||||
if (i >= MAX_BUF_SIZE) {
|
||||
ssh_set_error(session, SSH_FATAL,
|
||||
"String too long");
|
||||
free(buf);
|
||||
free(x);
|
||||
free(r);
|
||||
return NULL;
|
||||
}
|
||||
l = strlen(buf);
|
||||
strncpy(buf + l, x, sizeof(buf) - l - 1);
|
||||
strncpy(buf + l, x, MAX_BUF_SIZE - l - 1);
|
||||
buf[i] = '\0';
|
||||
SAFE_FREE(x);
|
||||
}
|
||||
|
||||
free(r);
|
||||
return strdup(buf);
|
||||
#undef MAX_BUF_SIZE
|
||||
|
||||
/* strip the unused space by realloc */
|
||||
x = realloc(buf, strlen(buf) + 1);
|
||||
if (x == NULL) {
|
||||
ssh_set_error_oom(session);
|
||||
free(buf);
|
||||
}
|
||||
return x;
|
||||
}
|
||||
|
||||
/**
|
||||
|
Загрузка…
Ссылка в новой задаче
Block a user