diff options
author | Lukas Larsson <lukas@erlang-solutions.com> | 2011-07-22 12:31:56 +0200 |
---|---|---|
committer | Lukas Larsson <lukas@erlang-solutions.com> | 2011-08-01 17:29:46 +0200 |
commit | f2490767798cbc3ea94d603afb75be2eb65fe564 (patch) | |
tree | 484e6102aedf6a70ed8523ad59a7a29478b48c44 /lib/asn1 | |
parent | 82b32acf0201a6e389b41d0cce18fd586043d155 (diff) | |
download | otp-f2490767798cbc3ea94d603afb75be2eb65fe564.tar.gz otp-f2490767798cbc3ea94d603afb75be2eb65fe564.tar.bz2 otp-f2490767798cbc3ea94d603afb75be2eb65fe564.zip |
Update to use enif_alloc instead of malloc
Diffstat (limited to 'lib/asn1')
-rw-r--r-- | lib/asn1/c_src/asn1_erl_nif.c | 15 |
1 files changed, 9 insertions, 6 deletions
diff --git a/lib/asn1/c_src/asn1_erl_nif.c b/lib/asn1/c_src/asn1_erl_nif.c index 1a1ec09c39..9c9f83bc2a 100644 --- a/lib/asn1/c_src/asn1_erl_nif.c +++ b/lib/asn1/c_src/asn1_erl_nif.c @@ -1155,11 +1155,11 @@ int ber_encode_length(size_t size, mem_chunk_t **curr, unsigned int *count) { } mem_chunk_t *ber_new_chunk(unsigned int length) { - mem_chunk_t *new = malloc(sizeof(mem_chunk_t)); + mem_chunk_t *new = enif_alloc(sizeof(mem_chunk_t)); if (new == NULL) return NULL; new->next = NULL; - new->top = malloc(sizeof(char) * length); + new->top = enif_alloc(sizeof(char) * length); if (new->top == NULL) { free(new); return NULL; @@ -1174,8 +1174,8 @@ void ber_free_chunks(mem_chunk_t *chunk) { while (next != NULL) { curr = next; next = curr->next; - free(curr->top); - free(curr); + enif_free(curr->top); + enif_free(curr); } } @@ -1243,18 +1243,21 @@ static ERL_NIF_TERM encode_ber_tlv(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) { ErlNifBinary out_binary; unsigned int length = 0, pos = 0; + int encode_err; mem_chunk_t *curr, *top; ERL_NIF_TERM err_code; curr = ber_new_chunk(40); - if ((ber_encode(env, argv[0], &curr, &length)) + if ((encode_err = ber_encode(env, argv[0], &curr, &length)) <= ASN1_ERROR) { - err_code = enif_make_uint(env, 0); + ber_free_chunks(curr); + err_code = enif_make_int(env, encode_err); return enif_make_tuple2(env, enif_make_atom(env, "error"), err_code); } if (!enif_alloc_binary(length, &out_binary)) { + ber_free_chunks(curr); return enif_make_tuple2(env, enif_make_atom(env, "error"), enif_make_atom(env,"oom")); } |