diff options
| author | Doug Hogan <[email protected]> | 2019-01-15 08:04:32 -0800 |
|---|---|---|
| committer | Doug Hogan <[email protected]> | 2019-01-15 08:38:09 -0800 |
| commit | 7cb2e25a026e6d9ff2cdb078504163de5f71c29a (patch) | |
| tree | 5e754c51987515e2688d6fb1f7a70512207fe27e /lib/crypto/c_src/hmac.c | |
| parent | b36527d7bee7c264bfac9b36fd6dc58b2f15a7f0 (diff) | |
| download | otp-7cb2e25a026e6d9ff2cdb078504163de5f71c29a.tar.gz otp-7cb2e25a026e6d9ff2cdb078504163de5f71c29a.tar.bz2 otp-7cb2e25a026e6d9ff2cdb078504163de5f71c29a.zip | |
Convert argc checks to ASSERTs per PR
Diffstat (limited to 'lib/crypto/c_src/hmac.c')
| -rw-r--r-- | lib/crypto/c_src/hmac.c | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/lib/crypto/c_src/hmac.c b/lib/crypto/c_src/hmac.c index 9418e7950e..c41e50eb35 100644 --- a/lib/crypto/c_src/hmac.c +++ b/lib/crypto/c_src/hmac.c @@ -56,8 +56,7 @@ ERL_NIF_TERM hmac_nif(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) ERL_NIF_TERM ret; unsigned char *outp; - if (argc != 3 && argc != 4) - goto bad_arg; + ASSERT(argc == 3 || argc == 4); if ((digp = get_digest_type(argv[0])) == NULL) goto bad_arg; @@ -125,8 +124,8 @@ ERL_NIF_TERM hmac_init_nif(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) ERL_NIF_TERM ret; struct hmac_context *obj = NULL; - if (argc != 2) - goto bad_arg; + ASSERT(argc == 2); + if ((digp = get_digest_type(argv[0])) == NULL) goto bad_arg; if (!enif_inspect_iolist_as_binary(env, argv[1], &key)) @@ -180,8 +179,8 @@ ERL_NIF_TERM hmac_update_nif(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[] ErlNifBinary data; struct hmac_context *obj = NULL; - if (argc != 2) - goto bad_arg; + ASSERT(argc == 2); + if (!enif_get_resource(env, argv[0], hmac_context_rtype, (void**)&obj)) goto bad_arg; if (!enif_inspect_iolist_as_binary(env, argv[1], &data)) @@ -223,8 +222,8 @@ ERL_NIF_TERM hmac_final_nif(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]) unsigned int req_len = 0; unsigned int mac_len; - if (argc != 1 && argc != 2) - goto bad_arg; + ASSERT(argc == 1 || argc == 2); + if (!enif_get_resource(env, argv[0], hmac_context_rtype, (void**)&obj)) goto bad_arg; if (argc == 2) { |