zjgcjy/pykd
1
0
Fork 0
mirror of https://github.com/ivellioscolin/pykd.git synced 2025-04-20 03:23:23 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

[0.3.x] fixed: run script without .py extension

Browse Source 
git-svn-id: https://pykd.svn.codeplex.com/svn@89655 9b283d60-5439-405e-af05-b73fd8c4d996
This commit is contained in:
SND\EreTIk_cp 2015-01-20 09:35:31 +00:00 committed by Mikhail I. Izmestev
parent dd2138393c
commit d7672c3ffe
2 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
pykd/windbgext.cpp
View File

@ -117,7 +117,7 @@ KDLIB_EXT_COMMAND_METHOD_IMPL(PykdExt, py)
std::string scriptFileName;
if ( args.size() > 0 )
{
scriptFileName = getScriptFileName( args[0] );
scriptFileName = getScriptFileName( args[0] );
if ( scriptFileName.empty() )
{
@ -247,16 +247,18 @@ void PykdExt::printUsage()
std::string PykdExt::getScriptFileName( const std::string &scriptName )
{
bool fileHasPyExt = false;
std::string scriptFileName = findScript( scriptName );
if ( scriptName.length() > 3 )
fileHasPyExt = scriptName.rfind(".py") == scriptName.length() - 3;
std::string fullFileName = scriptName;
if ( scriptFileName.empty() && (scriptName.rfind(".py") != scriptName.length() - 3) )
scriptFileName = findScript( scriptName + ".py" );
if (!fileHasPyExt)
fullFileName += ".py";
return scriptFileName;
}
///////////////////////////////////////////////////////////////////////////////
std::string PykdExt::findScript( const std::string &fullFileName )
{
if ( GetFileAttributesA(fullFileName.c_str()) != INVALID_FILE_ATTRIBUTES )
return fullFileName;
@ -287,10 +289,9 @@ std::string PykdExt::getScriptFileName( const std::string &scriptName )
if (bufSize > 0)
{
fullFileName = std::string(&fullFileNameCStr[0]);
return fullFileName;
return std::string(&fullFileNameCStr[0]);
}
}
}
}
return "";

1
pykd/windbgext.h
View File

@ -25,6 +25,7 @@ private:
virtual void tearDown();
std::string getScriptFileName( const std::string &scriptName );
std::string findScript( const std::string &fullFileName );
std::vector<std::string> m_paths;