问题描述
我正在本地使用以下库来连接到运行正常的远程服务器:
https://pypi.org/project/sshtunnel/
但是我需要将Python函数托管为Google Cloud Function。不幸的是,该库似乎只能接受文件,不能直接将键作为字符串接受。这是配置:
server = SSHTunnelForwarder(
SERVER_HOST,ssh_username=SSH_USERNAME,ssh_pkey="my_filename.pem",remote_bind_address=('127.0.0.1',5412)
)
如果我尝试插入以下内容:
SSH_KEY = """-----BEGIN RSA PRIVATE KEY-----
-----END RSA PRIVATE KEY-----"""
然后将ssh_pkey行修改为:
ssh_pkey=SSH_KEY
我希望它可以工作,但看起来好像该库不允许这样做。我在这里查看了源代码,看来是this引起了问题。
@staticmethod
def get_keys(logger=None,host_pkey_directories=None,allow_agent=False):
"""
Load public keys from any available SSH agent or local
.ssh directory.
Arguments:
logger (Optional[logging.Logger])
host_pkey_directories (Optional[list[str]]):
List of local directories where host SSH pkeys in the format
"id_*" are searched. For example,['~/.ssh']
.. versionadded:: 0.1.0
allow_agent (Optional[boolean]):
Whether or not load keys from agent
Default: False
Return:
list
"""
keys = SSHTunnelForwarder.get_agent_keys(logger=logger) \
if allow_agent else []
if host_pkey_directories is not None:
paramiko_key_types = {'rsa': paramiko.RSAKey,'dsa': paramiko.DSSKey,'ecdsa': paramiko.ECDSAKey,'ed25519': paramiko.Ed25519Key}
for directory in host_pkey_directories or [DEFAULT_SSH_DIRECTORY]:
for keytype in paramiko_key_types.keys():
ssh_pkey_expanded = os.path.expanduser(
os.path.join(directory,'id_{}'.format(keytype))
)
if os.path.isfile(ssh_pkey_expanded):
ssh_pkey = SSHTunnelForwarder.read_private_key_file(
pkey_file=ssh_pkey_expanded,logger=logger,key_type=paramiko_key_types[keytype]
)
if ssh_pkey:
keys.append(ssh_pkey)
if logger:
logger.info('{0} keys loaded from host directory'.format(
len(keys))
)
return keys
在此之前,我从来没有猴子打过补丁,我能以某种方式手动覆盖它吗?
解决方法
这就是我解决的方法,希望它能对某人有所帮助。首先,我使用Python打印出密钥文件'temp_key.pem'的base64编码密钥:
import base64
with open('temp_key.pem','rb') as f:
blob = base64.b64encode(f.read())
print(blob)
for_google_cloud_function = blob.decode('utf-8')
print(for_google_cloud_function)
我使用我的环境变量SSH_KEY_BLOB的输出。然后,在我的GCP Cloud功能中添加了此内容:
# decode key back into a useable form from base64
SSH_KEY_BLOB_DECODED = base64.b64decode(SSH_KEY_BLOB)
SSH_KEY = SSH_KEY_BLOB_DECODED.decode('utf-8')
# pass key to parmiko to get your pkey
pkey = paramiko.RSAKey.from_private_key(io.StringIO(SSH_KEY))
# setup your SSHTunnel like normal
server = SSHTunnelForwarder(
remote_server_ip,ssh_username=SSH_USERNAME,ssh_pkey=pkey,remote_bind_address=('127.0.0.1',27017)
)
这样,就不会对密钥进行硬编码,并且该功能可以从文件中自给自足。他们可能是更好的方法,但这对我有用。