fastdfs实现防盗链功能

news2024/11/15 12:24:06

一、实现原理

二、开启防盗链

三、重启nginx

四、 Java代码生成token

1、token生成规则

2、java生成token

五、测试

1 带正确token访问

2 带错误token访问，或直接访问

一、实现原理

fastdfs是一个分布式文件系统，如果我们的fastdfs部署在外网，那么任何一个人知道了我们的上传接口，那么它就可以文件的上传和访问。那么我们如何阻止他人访问我们fastdfs服务器上的文件呢？因此就需要使用fastdfs的防盗链功能。

原理：fastdfs的防盗链是通过token机制来实现的。当我们开启防盗链功能后，需要在url后增加2个额外的参数token和ts。token和ts的生成都是需要在服务端。

二、开启防盗链

vi /etc/fdfs/http.conf

# true 表示开启防盗链
http.anti_steal.check_token = true
# token的过期时间，单位为秒
http.anti_steal.token_ttl = 300
# 密钥，不可泄漏，用于生成token
http.anti_steal.secret_key = thisisasecuritykey
# 当图片拒绝访问后，显示的图片，此图片需要可访问，不然可能会出现问题
http.anti_steal.token_check_fail = /data/fastdfs/404.jpg

http.anti_steal.token_check_fail 指定的图片需要可访问，否则可能会出现问题

三、重启nginx

/usr/local/nginx/sbin/nginx -s reload

四、 Java代码生成token

1、token生成规则

token = md5(文件ID+私钥+时间戳) 文件ID：不能包含group

group1/M00/00/00/wKh5iWNBl7-AKvj1AAAwWD4VeAg577.jpg
`需要替换成`
M00/00/00/wKh5iWNBl7-AKvj1AAAwWD4VeAg577.jpg

私钥：需要和 /etc/fdfs/http.conf 中的 http.anti_steal.secret_key 值一致
时间戳：单位秒

2、java生成token

/**
     * 生成token
     *
     * @param fileId          the filename return by FastDFS server,不能含有组
     * @param timestampSecond 时间戳 单位秒
     * @return token
     * @throws NoSuchAlgorithmException
     */
    private String generatorToken(String fileId, Long timestampSecond) throws NoSuchAlgorithmException {
        // 需要去掉 group
        fileId = fileId.substring(fileId.indexOf("/") + 1);
        byte[] bsFilename = fileId.getBytes(StandardCharsets.UTF_8);
        byte[] bsTimestamp = timestampSecond.toString().getBytes(StandardCharsets.UTF_8);
        // thisisasecuritykey 需要和 /etc/fdfs/http.conf 中的 http.anti_steal.secret_key 值一致
        byte[] bsKey = "thisisasecuritykey".getBytes(StandardCharsets.UTF_8);

        byte[] buff = new byte[bsFilename.length + bsKey.length + bsTimestamp.length];
        System.arraycopy(bsFilename, 0, buff, 0, bsFilename.length);
        System.arraycopy(bsKey, 0, buff, bsFilename.length, bsKey.length);
        System.arraycopy(bsTimestamp, 0, buff, bsFilename.length + bsKey.length, bsTimestamp.length);

        return md5(buff);
    }

    public static String md5(byte[] source) throws NoSuchAlgorithmException {
        char hexDigits[] = {'0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd', 'e', 'f'};
        java.security.MessageDigest md = java.security.MessageDigest.getInstance("MD5");
        md.update(source);
        byte tmp[] = md.digest();
        char str[] = new char[32];
        int k = 0;
        for (int i = 0; i < 16; i++) {
            str[k++] = hexDigits[tmp[i] >>> 4 & 0xf];
            str[k++] = hexDigits[tmp[i] & 0xf];
        }

        return new String(str);
    }