@marshmellows memory fixes for reveng1.30 in client..

This commit is contained in:
iceman1001 2015-06-09 21:32:22 +02:00
parent 817d709615
commit dd1df4901d
2 changed files with 22 additions and 45 deletions

View file

@ -90,9 +90,9 @@ int GetModels(char *Models[], int *count, uint32_t *width){
PZERO, /* check value unused */ PZERO, /* check value unused */
NULL /* no model name */ NULL /* no model name */
}; };
int ibperhx = 8;//, obperhx = 8; int ibperhx = 8;//, obperhx = 8;
int rflags = 0, uflags = 0; /* search and UI flags */ int rflags = 0, uflags = 0; /* search and UI flags */
poly_t apoly, crc, qpoly = PZERO, *apolys = NULL, *pptr = NULL, *qptr = NULL; poly_t apoly, crc, qpoly = PZERO, *apolys = NULL, *pptr = NULL, *qptr = NULL;
model_t pset = model, *candmods, *mptr; model_t pset = model, *candmods, *mptr;
@ -107,7 +107,6 @@ int GetModels(char *Models[], int *count, uint32_t *width){
int Cnt = 0; int Cnt = 0;
if (*width == 0) { //reveng -D if (*width == 0) { //reveng -D
*count = mcount(); *count = mcount();
if(!*count) if(!*count)
return uerr("no preset models available"); return uerr("no preset models available");
@ -115,7 +114,6 @@ int GetModels(char *Models[], int *count, uint32_t *width){
mbynum(&model, mode); mbynum(&model, mode);
mcanon(&model); mcanon(&model);
size_t size = (model.name && *model.name) ? strlen(model.name) : 6; size_t size = (model.name && *model.name) ? strlen(model.name) : 6;
char *tmp = calloc(size+1, sizeof(char)); char *tmp = calloc(size+1, sizeof(char));
if (tmp==NULL) if (tmp==NULL)
return uerr("out of memory?"); return uerr("out of memory?");
@ -123,6 +121,7 @@ int GetModels(char *Models[], int *count, uint32_t *width){
memcpy(tmp, model.name, size); memcpy(tmp, model.name, size);
Models[mode] = tmp; Models[mode] = tmp;
} }
mfree(&model);
} else { //reveng -s } else { //reveng -s
if(~model.flags & P_MULXN) if(~model.flags & P_MULXN)
@ -209,21 +208,16 @@ int GetModels(char *Models[], int *count, uint32_t *width){
if(uflags & C_RESULT) { if(uflags & C_RESULT) {
for(qptr = apolys; qptr < pptr; ++qptr) for(qptr = apolys; qptr < pptr; ++qptr)
pfree(qptr); pfree(qptr);
return 1;
} }
if(!(model.flags & P_REFIN) != !(model.flags & P_REFOUT)) if(!(model.flags & P_REFIN) != !(model.flags & P_REFOUT))
return uerr("cannot search for crossed-endian models"); return uerr("cannot search for crossed-endian models");
pass = 0; pass = 0;
do { do {
mptr = candmods = reveng(&model, qpoly, rflags, args, apolys); mptr = candmods = reveng(&model, qpoly, rflags, args, apolys);
if(mptr && plen(mptr->spoly)) if(mptr && plen(mptr->spoly))
uflags |= C_RESULT; uflags |= C_RESULT;
while(mptr && plen(mptr->spoly)) { while(mptr && plen(mptr->spoly)) {
/* results were printed by the callback
* string = mtostr(mptr);
* puts(string);
* free(string);
*/
mfree(mptr++); mfree(mptr++);
} }
free(candmods); free(candmods);
@ -238,29 +232,8 @@ int GetModels(char *Models[], int *count, uint32_t *width){
free(apolys); free(apolys);
if(~uflags & C_RESULT) if(~uflags & C_RESULT)
return uerr("no models found"); return uerr("no models found");
mfree(&model);
}
return 1;
}
//test call to GetModels
int CmdrevengTest(const char *Cmd){
char *Models[80];
int count = 0;
uint32_t width = 0;
width = param_get8(Cmd, 0);
//PrintAndLog("width: %d",width);
if (width > 89)
return uerr("Width cannot exceed 89");
int ans = GetModels(Models, &count, &width);
if (!ans) return 0;
PrintAndLog("Count: %d",count);
for (int i = 0; i < count; i++){
PrintAndLog("Model %d: %s",i,Models[i]);
free(Models[i]);
} }
return 1; return 1;
} }
@ -298,7 +271,7 @@ int RunModel(char *inModel, char *inHexStr, bool reverse, char endian, char *res
SETBMP(); SETBMP();
//set model //set model
if(!(c = mbynam(&model, inModel))) { if(!(c = mbynam(&model, inModel))) {
fprintf(stderr,"error: preset model '%s' not found. Use reveng -D to list presets.\n", inModel); PrintAndLog("error: preset model '%s' not found. Use reveng -D to list presets.", inModel);
return 0; return 0;
} }
if(c < 0) if(c < 0)

View file

@ -393,13 +393,15 @@ static int l_reveng_models(lua_State *L){
char *models[80]; char *models[80];
int count = 0; int count = 0;
lua_Integer in_width = luaL_checkinteger(L, 1); int in_width = luaL_checkinteger(L, 1);
if( in_width > 89 ) return returnToLuaWithError(L,"Width cannot exceed 89, got %d", in_width); if( in_width > 89 ) return returnToLuaWithError(L,"Width cannot exceed 89, got %d", in_width);
uint32_t width = (uint32_t)in_width; uint32_t width = (uint32_t)in_width;
int ans = GetModels(models, &count, &width); int ans = GetModels(models, &count, &width);
if (!ans) return 0; if (!ans){
return 0;
}
lua_newtable(L); lua_newtable(L);
@ -412,6 +414,13 @@ static int l_reveng_models(lua_State *L){
return 1; return 1;
} }
//Called with 4 parameters.
// inModel ,string containing the crc model name: 'CRC-8'
// inHexStr ,string containing the hex representation of the data that will be used for CRC calculations.
// reverse ,int 0/1 (bool) if 1, calculate the reverse CRC
// endian ,char, 'B','b','L','l','t','r' describing if Big-Endian or Little-Endian should be used in different combinations.
//
// outputs: string with hex representation of the CRC result
static int l_reveng_RunModel(lua_State *L){ static int l_reveng_RunModel(lua_State *L){
//-c || -v //-c || -v
//inModel = valid model name string - CRC-8 //inModel = valid model name string - CRC-8
@ -422,19 +431,14 @@ static int l_reveng_RunModel(lua_State *L){
//result = calculated crc hex string //result = calculated crc hex string
char result[50]; char result[50];
size_t dataLen; const char *inModel = luaL_checkstring(L, 1);
const char *inModel = luaL_checklstring(L, 1, &dataLen); const char *inHexStr = luaL_checkstring(L, 2);
if ( dataLen < 4 ) return returnToLuaWithError(L,"Can't find model, got %s", inModel); bool reverse = lua_toboolean(L, 3);
const char endian = luaL_checkstring(L, 4)[0];
const char *inHexStr = luaL_checklstring(L, 2, &dataLen);
if ( dataLen < 4 ) return returnToLuaWithError(L,"Hex string too short, got %d", dataLen);
int reverse = luaL_checkinteger(L, 3);
const char *endian = luaL_checklstring(L, 4, &dataLen);
//PrintAndLog("mod: %s, hex: %s, rev %d", inModel, inHexStr, reverse); //PrintAndLog("mod: %s, hex: %s, rev %d", inModel, inHexStr, reverse);
// int RunModel(char *inModel, char *inHexStr, bool reverse, char endian, char *result) // int RunModel(char *inModel, char *inHexStr, bool reverse, char endian, char *result)
int ans = RunModel( (char*)inModel, (char*)inHexStr, (bool)reverse, (char*)endian, result); int ans = RunModel( (char *)inModel, (char *)inHexStr, reverse, endian, result);
if (!ans) if (!ans)
return returnToLuaWithError(L,"Reveng failed"); return returnToLuaWithError(L,"Reveng failed");