Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
openeuler
A-Tune
提交
391fb9dc
A
A-Tune
项目概览
openeuler
/
A-Tune
通知
5
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
A
A-Tune
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
391fb9dc
编写于
1月 07, 2020
作者:
O
openeuler-ci-bot
提交者:
Gitee
1月 07, 2020
浏览文件
操作
浏览文件
下载
差异文件
!8 atuned: add input validation for collection
Merge pull request !8 from 谢志鹏/master
上级
fa881521
0104bad1
变更
3
显示空白变更内容
内联
并排
Showing
3 changed file
with
34 addition
and
0 deletion
+34
-0
.gitignore
.gitignore
+3
-0
common/utils/utils.go
common/utils/utils.go
+11
-0
modules/server/profile/profile.go
modules/server/profile/profile.go
+20
-0
未找到文件。
.gitignore
0 → 100644
浏览文件 @
391fb9dc
/pkg/
*.pyc
database/atuned.db
common/utils/utils.go
浏览文件 @
391fb9dc
...
@@ -25,6 +25,7 @@ import (
...
@@ -25,6 +25,7 @@ import (
"path"
"path"
"path/filepath"
"path/filepath"
"plugin"
"plugin"
"regexp"
"strconv"
"strconv"
"strings"
"strings"
"syscall"
"syscall"
...
@@ -278,3 +279,13 @@ func DiskByName(disk string) error {
...
@@ -278,3 +279,13 @@ func DiskByName(disk string) error {
return
fmt
.
Errorf
(
"disk %s is not exist"
,
disk
)
return
fmt
.
Errorf
(
"disk %s is not exist"
,
disk
)
}
}
// common input string validator
func
IsInputStringValid
(
input
string
)
bool
{
if
input
!=
""
{
if
isOk
,
_
:=
regexp
.
MatchString
(
"^[a-zA-Z0-9/.-_]*$"
,
input
);
isOk
{
return
isOk
}
}
return
false
}
modules/server/profile/profile.go
浏览文件 @
391fb9dc
...
@@ -691,6 +691,26 @@ func (s *ProfileServer) ProfileRollback(profileInfo *PB.ProfileInfo, stream PB.P
...
@@ -691,6 +691,26 @@ func (s *ProfileServer) ProfileRollback(profileInfo *PB.ProfileInfo, stream PB.P
Collection method call collection script to collect system data.
Collection method call collection script to collect system data.
*/
*/
func
(
s
*
ProfileServer
)
Collection
(
message
*
PB
.
CollectFlag
,
stream
PB
.
ProfileMgr_CollectionServer
)
error
{
func
(
s
*
ProfileServer
)
Collection
(
message
*
PB
.
CollectFlag
,
stream
PB
.
ProfileMgr_CollectionServer
)
error
{
if
valid
:=
utils
.
IsInputStringValid
(
message
.
GetWorkload
());
!
valid
{
return
fmt
.
Errorf
(
"input:%s is invalid"
,
message
.
GetWorkload
())
}
if
valid
:=
utils
.
IsInputStringValid
(
message
.
GetOutputPath
());
!
valid
{
return
fmt
.
Errorf
(
"input:%s is invalid"
,
message
.
GetOutputPath
())
}
if
valid
:=
utils
.
IsInputStringValid
(
message
.
GetType
());
!
valid
{
return
fmt
.
Errorf
(
"input:%s is invalid"
,
message
.
GetType
())
}
if
valid
:=
utils
.
IsInputStringValid
(
message
.
GetBlock
());
!
valid
{
return
fmt
.
Errorf
(
"input:%s is invalid"
,
message
.
GetBlock
())
}
if
valid
:=
utils
.
IsInputStringValid
(
message
.
GetNetwork
());
!
valid
{
return
fmt
.
Errorf
(
"input:%s is invalid"
,
message
.
GetNetwork
())
}
classApps
:=
&
sqlstore
.
GetClassApp
{
Class
:
message
.
GetType
()}
classApps
:=
&
sqlstore
.
GetClassApp
{
Class
:
message
.
GetType
()}
err
:=
sqlstore
.
GetClassApps
(
classApps
)
err
:=
sqlstore
.
GetClassApps
(
classApps
)
if
err
!=
nil
{
if
err
!=
nil
{
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录