java编写shell命令解析规则

因业务需求,要对shell命令添加一个白名单,因此需要对shell进行一个简单的解析,代码跑了两天基本满足现阶段要求,代码如下:

public static void main(String[] args) {
		System.out.println(checkCmd("zgrep oHQwN0ZYayuTyNZHAosTKc7uP6vk $logpath | grep -o \"cmd=[^ ]*\" | sort | uniq -c".trim()));
	}

    public static boolean checkCmd(String grep) {
		List<List<String>> unitList = new ArrayList<>();
		List<String> tokenList = new ArrayList<>();
		boolean flag = false;
		char punc = '~';
		StringBuilder sb = new StringBuilder();
		int pre = 0, i;
		for (i = 0; i < grep.length() && pre < grep.length(); i++) {
			char c = grep.charAt(i);
			if (flag) {
				if (c == punc) {
					flag = false;
					tokenList.add(grep.substring(pre + 1, i));
					pre = i + 1;
				}
			} else {
				if (c == ' ' || c == '|') {
					tokenList.add(grep.substring(pre, i));
					pre = i + 1;
					if (c == '|') {
						unitList.add(new ArrayList<>(tokenList));
						tokenList.clear();
					}
				} else if ((c == '\'' || c == '"') && !flag) {
					flag = true;
					punc = c;
				}
			}
		}
		if (pre < grep.length()) {
			tokenList.add(grep.substring(pre, i));
			unitList.add(new ArrayList<>(tokenList));
		}
		//拿到白名单
		List<String> wightTokenList = Utils.getTokenWightList();
		for (List<String> list : unitList) {
			List<String> tokens = list.stream().filter(s -> !s.trim().equals("")).collect(Collectors.toList());
			if (!wightTokenList.contains(tokens.get(0)) || !legalPath(tokens)) {
				return false;
			}
		}
		return true;
    }

	private static boolean legalPath(List<String> tokens) {
		if (tokens.get(0).equals("less") || tokens.get(0).equals("cat") || tokens.get(0).equals("nl")) {
			String last = tokens.get(tokens.size() - 1);
			if (last.indexOf("/") > -1) {
				return false;
			}
		}
		return true;
	}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

请保持优秀。

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值