Пример #1
0
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 ;
}
Пример #2
0
   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 ;
   }
Пример #3
0
   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 ;
   }
Пример #4
0
   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 ;
   }
Пример #5
0
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 ;
}
Пример #6
0
   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 ;
   }
Пример #7
0
   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 ;
   }