mirror of
https://github.com/ivellioscolin/pykd.git
synced 2025-04-21 12:53:23 +08:00
151 lines
4.2 KiB
C++
151 lines
4.2 KiB
C++
#include "stdafx.h"
|
||
|
||
#include <vector>
|
||
|
||
#include "dbgpath.h"
|
||
|
||
#include <boost/tokenizer.hpp>
|
||
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
|
||
|
||
|
||
DbgPythonPath::DbgPythonPath()
|
||
{
|
||
DWORD enviromentSize = 0;
|
||
|
||
enviromentSize = GetEnvironmentVariableA( "PYTHONPATH", NULL, enviromentSize );
|
||
|
||
std::vector<char> enviromentBuffer(enviromentSize);
|
||
|
||
if (!enviromentBuffer.empty())
|
||
{
|
||
GetEnvironmentVariableA( "PYTHONPATH", &enviromentBuffer[0], enviromentSize );
|
||
|
||
typedef boost::escaped_list_separator<char> char_separator_t;
|
||
typedef boost::tokenizer< char_separator_t > char_tokenizer_t;
|
||
|
||
std::string pytonPath( &enviromentBuffer[0] );
|
||
|
||
char_tokenizer_t token( pytonPath, char_separator_t( "", "; \t", "\"" ) );
|
||
|
||
for ( char_tokenizer_t::iterator it = token.begin(); it != token.end(); ++it )
|
||
{
|
||
if ( *it != "" )
|
||
m_pathList.push_back( *it );
|
||
}
|
||
}
|
||
}
|
||
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
|
||
std::string
|
||
DbgPythonPath::getStr() const
|
||
{
|
||
std::string str;
|
||
std::vector<std::string>::const_iterator it = m_pathList.begin();
|
||
|
||
for ( ; it != m_pathList.end(); ++it )
|
||
{
|
||
str += *it;
|
||
str += ";";
|
||
}
|
||
|
||
return str;
|
||
}
|
||
|
||
///////////////////////////////////////////////////////////////////////////////
|
||
|
||
bool
|
||
DbgPythonPath::findPath(
|
||
const std::string &fileName,
|
||
std::string &fullFileName,
|
||
std::string &filePath ) const
|
||
{
|
||
std::vector< std::string > extPathList;
|
||
|
||
boost::python::object sys = boost::python::import( "sys");
|
||
|
||
boost::python::list pathList( sys.attr("path") );
|
||
|
||
boost::python::ssize_t n = boost::python::len(pathList);
|
||
for(boost::python::ssize_t i=0;i<n;i++)
|
||
extPathList.push_back( boost::python::extract<std::string>( pathList[i] ) );
|
||
|
||
bool pyExt = fileName.rfind( ".py" ) == fileName.length() - 3;
|
||
|
||
// 1. <20><><EFBFBD><EFBFBD> <20> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>
|
||
DWORD bufSize =
|
||
SearchPathA(
|
||
NULL,
|
||
fileName.c_str(),
|
||
pyExt ? NULL : ".py",
|
||
0,
|
||
NULL,
|
||
NULL );
|
||
|
||
if ( bufSize > 0 )
|
||
{
|
||
bufSize += 1;
|
||
std::vector<char> fullFileNameCStr(bufSize);
|
||
char *partFileNameCStr = NULL;
|
||
|
||
SearchPathA(
|
||
NULL,
|
||
fileName.c_str(),
|
||
pyExt ? NULL : ".py",
|
||
bufSize,
|
||
&fullFileNameCStr[0],
|
||
&partFileNameCStr );
|
||
|
||
fullFileName = std::string( &fullFileNameCStr[0] );
|
||
if ( !fullFileName.empty() )
|
||
{
|
||
filePath = std::string( &fullFileNameCStr[0], partFileNameCStr );
|
||
return true;
|
||
}
|
||
}
|
||
|
||
// 2. <20><><EFBFBD><EFBFBD> <20><> <20><><EFBFBD><EFBFBD> <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>, <20><><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD> <20> m_pathList
|
||
|
||
std::vector<std::string>::const_iterator it = extPathList.begin();
|
||
|
||
for ( ; it != extPathList.end(); ++it )
|
||
{
|
||
DWORD bufSize =
|
||
SearchPathA(
|
||
(*it).c_str(),
|
||
fileName.c_str(),
|
||
pyExt ? NULL : ".py",
|
||
0,
|
||
NULL,
|
||
NULL );
|
||
|
||
if ( bufSize > 0 )
|
||
{
|
||
bufSize += 1;
|
||
std::vector<char> fullFileNameCStr(bufSize);
|
||
char *partFileNameCStr = NULL;
|
||
|
||
bufSize = SearchPathA(
|
||
(*it).c_str(),
|
||
fileName.c_str(),
|
||
pyExt ? NULL : ".py",
|
||
bufSize,
|
||
&fullFileNameCStr[0],
|
||
&partFileNameCStr );
|
||
|
||
fullFileName = std::string( &fullFileNameCStr[0] );
|
||
if ( !fullFileName.empty() )
|
||
{
|
||
filePath = std::string( &fullFileNameCStr[0], partFileNameCStr );
|
||
return true;
|
||
}
|
||
}
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
//////////////////////////////////////////////////////////////////////////////
|