1. use jsoncpp c++11 interface in telnet module. 2. fix timeout when create ssh channel on libssh session.

pull/236/head
Apex Liu 2019-11-22 02:59:52 +08:00
parent e91f057692
commit 958712745b
5 changed files with 1290 additions and 80 deletions

View File

@ -231,15 +231,12 @@ class BuilderWin(BuilderBase):
utils.unzip(os.path.join(PATH_DOWNLOAD, file_name), PATH_EXTERNAL) utils.unzip(os.path.join(PATH_DOWNLOAD, file_name), PATH_EXTERNAL)
os.rename(os.path.join(PATH_EXTERNAL, 'libssh-{}'.format(env.ver_libssh)), self.LIBSSH_PATH_SRC) os.rename(os.path.join(PATH_EXTERNAL, 'libssh-{}'.format(env.ver_libssh)), self.LIBSSH_PATH_SRC)
# cc.n('fix libssh source code... ', end='')
# utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', 'src', 'sftp.c'))
# utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'sftp.c')
cc.n('fix libssh source code... ', end='') cc.n('fix libssh source code... ', end='')
s_name = 'libssh-{}'.format(env.ver_libssh) s_name = 'libssh-{}'.format(env.ver_libssh)
# utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'session.c')) utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'session.c'))
# utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto.c')) # utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto.c'))
utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto-compat.c')) utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto-compat.c'))
# utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'session.c') utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'session.c')
# utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto.c') # utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto.c')
utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto-compat.c') utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto-compat.c')
@ -521,10 +518,10 @@ class BuilderLinux(BuilderBase):
cc.n('fix libssh source code... ', end='') cc.n('fix libssh source code... ', end='')
s_name = 'libssh-{}'.format(env.ver_libssh) s_name = 'libssh-{}'.format(env.ver_libssh)
# utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'session.c')) utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'session.c'))
# utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto.c')) # utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto.c'))
utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto-compat.c')) utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto-compat.c'))
# utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'session.c') utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'session.c')
# utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto.c') # utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto.c')
utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto-compat.c') utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto-compat.c')
@ -578,15 +575,6 @@ class BuilderLinux(BuilderBase):
return return
cc.v('') cc.v('')
# cc.n('fix libssh source code... ', end='')
# s_name = 'libssh-{}'.format(env.ver_libssh)
# # utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'session.c'))
# # utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto.c'))
# utils.ensure_file_exists(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src', 'libcrypto-compat.c'))
# # utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'session.c')
# # utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto.c')
# utils.copy_file(os.path.join(PATH_EXTERNAL, 'fix-external', 'libssh', s_name, 'src'), os.path.join(self.LIBSSH_PATH_SRC, 'src'), 'libcrypto-compat.c')
build_path = os.path.join(self.ZLIB_PATH_SRC, 'build') build_path = os.path.join(self.ZLIB_PATH_SRC, 'build')
cmake_define = ' -DCMAKE_INSTALL_PREFIX={path_release}' \ cmake_define = ' -DCMAKE_INSTALL_PREFIX={path_release}' \

Binary file not shown.

View File

@ -322,7 +322,7 @@ int HMAC_CTX_reset(HMAC_CTX *ctx)
return 1; return 1;
} }
#if 0 // by Apex #if 0 // by apex.liu
#ifndef HAVE_OPENSSL_EVP_CIPHER_CTX_NEW #ifndef HAVE_OPENSSL_EVP_CIPHER_CTX_NEW
EVP_CIPHER_CTX *EVP_CIPHER_CTX_new(void) EVP_CIPHER_CTX *EVP_CIPHER_CTX_new(void)
{ {
@ -336,7 +336,7 @@ void EVP_CIPHER_CTX_free(EVP_CIPHER_CTX *ctx)
OPENSSL_free(ctx); OPENSSL_free(ctx);
} }
#endif #endif
#endif #endif // endif by apex.liu
void DH_get0_pqg(const DH *dh, void DH_get0_pqg(const DH *dh,
const BIGNUM **p, const BIGNUM **q, const BIGNUM **g) const BIGNUM **p, const BIGNUM **q, const BIGNUM **g)

File diff suppressed because it is too large Load Diff

View File

