JAVA / 安全 / 小秘密 · 2020-07-23 0

java安全编码指南之:表达式规则

简介

在java编写过程中,我们会使用到各种各样的表达式,在使用表达式的过程中,有哪些安全问题需要我们注意的呢?一起来看看吧。

注意表达式的返回值

我们在使用JDK库的时候,一定要注意认真的读一下JDK中方法的含义和它的返回值。

有些返回值可能表示这个操作是否成功,有的返回值可能是方法操作的结果。我们看两个常见的例子:

    public void deleteFileWrong(){
        File file= new File("/tmp/www.flydean.com.txt");
        file.delete();
        System.out.println("File delete success!");
    }

    public void deleteFileRight(){
        File file= new File("/tmp/www.flydean.com.txt");
        if(file.delete()){
            System.out.println("File delete success!");
        }
    }

先看一个文件删除的例子,delete方法是有返回值的,所以我们在调用delete方法之后,一定要判断一下返回值,看是否删除成功。

再看一个常见的String中字符替换的例子:

    public void stringReplaceWrong(){
        String url="www.flydean.com";
        url.replace("www","WWW");
        System.out.println("replaced url..."+url);
    }
    public void stringReplaceRight(){
        String url="www.flydean.com";
        url=url.replace("www","WWW");
        System.out.println("replaced url..."+url);
    }

我们要记住,String是不可变的,所以它的replace方法,会返回一个替换过后的String,但是原String是不变的,所以我们需要将返回值重新赋值。

注意避免NullPointerException

NullPointerException应该是最最常见的运行时异常了。怎么避免这个异常呢?

我们要做的就是在调用object的方法时候,一定要判断这个object是不是为空。

在JDK8之后,我们引入了Stream操作:

    public void streamWrong(Collection<Object> collection){
        collection.stream().filter(obj->obj.equals("www.flydean.com"));
    }

Stream操作的过程中,我们需要注意stream中的元素是否为空。

有时候,我们可能觉得已经判断是为空了,但是条件判断不准确,导致未知的异常,看下面这个例子:

    public int countWrong(Collection<Object> collection, Object object){
        int count=0;
        if(collection ==null){
            return count;
        }
        for(Object element: collection){
            if((element ==null && object== null)
                || element.equals(object)){
                count++;
            }
        }
        return count;
    }

这个例子是用来查找collection中到底有多少元素和object相同,如果两者都为空,也记为相同。

但是上面的例子有一个漏洞,它没有考虑element ==null 而 object !=null的情况,所以会导致NullPointerException的生成。

我们需要这样修改:

   public int countRight(Collection<Object> collection, Object object){
        int count=0;
        if(collection ==null){
            return count;
        }
        for(Object element: collection){
            if((element ==null && object== null)
                    || (element !=null && element.equals(object))){
                count++;
            }
        }
        return count;
    }

数组相等的判断

如果我们需要比较两个数组是否相等,其实我们想比较的是两个数组中的元素是否相等。

我们知道数组是一个特殊的Object,那么数组对象也有一个equals方法,考虑下面的例子:

    public boolean compareWrong(){
        int[] array1 = new int[10];
        int[] array2 = new int[10];
        return array1.equals(array2);
    }

返回的结果是false,因为数组直接使用了Object中定义的equals方法,我们看下该方法的定义:

    public boolean equals(Object obj) {
        return (this == obj);
    }

可以看到,该方法比较的是两个地址是否相等。所以我们的到了false结果。

其实,我们可以使用Arrays.equals工具类中的方法来进行两个数组的比较:

    public boolean compareRight(){
        int[] array1 = new int[10];
        int[] array2 = new int[10];
        return Arrays.equals(array1, array2);
    }

基础类型的封装类间的比较

在java中,我们知道有一些基础类型像boolean, byte,char, short, int他们会有相对应的封装类型:Boolean,Byte,Character,Short,Integer等。

我们可以直接将基础类型的值赋值给封装类型,封装类型会自行进行转换。

考虑下面的例子:

        Boolean boolA=true;
        Boolean boolB=true;
        System.out.println(boolA==boolB);

结果是多少呢?

答案是true。为什么两个不同对象的比较会是true呢?

在回答这个问题之前,我们看一下字符串的比较:

        String stringA="www.flydean.com";
        String stringB="www.flydean.com";
        System.out.println(stringA==stringB);

这个我们大家应该都知道,因为String有一个字符串常量池,直接从字符串常量构建的String对象,其实是同一个对象。

同样的对于Boolean和Byte来说,如果直接从基础类值构建的话,也是同一个对象。

而对于Character来说,如果值的范围在\u0000 to \u007f,则属于同一个对象,如果超出了这个范围,则是不同的对象。

对于Integer和Short来说,如果值的范围在-128 and 127,则属于同一个对象,如果超出了这个范围,则是不同的对象。

再考虑下面的例子:

Boolean boolA=true;
Boolean boolC=new Boolean(true);
System.out.println(boolA==boolC);

输出的结果是false,因为boolC使用了new关键字,构建了一个新的对象。

集合中类型不匹配

现在java集合可以通过指定类型,从而只存储特定类型的对象。考虑下面的一个例子:

    public void typeMismatch(){
        HashSet<Short> shortSet= new HashSet<>();
        for(int i=0;i<10;i++){
            shortSet.add((short)i);
            shortSet.remove(i);
        }
        System.out.println(shortSet.size());
    }

上面代码我们定义了一个Short的集合,然后将0-9添加进去,接着我们又调用了remove方法把i从集合删除。

但是最后输出结果是10,表明我们并没有删除成功。为什么呢?

看下HashSet的remove方法:

    public boolean remove(Object o) {
        return map.remove(o)==PRESENT;
    }

remove方法的参数是Object,我们传入的i是int类型的,跟short不匹配,所以导致删除失败。

我们需要这样修改:

    public void typeMatch(){
        HashSet<Short> shortSet= new HashSet<>();
        for(int i=0;i<10;i++){
            shortSet.add((short)i);
            shortSet.remove((short)i);
        }
        System.out.println(shortSet.size());
    }

Asset的副作用

我们会使用Asset语句在代码中做调试使用,在使用的过程中需要注意Asset语句不要对系统的业务逻辑产生副作用,也就是说即使Asset语句不运行,也不会修改代码的业务逻辑。

看下面的例子:

    public void assetWrong(ArrayList<Integer> list){
        assert  list.remove(0)>0;
    }

上的代码我们从list中删除第一个元素,并判断删除的元素是否大于0.

上面的代码如果assert语句不执行的话,会导致业务逻辑也不执行,所以需要修改成下面这样:

    public void assetRight(ArrayList<Integer> list){
        int result=list.remove(0);
        assert  result>0;
    }

本文的例子:

learn-java-base-9-to-20/tree/master/security

本文已收录于 http://www.flydean.com/java-security-code-line-expresion/

最通俗的解读,最深刻的干货,最简洁的教程,众多你不知道的小技巧等你来发现!

欢迎关注我的公众号:「程序那些事」,懂技术,更懂你!