hcq/oceanbase
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

[FEAT MERGE] rpc authentication improvement

Browse Source
This commit is contained in:
496148326@qq.com
2023-12-12 03:31:09 +00:00
committed by ob-robot
parent 8d9120d7fd
commit e03554cea8
44 changed files with 1594 additions and 77 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/sql/CMakeLists.txt
View File

@ -648,6 +648,7 @@ ob_set_subtarget(ob_sql engine_expr
engine/expr/ob_expr_xmlcast.cpp
engine/expr/ob_expr_update_xml.cpp
engine/expr/ob_expr_temp_table_ssid.cpp
engine/expr/ob_expr_extract_cert_expired_time.cpp
)
ob_set_subtarget(ob_sql engine_join

4
src/sql/engine/expr/ob_expr_eval_functions.cpp
View File

@ -335,6 +335,7 @@
#include "ob_expr_initcap.h"
#include "ob_expr_temp_table_ssid.h"
#include "ob_expr_align_date4cmp.h"
#include "ob_expr_extract_cert_expired_time.h"
namespace oceanbase
{
@ -1097,7 +1098,8 @@ static ObExpr::EvalFunc g_expr_eval_functions[] = {
eval_questionmark_decint2nmb, /* 658 */
eval_questionmark_nmb2decint_eqcast, /* 659 */
eval_questionmark_decint2decint_eqcast, /* 660 */
eval_questionmark_decint2decint_normalcast /* 661 */
eval_questionmark_decint2decint_normalcast, /* 661 */
ObExprExtractExpiredTime::eval_extract_cert_expired_time, /* 662 */
};
static ObExpr::EvalBatchFunc g_expr_eval_batch_functions[] = {

112
src/sql/engine/expr/ob_expr_extract_cert_expired_time.cpp Normal file
View File

@ -0,0 +1,112 @@
/**
* Copyright (c) 2021 OceanBase
* OceanBase CE is licensed under Mulan PubL v2.
* You can use this software according to the terms and conditions of the Mulan PubL v2.
* You may obtain a copy of Mulan PubL v2 at:
* http://license.coscl.org.cn/MulanPubL-2.0
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PubL v2 for more details.
*/
#define USING_LOG_PREFIX SQL_ENG
#include "ob_expr_extract_cert_expired_time.h"
#include "lib/utility/utility.h"
#include "sql/session/ob_sql_session_info.h"
#include "sql/engine/ob_exec_context.h"
using namespace oceanbase::share;
using namespace oceanbase::common;
namespace oceanbase {
namespace sql {
ObExprExtractExpiredTime::ObExprExtractExpiredTime(common::ObIAllocator& alloc)
: ObFuncExprOperator(alloc,
T_FUN_SYS_EXTRACT_CERT_EXPIRED_TIME,
N_EXTRACT_CERT_EXPIRED_TIME,
1,
NOT_VALID_FOR_GENERATED_COL,
NOT_ROW_DIMENSION)
{
}
ObExprExtractExpiredTime::~ObExprExtractExpiredTime() {}
int ObExprExtractExpiredTime::calc_result_type1(ObExprResType &type,
ObExprResType &text,
common::ObExprTypeCtx &type_ctx) const
{
UNUSED(type_ctx);
int ret = OB_SUCCESS;
if (text.is_null()) {
type.set_null();
} else {
if (!is_type_valid(text.get_type())) {
ret = OB_INVALID_ARGUMENT_NUM;
LOG_WARN("the param is not castable", K(text), K(ret));
} else {
type.set_timestamp();
type.set_scale(common::MAX_SCALE_FOR_TEMPORAL);
}
}
return ret;
}
int ObExprExtractExpiredTime::eval_extract_cert_expired_time(const ObExpr &expr, ObEvalCtx &ctx,
ObDatum &res)
{
int ret = OB_SUCCESS;
if (OB_UNLIKELY(expr.arg_cnt_ != 1)
|| OB_ISNULL(expr.args_)
|| OB_ISNULL(expr.args_[0])) {
ret = OB_INVALID_ARGUMENT;
LOG_WARN("invalid argument", K(ret));
} else {
ObDatum *arg = NULL;
int64_t expired_time = 0;
ObObjTypeClass in_tc = ob_obj_type_class(expr.args_[0]->datum_meta_.type_);
if (!ob_is_castable_type_class(in_tc)) {
res.set_null();
} else if (OB_FAIL(expr.eval_param_value(ctx, arg))) {
LOG_WARN("eval arg failed", K(ret));
} else if (arg->is_null()) {
res.set_null();
} else if (in_tc != ObTextTC) {
const ObString &arg_str = arg->get_string();
if (OB_FAIL(extract_cert_expired_time(arg_str.ptr(), arg_str.length(), expired_time))) {
LOG_WARN("failed to extract expired time", K(ret), K(*arg), K(in_tc));
} else {
res.set_timestamp(expired_time);
}
} else {
ObEvalCtx::TempAllocGuard alloc_guard(ctx);
ObIAllocator &tmp_alloc = alloc_guard.get_allocator();
ObString text_str;
if (OB_FAIL(ObTextStringHelper::get_string(expr, tmp_alloc,
0, arg, text_str))) {
LOG_WARN("failed to read realdata", K(ret));
} else if (OB_FAIL(extract_cert_expired_time(text_str.ptr(), text_str.length(), expired_time))) {
LOG_WARN("failed to extract expired time", K(ret), K(*arg), K(in_tc));
} else {
res.set_timestamp(expired_time);
}
}
}
return ret;
}
int ObExprExtractExpiredTime::cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const
{
int ret = OB_SUCCESS;
UNUSED(expr_cg_ctx);
UNUSED(raw_expr);
rt_expr.eval_func_ = eval_extract_cert_expired_time;
return ret;
}
}
}

37
src/sql/engine/expr/ob_expr_extract_cert_expired_time.h Normal file
View File

@ -0,0 +1,37 @@
/**
* Copyright (c) 2021 OceanBase
* OceanBase CE is licensed under Mulan PubL v2.
* You can use this software according to the terms and conditions of the Mulan PubL v2.
* You may obtain a copy of Mulan PubL v2 at:
* http://license.coscl.org.cn/MulanPubL-2.0
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PubL v2 for more details.
*/
#ifndef OCEANBASE_SQL_ENGINE_EXPR_OB_EXTRACT_CERT_EXPIRED_TIME_H
#define OCEANBASE_SQL_ENGINE_EXPR_OB_EXTRACT_CERT_EXPIRED_TIME_H
#include "sql/engine/expr/ob_expr_operator.h"
namespace oceanbase {
namespace sql {
class ObExprExtractExpiredTime : public ObFuncExprOperator
{
public:
ObExprExtractExpiredTime();
explicit ObExprExtractExpiredTime(common::ObIAllocator& alloc);
virtual ~ObExprExtractExpiredTime();
virtual int calc_result_type1(ObExprResType &type,
ObExprResType &text,
common::ObExprTypeCtx &type_ctx) const;
static int eval_extract_cert_expired_time(const ObExpr &expr, ObEvalCtx &ctx, ObDatum &res);
virtual int cg_expr(ObExprCGCtx &expr_cg_ctx, const ObRawExpr &raw_expr,
ObExpr &rt_expr) const override;
private:
DISALLOW_COPY_AND_ASSIGN(ObExprExtractExpiredTime);
};
}
}
#endif

2
src/sql/engine/expr/ob_expr_operator_factory.cpp
View File

@ -403,6 +403,7 @@
#include "sql/engine/expr/ob_expr_update_xml.h"
#include "sql/engine/expr/ob_expr_temp_table_ssid.h"
#include "sql/engine/expr/ob_expr_align_date4cmp.h"
#include "sql/engine/expr/ob_expr_extract_cert_expired_time.h"
using namespace oceanbase::common;
namespace oceanbase
@ -999,6 +1000,7 @@ void ObExprOperatorFactory::register_expr_operators()
REG_OP(ObExprRandstr);
REG_OP(ObExprPrefixPattern);
REG_OP(ObExprAlignDate4Cmp);
REG_OP(ObExprExtractExpiredTime);
}();
// 注册oracle系统函数
REG_OP_ORCL(ObExprSysConnectByPath);