@ -1,60 +1,66 @@
#include "telnet_proxy.h" #include "telnet_proxy.h"
#include "tpp_env.h" #include "tpp_env.h"
#include <teleport_const.h> #include <teleport_const.h>
#include <json/json.h> #include <json/json.h>
TPP_API ex_rv tpp_init(TPP_INIT_ARGS* init_args) TPP_API ex_rv tpp_init(TPP_INIT_ARGS* init_args)
{ {
if (!g_telnet_env.init(init_args)) if (!g_telnet_env.init(init_args))
return TPE_FAILED; return TPE_FAILED;
return 0; return 0;
} }
TPP_API ex_rv tpp_start(void) TPP_API ex_rv tpp_start(void)
{ {
if (!g_telnet_proxy.init()) if (!g_telnet_proxy.init())
return TPE_FAILED; return TPE_FAILED;
if (!g_telnet_proxy.start()) if (!g_telnet_proxy.start())
return TPE_FAILED; return TPE_FAILED;
return 0; return 0;
} }
TPP_API ex_rv tpp_stop(void) TPP_API ex_rv tpp_stop(void)
{ {
g_telnet_proxy.stop(); g_telnet_proxy.stop();
return 0; return 0;
} }
TPP_API void tpp_timer(void) { TPP_API void tpp_timer(void) {
// be called per one second. // be called per one second.
g_telnet_proxy.timer(); g_telnet_proxy.timer();
} }
// TPP_API void tpp_set_cfg(TPP_SET_CFG_ARGS* cfg_args) {
// g_telnet_proxy.set_cfg(cfg_args);
// }
// TPP_API void tpp_set_cfg(TPP_SET_CFG_ARGS* cfg_args) {
// g_telnet_proxy.set_cfg(cfg_args);
// }
static ex_rv _set_runtime_config(const char* param) { static ex_rv _set_runtime_config(const char* param) {
Json::Value jp; // Json::Value jp;
Json::Reader jreader; // Json::Reader jreader;
//
// if (!jreader.parse(param, jp))
Json::CharReaderBuilder jcrb;
std::unique_ptr<Json::CharReader> const jreader(jcrb.newCharReader());
if (!jreader.parse(param, jp)) ex_astr err;
return TPE_JSON_FORMAT; Json::Value jp;
if (!jreader->parse(param, param + strlen(param), &jp, &err))
return TPE_JSON_FORMAT;
if (!jp.isObject()) if (!jp.isObject())
return TPE_PARAM; return TPE_PARAM;
if (jp["noop_timeout"].isNull() || !jp["noop_timeout"].isUInt()) if (jp["noop_timeout"].isNull() || !jp["noop_timeout"].isUInt())
return TPE_PARAM; return TPE_PARAM;
ex_u32 noop_timeout = jp["noop_timeout"].asUInt(); ex_u32 noop_timeout = jp["noop_timeout"].asUInt();
if (noop_timeout == 0) if (noop_timeout == 0)
return TPE_PARAM; return TPE_PARAM;
g_telnet_proxy.set_cfg(noop_timeout * 60); g_telnet_proxy.set_cfg(noop_timeout * 60);
@ -62,25 +68,31 @@ static ex_rv _set_runtime_config(const char* param) {
} }
static ex_rv _kill_sessions(const char* param) { static ex_rv _kill_sessions(const char* param) {
Json::Value jp; // Json::Value jp;
Json::Reader jreader; // Json::Reader jreader;
//
// if (!jreader.parse(param, jp))
Json::CharReaderBuilder jcrb;
std::unique_ptr<Json::CharReader> const jreader(jcrb.newCharReader());
if (!jreader.parse(param, jp)) ex_astr err;
return TPE_JSON_FORMAT; Json::Value jp;
if (!jreader->parse(param, param + strlen(param), &jp, &err))
return TPE_JSON_FORMAT;
if (!jp.isArray()) if (!jp.isArray())
return TPE_PARAM; return TPE_PARAM;
ex_astrs ss; ex_astrs ss;
int cnt = jp.size(); int cnt = jp.size();
for (int i = 0; i < cnt; ++i) for (int i = 0; i < cnt; ++i)
{ {
if (!jp[i].isString()) { if (!jp[i].isString()) {
return TPE_PARAM; return TPE_PARAM;
} }
ss.push_back(jp[i].asString()); ss.push_back(jp[i].asString());
} }
g_telnet_proxy.kill_sessions(ss); g_telnet_proxy.kill_sessions(ss);