INT32 _sptUsrHash::md5( const _sptArguments &arg,
_sptReturnVal &rval,
bson::BSONObj &detail )
{
INT32 rc = SDB_OK ;
stringstream ss ;
string code ;
string str ;
if ( 0 == arg.argc() )
{
rc = SDB_INVALIDARG ;
ss << "not specified the argument" ;
goto error ;
}
rc = arg.getString( 0, str ) ;
if ( rc )
{
PD_LOG ( PDERROR, "failed to get argument: %d", rc ) ;
ss << "argument \'str\' must be a string" ;
goto error ;
}
code = md5::md5simpledigest( str ) ;
rval.setStringVal( "", code.c_str() ) ;
done:
return rc ;
error:
detail = BSON( SPT_ERR << ss.str() ) ;
goto done ;
}
INT32 _sptUsrSsh::construct( const _sptArguments &arg,
_sptReturnVal &rval,
bson::BSONObj &detail)
{
INT32 rc = SDB_OK ;
string passwd ;
string errmsg ;
INT32 port = SPT_SSH_PORT ;
rc = arg.getString( 0, _host ) ;
if ( SDB_OUT_OF_BOUND == rc )
{
detail = BSON( SPT_ERR << "hostname must be config" ) ;
}
else if ( rc )
{
detail = BSON( SPT_ERR << "hostname must be string" ) ;
}
PD_RC_CHECK( rc, PDERROR, "Failed to get hostname, rc: %d", rc ) ;
rc = arg.getString( 1, _user ) ;
if ( rc && SDB_OUT_OF_BOUND != rc )
{
detail = BSON( SPT_ERR << "user must be string" ) ;
PD_RC_CHECK( rc, PDERROR, "Failed to get user, rc: %d", rc ) ;
}
rc = arg.getString( 2, passwd ) ;
if ( rc && SDB_OUT_OF_BOUND != rc )
{
detail = BSON( SPT_ERR << "password must be string" ) ;
PD_RC_CHECK( rc, PDERROR, "Failed to get password, rc: %d", rc ) ;
}
rc = arg.getNative( 3, (void*)&port, SPT_NATIVE_INT32 ) ;
if ( rc && SDB_OUT_OF_BOUND != rc )
{
detail = BSON( SPT_ERR << "port must be uint or int" ) ;
PD_RC_CHECK( rc, PDERROR, "Failed to get port, rc: %d", rc ) ;
}
_session = SDB_OSS_NEW _sptLibssh2Session( _host.c_str(),
_user.c_str(),
passwd.c_str(),
&port ) ;
if ( NULL == _session )
{
PD_LOG( PDERROR, "failed to allocate mem." ) ;
rc = SDB_OOM ;
goto error ;
}
rc = _session->open() ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to open ssh session:%d", rc ) ;
_session->getLastError( errmsg ) ;
goto error ;
}
_localIP = _session->getLocalIPAddr() ;
_peerIP = _session->getPeerIPAddr() ;
rval.addSelfProperty("_host")->setValue( _host ) ;
rval.addSelfProperty("_port")->setValue( port ) ;
rval.addSelfProperty("_usrname")->setValue( _user ) ;
done:
return rc ;
error:
SAFE_OSS_DELETE( _session ) ;
if ( detail.isEmpty() )
{
if ( !errmsg.empty() )
{
detail = BSON( SPT_ERR << errmsg ) ;
}
else
{
detail = BSON( SPT_ERR << "failed to ssh to specified host" ) ;
}
}
goto done ;
}
INT32 _sptUsrSsh::exec( const _sptArguments &arg,
_sptReturnVal &rval,
bson::BSONObj &detail )
{
INT32 rc = SDB_OK ;
SDB_ASSERT( NULL != _session, "can not be null" ) ;
string cmd ;
string errMsg ;
string sig ;
_lastRet = 0 ;
_lastOutStr = "" ;
rc = arg.getString( 0, cmd ) ;
if ( SDB_OUT_OF_BOUND == rc )
{
detail = BSON( SPT_ERR << "command must be config" ) ;
}
else if ( rc )
{
detail = BSON( SPT_ERR << "command must be string" ) ;
}
PD_RC_CHECK( rc, PDERROR, "Failed to get command, rc: %d", rc ) ;
if ( !_session->isOpened() )
{
detail = BSON( SPT_ERR << "connection is shutdown" ) ;
rc = SDB_NETWORK ;
goto error ;
}
rc = _session->exec( cmd.c_str(), _lastRet, _lastOutStr ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to exec cmd[%s], rc: %d", cmd.c_str(), rc ) ;
if ( _lastOutStr.empty() )
{
_session->getLastError( errMsg ) ;
}
else
{
errMsg = _lastOutStr ;
}
goto error ;
}
else if ( SDB_OK != _lastRet )
{
errMsg = _lastOutStr ;
rc = _lastRet ;
goto error ;
}
rval.getReturnVal().setValue( _lastOutStr ) ;
done:
return rc ;
error:
if ( !errMsg.empty() )
{
detail = BSON( SPT_ERR << errMsg ) ;
}
goto done ;
}
INT32 _sptUsrSsh::copyFromRemote( const _sptArguments &arg,
_sptReturnVal &rval,
bson::BSONObj &detail )
{
INT32 rc = SDB_OK ;
string remote ;
string local ;
INT32 mode = 0 ;
string errMsg ;
rc = arg.getString( 0, remote ) ;
if ( SDB_OUT_OF_BOUND == rc )
{
detail = BSON( SPT_ERR << "remote_file must be config" ) ;
}
else if ( rc )
{
detail = BSON( SPT_ERR << "remote_file must be string" ) ;
}
PD_RC_CHECK( rc, PDERROR, "Failed to get remote_file, rc: %d", rc ) ;
rc = arg.getString( 1, local ) ;
if ( SDB_OUT_OF_BOUND == rc )
{
detail = BSON( SPT_ERR << "local_file must be config" ) ;
}
else if ( rc )
{
detail = BSON( SPT_ERR << "local_file must be string" ) ;
}
PD_RC_CHECK( rc, PDERROR, "Failed to get local_file, rc: %d", rc ) ;
rc = arg.getNative( 2, &mode, SPT_NATIVE_INT32 ) ;
if ( rc && SDB_OUT_OF_BOUND != rc )
{
detail = BSON( SPT_ERR << "mode must be native type" ) ;
PD_RC_CHECK( rc, PDERROR, "Failed to get mode, rc: %d", rc ) ;
}
if ( !_session->isOpened() )
{
detail = BSON( SPT_ERR << "connection is shutdown" ) ;
rc = SDB_NETWORK ;
goto error ;
}
rc = _session->copyFromRemote( SPT_CP_PROTOCOL_SCP,
remote.c_str(),
local.c_str(),
mode ) ;
if ( SDB_OK != rc )
{
PD_LOG( PDERROR, "failed to copy file:%d", rc ) ;
_session->getLastError( errMsg ) ;
goto error ;
}
done:
return rc ;
error:
if ( !errMsg.empty() )
{
detail = BSON( SPT_ERR << errMsg ) ;
}
goto done ;
}
INT32 _sptUsrHash::fileMD5( const _sptArguments &arg,
_sptReturnVal &rval,
bson::BSONObj &detail )
{
INT32 rc = SDB_OK ;
SINT64 bufSize = SPT_MD5_READ_LEN ;
SINT64 hasRead = 0 ;
CHAR readBuf[SPT_MD5_READ_LEN + 1] = { 0 } ;
OSSFILE file ;
string filename ;
stringstream ss ;
BOOLEAN isOpen = FALSE ;
md5_state_t st ;
md5_init( &st ) ;
md5::md5digest digest ;
string code ;
if ( 0 == arg.argc() )
{
rc = SDB_INVALIDARG ;
ss << "not specified the file" ;
goto error ;
}
rc = arg.getString( 0, filename ) ;
if ( rc )
{
PD_LOG ( PDERROR, "failed to get filename argument: %d", rc ) ;
ss << "argument \'filename\' must be a string" ;
goto error ;
}
rc = ossOpen( filename.c_str(), OSS_READONLY, 0, file ) ;
if ( rc )
{
ss << "open file[" << filename.c_str() << "] failed: " << rc ;
goto error ;
}
isOpen = TRUE ;
while ( TRUE )
{
rc = ossReadN( &file, bufSize, readBuf, hasRead ) ;
if ( SDB_EOF == rc || 0 == hasRead )
{
rc = SDB_OK ;
break ;
}
else if ( rc )
{
ss << "Read file[" << filename.c_str() << "] failed, rc: " << rc ;
goto error ;
}
md5_append( &st, (const md5_byte_t *)readBuf, hasRead ) ;
}
md5_finish( &st, digest ) ;
code = md5::digestToString( digest ) ;
rval.setStringVal( "", code.c_str() ) ;
done:
if ( TRUE == isOpen )
ossClose( file ) ;
return rc ;
error:
detail = BSON( SPT_ERR << ss.str() ) ;
goto done ;
}
INT32 _sptUsrCmd::start( const _sptArguments & arg,
_sptReturnVal & rval,
BSONObj & detail )
{
INT32 rc = SDB_OK ;
string ev ;
ossCmdRunner runner ;
UINT32 useShell = TRUE ;
UINT32 usePipe = TRUE ;
_command.clear() ;
rc = arg.getString( 0, _command ) ;
if ( SDB_OK != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "cmd must be config" ) ;
goto error ;
}
utilStrTrim( _command ) ;
rc = arg.getString( 1, ev ) ;
if ( SDB_OK != rc && SDB_OUT_OF_BOUND != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "environment should be a string" ) ;
goto error ;
}
else if ( SDB_OK == rc )
{
_command += " " ;
_command += ev ;
}
// useShell, default : 1
rc = arg.getNative( 2, (void*)&useShell, SPT_NATIVE_INT32 ) ;
if ( SDB_OK != rc && SDB_OUT_OF_BOUND != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "useShell should be a number" ) ;
goto error ;
}
rc = SDB_OK ;
// usePipe, default : 1
rc = arg.getNative( 3, (void*)&usePipe, SPT_NATIVE_INT32 ) ;
if ( SDB_OK != rc && SDB_OUT_OF_BOUND != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "usePipe should be a number" ) ;
goto error ;
}
rc = SDB_OK ;
_strOut = "" ;
_retCode = 0 ;
rc = runner.exec( _command.c_str(), _retCode, TRUE, -1, FALSE, NULL,
useShell ? TRUE : FALSE, usePipe ? TRUE : FALSE ) ;
if ( SDB_OK != rc )
{
stringstream ss ;
ss << "run[" << _command << "] failed" ;
detail = BSON( SPT_ERR << ss.str() ) ;
goto error ;
}
else
{
OSSPID pid = runner.getPID() ;
rval.setNativeVal( "", NumberInt, (const void*)&pid ) ;
if ( usePipe )
{
ossSleep( 100 ) ;
if ( !ossIsProcessRunning( pid ) )
{
rc = runner.read( _strOut ) ;
if ( rc )
{
stringstream ss ;
ss << "read run command[" << _command << "] result failed" ;
detail = BSON( SPT_ERR << ss.str() ) ;
goto error ;
}
}
}
}
done:
return rc ;
error:
goto done ;
}
INT32 _sptUsrCmd::exec( const _sptArguments &arg,
_sptReturnVal &rval,
bson::BSONObj &detail )
{
INT32 rc = SDB_OK ;
string ev ;
UINT32 timeout = 0 ;
UINT32 useShell = TRUE ;
ossCmdRunner runner ;
_command.clear() ;
rc = arg.getString( 0, _command ) ;
if ( SDB_OK != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "cmd must be config" ) ;
goto error ;
}
utilStrTrim( _command ) ;
rc = arg.getString( 1, ev ) ;
if ( SDB_OK != rc && SDB_OUT_OF_BOUND != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "environment should be a string" ) ;
goto error ;
}
else if ( SDB_OK == rc && !ev.empty() )
{
_command += " " ;
_command += ev ;
}
rc = arg.getNative( 2, (void*)&timeout, SPT_NATIVE_INT32 ) ;
if ( SDB_OK != rc && SDB_OUT_OF_BOUND != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "timeout should be a number" ) ;
goto error ;
}
rc = SDB_OK ;
// useShell, default : 1
rc = arg.getNative( 3, (void*)&useShell, SPT_NATIVE_INT32 ) ;
if ( SDB_OK != rc && SDB_OUT_OF_BOUND != rc )
{
rc = SDB_INVALIDARG ;
detail = BSON( SPT_ERR << "useShell should be a number" ) ;
goto error ;
}
rc = SDB_OK ;
_strOut = "" ;
_retCode = 0 ;
rc = runner.exec( _command.c_str(), _retCode, FALSE,
0 == timeout ? -1 : (INT64)timeout,
FALSE, NULL, useShell ? TRUE : FALSE ) ;
if ( SDB_OK != rc )
{
stringstream ss ;
ss << "run[" << _command << "] failed" ;
detail = BSON( SPT_ERR << ss.str() ) ;
goto error ;
}
else
{
rc = runner.read( _strOut ) ;
if ( rc )
{
stringstream ss ;
ss << "read run command[" << _command << "] result failed" ;
detail = BSON( SPT_ERR << ss.str() ) ;
goto error ;
}
else if ( SDB_OK != _retCode )
{
detail = BSON( SPT_ERR << _strOut ) ;
rc = _retCode ;
goto error ;
}
rval.setStringVal( "", _strOut.c_str() ) ;
}
done:
return rc ;
error:
goto done ;
}