add: added a test function for nonce distance

[proxmark3-svn] / client / reveng / reveng.c

diff --git a/client/reveng/reveng.c b/client/reveng/reveng.c
index dd50987c308fd7b47c2ccc503e99289b8897880c..d82353f8ad0d4046def9f7b0c22c5f12b5367373 100644 (file)
--- a/client/reveng/reveng.c
+++ b/client/reveng/reveng.c
@@ -141,8 +141,10 @@ reveng(const model_t *guess, const poly_t qpoly, int rflags, int args, const pol
                engini(&resc, &result, guess->spoly, guess->flags, args, argpolys);
 
 requit:
                engini(&resc, &result, guess->spoly, guess->flags, args, argpolys);
 
 requit:

-       if(!(result = realloc(result, ++resc * sizeof(model_t))))
+       if(!(result = realloc(result, ++resc * sizeof(model_t)))) {

                uerror("cannot reallocate result array");
                uerror("cannot reallocate result array");

+               return NULL;
+       }

        rptr = result + resc - 1;
        rptr->spoly  = pzero;
        rptr->init   = pzero;
        rptr = result + resc - 1;
        rptr->spoly  = pzero;
        rptr->init   = pzero;


@@ -471,9 +473,12 @@ chkres(int *resc, model_t **result, const poly_t divisor, const poly_t init, int
        pfree(&xor);
        if(aptr != eptr) return;
 
        pfree(&xor);
        if(aptr != eptr) return;
 

-       if(!(*result = realloc(*result, ++*resc * sizeof(model_t))))
+       *result = realloc(*result, ++*resc * sizeof(model_t));
+       if (!*result) {

                uerror("cannot reallocate result array");
                uerror("cannot reallocate result array");

-
+               return;
+       }
+       

        rptr = *result + *resc - 1;
        rptr->spoly  = pclone(divisor);
        rptr->init   = pclone(init);
        rptr = *result + *resc - 1;
        rptr->spoly  = pclone(divisor);
        rptr->init   = pclone(init);