Merge pull request #1809 from DidierA/fix_cmd_data

Fix read oob in data diff
This commit is contained in:
Iceman 2022-11-08 07:44:06 +01:00 committed by GitHub
commit fac513b9bd
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -3063,8 +3063,8 @@ static int CmdDiff(const char *Cmd) {
} }
*/ */
size_t n = (datalenA > datalenB) ? datalenB : datalenA; size_t biggest = (datalenA > datalenB) ? datalenA : datalenB;
PrintAndLogEx(DEBUG, "data len: %zu A %zu B %zu", n, datalenA, datalenB); PrintAndLogEx(DEBUG, "data len: %zu A %zu B %zu", biggest, datalenA, datalenB);
if (inA == NULL) if (inA == NULL)
PrintAndLogEx(INFO, "inA null"); PrintAndLogEx(INFO, "inA null");
@ -3091,102 +3091,68 @@ static int CmdDiff(const char *Cmd) {
char line[880] = {0}; char line[880] = {0};
// print data diff loop // print data diff loop
int i; for (int i = 0 ; i < biggest ; i += width) {
for (i = 0; i < n; i += width) { char dlnA[240] = {0};
char dlnB[240] = {0};
char dlnAii[180] = {0};
char dlnBii[180] = {0};
memset(line, 0, sizeof(line)); memset(dlnA, 0, sizeof(dlnA));
memset(dlnB, 0, sizeof(dlnB));
memset(dlnAii, 0, sizeof(dlnAii));
memset(dlnBii, 0, sizeof(dlnBii));
int diff = memcmp(inA + i, inB + i, width); for (int j = i; j < i + width; j++) {
int dlnALen = strlen(dlnA);
int dlnBLen = strlen(dlnB);
int dlnAiiLen = strlen(dlnAii);
int dlnBiiLen = strlen(dlnBii);
// if ok, just print //both files ended
if (diff == 0) { if (j >= datalenA && j >= datalenB) {
hex_to_buffer((uint8_t *)line, inA + i, width, sizeof(line), 0, 1, true); snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, "-- ");
ascii_to_buffer((uint8_t *)(line + strlen(line)), inA + i, width, width, 0); snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, ".") ;
strncat(line + strlen(line), " | ", sizeof(line) - strlen(line)); snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, "-- ");
hex_to_buffer((uint8_t *)(line + strlen(line)), inB + i, width, sizeof(line), 0, 1, true); snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, ".") ;
ascii_to_buffer((uint8_t *)(line + strlen(line)), inB + i, width, width, 0); continue ;
} else { }
char dlnA[240] = {0}; char ca, cb;
char dlnB[240] = {0};
char dlnAii[180] = {0}; if (j >= datalenA) {
char dlnBii[180] = {0}; // file A ended. print B without colors
cb = inB[j];
memset(dlnA, 0, sizeof(dlnA)); snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, "-- ");
memset(dlnB, 0, sizeof(dlnB)); snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, ".") ;
memset(dlnAii, 0, sizeof(dlnAii)); snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, "%02X ", inB[j]);
memset(dlnBii, 0, sizeof(dlnBii)); snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, "%c", ((cb < 32) || (cb == 127)) ? '.' : cb);
continue ;
// if diff, time to find it }
for (int j = i; j < (i + width); j++) { ca = inA[j];
if (j >= datalenB) {
char ca = inA[j]; // file B ended. print A without colors
char cb = inB[j]; snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, "%02X ", inA[j]);
snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, "%c", ((ca < 32) || (ca == 127)) ? '.' : ca);
int dlnALen = strlen(dlnA); snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, "-- ");
int dlnBLen = strlen(dlnB); snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, ".") ;
int dlnAiiLen = strlen(dlnAii); continue ;
int dlnBiiLen = strlen(dlnBii); }
cb = inB[j];
if (inA[j] != inB[j]) { if (inA[j] != inB[j]) {
// diff / add colors
// diff / add colors snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, _GREEN_("%02X "), inA[j]);
snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, _GREEN_("%02X "), inA[j]); snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, _RED_("%02X "), inB[j]);
snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, _RED_("%02X "), inB[j]); snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, _GREEN_("%c"), ((ca < 32) || (ca == 127)) ? '.' : ca);
snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, _GREEN_("%c"), ((ca < 32) || (ca == 127)) ? '.' : ca); snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, _RED_("%c"), ((cb < 32) || (cb == 127)) ? '.' : cb);
snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, _RED_("%c"), ((cb < 32) || (cb == 127)) ? '.' : cb); } else {
// normal
} else { snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, "%02X ", inA[j]);
// normal snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, "%02X ", inB[j]);
snprintf(dlnA + dlnALen, sizeof(dlnA) - dlnALen, "%02X ", inA[j]); snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, "%c", ((ca < 32) || (ca == 127)) ? '.' : ca);
snprintf(dlnB + dlnBLen, sizeof(dlnB) - dlnBLen, "%02X ", inB[j]); snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, "%c", ((cb < 32) || (cb == 127)) ? '.' : cb);
snprintf(dlnAii + dlnAiiLen, sizeof(dlnAii) - dlnAiiLen, "%c", ((ca < 32) || (ca == 127)) ? '.' : ca);
snprintf(dlnBii + dlnBiiLen, sizeof(dlnBii) - dlnBiiLen, "%c", ((cb < 32) || (cb == 127)) ? '.' : cb);
}
} }
snprintf(line, sizeof(line), "%s%s | %s%s", dlnA, dlnAii, dlnB, dlnBii);
}
PrintAndLogEx(INFO, "%03X | %s", i, line);
}
// mod
// print different length
bool tallestA = (datalenA > datalenB);
if (tallestA) {
n = datalenA;
} else {
n = datalenB;
}
// print data diff loop
for (; i < n; i += width) {
memset(line, 0, sizeof(line));
if (tallestA) {
hex_to_buffer((uint8_t *)line, inA + i, width, sizeof(line), 0, 1, true);
ascii_to_buffer((uint8_t *)(line + strlen(line)), inA + i, width, sizeof(line), 0);
strcat(line + strlen(line), " | ");
for (int j = 0; j < width; j++) {
strncat(line + strlen(line), "-- ", sizeof(line) - strlen(line));
}
for (int j = 0; j < width; j++) {
strncat(line + strlen(line), ".", sizeof(line) - strlen(line));
}
} else {
for (int j = 0; j < width; j++) {
strncat(line + strlen(line), "-- ", sizeof(line) - strlen(line));
}
for (int j = 0; j < width; j++) {
strncat(line + strlen(line), ".", sizeof(line) - strlen(line));
}
strncat(line + strlen(line), " | ", sizeof(line) - strlen(line));
hex_to_buffer((uint8_t *)(line + strlen(line)), inB + i, width, sizeof(line), 0, 1, true);
ascii_to_buffer((uint8_t *)(line + strlen(line)), inB + i, width, sizeof(line), 0);
} }
snprintf(line, sizeof(line), "%s%s | %s%s", dlnA, dlnAii, dlnB, dlnBii);
PrintAndLogEx(INFO, "%03X | %s", i, line); PrintAndLogEx(INFO, "%03X | %s", i, line);
} }