mirror of
https://github.com/ostreedev/ostree.git
synced 2025-01-23 02:05:01 +03:00
libotutil/checksum-utils: Fix memory management
Ostree uses the OtChecksum data structure as a wrapper around GChecksum (depending on what libraries are available at compile time). According to the docs for g_checksum_get_digest(), a GChecksum value can no longer be updated after that function is called. Ostree enforces this by setting "initialized" to FALSE after getting the digest, but this leads to ot_checksum_clear() avoiding freeing any memory, leading to leaks. So this commit adds a "closed" value that gets set when getting a digest and checked when updating the value, so the initialized value can be used only for memory management. Closes: #1521 Approved by: jlebon
This commit is contained in:
parent
2be4631738
commit
9721be34e1
@ -54,6 +54,7 @@ ot_bin2hex (char *out_buf, const guint8 *inbuf, gsize len)
|
||||
*/
|
||||
typedef struct {
|
||||
gboolean initialized;
|
||||
gboolean closed;
|
||||
#if defined(HAVE_OPENSSL)
|
||||
EVP_MD_CTX *checksum;
|
||||
#elif defined(HAVE_GNUTLS)
|
||||
@ -84,6 +85,7 @@ ot_checksum_init (OtChecksum *checksum)
|
||||
real->digest_len = g_checksum_type_get_length (G_CHECKSUM_SHA256);
|
||||
#endif
|
||||
g_assert_cmpint (real->digest_len, ==, _OSTREE_SHA256_DIGEST_LEN);
|
||||
real->closed = FALSE;
|
||||
real->initialized = TRUE;
|
||||
}
|
||||
|
||||
@ -94,6 +96,7 @@ ot_checksum_update (OtChecksum *checksum,
|
||||
{
|
||||
OtRealChecksum *real = (OtRealChecksum*)checksum;
|
||||
g_return_if_fail (real->initialized);
|
||||
g_return_if_fail (!real->closed);
|
||||
#if defined(HAVE_OPENSSL)
|
||||
g_assert (EVP_DigestUpdate (real->checksum, buf, len));
|
||||
#elif defined(HAVE_GNUTLS)
|
||||
@ -130,7 +133,7 @@ ot_checksum_get_digest (OtChecksum *checksum,
|
||||
{
|
||||
OtRealChecksum *real = (OtRealChecksum*)checksum;
|
||||
ot_checksum_get_digest_internal (real, buf, buflen);
|
||||
real->initialized = FALSE;
|
||||
real->closed = TRUE;
|
||||
}
|
||||
|
||||
void
|
||||
@ -143,7 +146,6 @@ ot_checksum_get_hexdigest (OtChecksum *checksum,
|
||||
guint8 digest_buf[digest_len];
|
||||
ot_checksum_get_digest (checksum, digest_buf, digest_len);
|
||||
ot_bin2hex (buf, (guint8*)digest_buf, digest_len);
|
||||
real->initialized = FALSE;
|
||||
}
|
||||
|
||||
void
|
||||
|
Loading…
x
Reference in New Issue
Block a user