QuasarApp/pe-parse
4
0
Fork 0
mirror of https://github.com/QuasarApp/pe-parse.git synced 2025-04-27 04:44:31 +00:00
Code Issues Projects Releases Wiki Activity

Merge pull request #22 from gsauthof/speedup-cstring-copy

Browse Source 
LGTM
This commit is contained in:
Peter Goodman 2016-09-08 12:18:36 -04:00 committed by GitHub
commit 6d1b49b74a
1 changed files with 25 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
parser-library/parse.cpp
View File

@ -24,6 +24,9 @@ THE SOFTWARE.
#include <string.h>
#include <list>
#include <algorithm>
#include <stdexcept>
#include <boost/algorithm/string/case_conv.hpp>
#include "parse.h"
#include "nt-headers.h"
#include "to_string.h"
@ -91,6 +94,23 @@ string GetPEErrLoc() {
return err_loc;
}
static bool readCString(const bounded_buffer &buffer, ::uint32_t off,
string &result)
{
if (off < buffer.bufLen) {
::uint8_t *p = buffer.buf;
::uint32_t n = buffer.bufLen;
::uint8_t *b = p + off;
::uint8_t *x = std::find(b, p + n, 0);
if (x == p + n)
return false;
result.insert(result.end(), b, x);
return true;
} else {
return false;
}
}
bool getSecForVA(list<section> &secs, VA v, section &sec) {
for(list<section>::iterator it = secs.begin(), e = secs.end();
it != e;
@ -723,23 +743,13 @@ parsed_pe *ParsePEFromFile(const char *filePath) {
::uint32_t nameOff = nameVA - nameSec.sectionBase;
string modName;
::uint8_t c;
do {
if(readByte(nameSec.sectionData, nameOff, c) == false) {
if (readCString(*nameSec.sectionData, nameOff, modName) == false) {
deleteBuffer(remaining);
deleteBuffer(p->fileBuffer);
delete p;
PE_ERR(PEERR_READ);
return NULL;
}
if(c == 0) {
break;
}
modName.push_back(c);
nameOff++;
}while(true);
}
//now, get all the named export symbols
::uint32_t numNames;
@ -1185,23 +1195,14 @@ parsed_pe *ParsePEFromFile(const char *filePath) {
::uint32_t nameOff = name - nameSec.sectionBase;
string modName;
::uint8_t c;
do {
if(readByte(nameSec.sectionData, nameOff, c) == false) {
if (readCString(*nameSec.sectionData, nameOff, modName) == false ) {
deleteBuffer(remaining);
deleteBuffer(p->fileBuffer);
delete p;
PE_ERR(PEERR_READ);
return NULL;
}
if(c == 0) {
break;
}
modName.push_back(toupper(c));
nameOff++;
}while(true);
}
boost::to_upper(modName);
//then, try and get all of the sub-symbols
VA lookupVA;