diff options
| author | Andreas Schneider <mail@cynapses.org> | 2009-04-01 21:24:16 +0000 |
|---|---|---|
| committer | Andreas Schneider <mail@cynapses.org> | 2009-04-01 21:24:16 +0000 |
| commit | 891539af6cbdbdfabb8d37fb491f7f8cadf9823c (patch) | |
| tree | 1699cb47cd68dc35a972f8b5e1d834946c48d944 /libssh/keys.c | |
| parent | 1b627b386763e22a7b8f7a7c8de9610c1da56cd2 (diff) | |
| download | libssh-891539af6cbdbdfabb8d37fb491f7f8cadf9823c.tar.gz libssh-891539af6cbdbdfabb8d37fb491f7f8cadf9823c.tar.xz libssh-891539af6cbdbdfabb8d37fb491f7f8cadf9823c.zip | |
Add memory error checks for crypto wrapper functions.
git-svn-id: svn+ssh://svn.berlios.de/svnroot/repos/libssh/trunk@330 7dcaeef0-15fb-0310-b436-a5af3365683c
Diffstat (limited to 'libssh/keys.c')
| -rw-r--r-- | libssh/keys.c | 11 |
1 files changed, 9 insertions, 2 deletions
diff --git a/libssh/keys.c b/libssh/keys.c index 82d5ce3..cdef39f 100644 --- a/libssh/keys.c +++ b/libssh/keys.c @@ -715,7 +715,11 @@ STRING *ssh_do_sign(SSH_SESSION *session,BUFFER *sigbuf, PRIVATE_KEY *privatekey return NULL; } string_fill(session_str,crypto->session_id,SHA_DIGEST_LEN); - ctx=sha1_init(); + ctx = sha1_init(); + if (ctx == NULL) { + SAFE_FREE(session_str); + return NULL; + } sha1_update(ctx,session_str,string_len(session_str)+4); SAFE_FREE(session_str); sha1_update(ctx,buffer_get(sigbuf),buffer_get_len(sigbuf)); @@ -818,7 +822,10 @@ STRING *ssh_sign_session_id(SSH_SESSION *session, PRIVATE_KEY *privatekey){ #ifdef HAVE_LIBGCRYPT gcry_sexp_t data_sexp; #endif - ctx=sha1_init(); + ctx = sha1_init(); + if (ctx == NULL) { + return NULL; + } sha1_update(ctx,crypto->session_id,SHA_DIGEST_LEN); sha1_final(hash+1,ctx); hash[0]=0; |