使用springboot实现上传文件时校验文件是否有病毒

clamd.exe --install

clamd.exe --status

        <dependency>
            <groupId>fi.solita.clamav</groupId>
            <artifactId>clamav-client</artifactId>
            <version>1.0.1</version>
        </dependency>

# clamav 文件杀毒服务器连接配置
clamav:
  # 服务器ip
  ipAddress: 172.16.1.1
  # 服务器端口 (Integer类型)
  port: 3310
  # 超时时间 (Integer类型)
  timeout: 5000000

import fi.solita.clamav.ClamAVClient;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

/**
 * ClamAV 客户端配置
 *
 * @author 
 */
@Configuration
public class ClamAVClientConfig {

    @Value("${clamav.ipAddress}")
    private String ipAddress;

    @Value("${clamav.port}")
    private Integer port;

    @Value("${clamav.timeout}")
    private Integer timeout;

    @Bean
    public ClamAVClient clamAVClient() {
        return new ClamAVClient(ipAddress, port, timeout);
    }
}

import lombok.Getter;

/**
 * 文件杀毒 返回结果枚举类
 */

@Getter
public enum ClamAVEnum {

    OK("stream: OK\u0000","ok",1,"文件无病毒"),
    ALLOCATE_MEMORY("stream: Can't allocate memory ERROR\u0000","allocate_memory",-1,"文件超出杀毒服务最大文件限制"),
    INFECTED("stream:INFECTED ","Infected",-1,"文件存在病毒");

    private String code;
    private String msg;
    private Integer result;
    private String desc;

    ClamAVEnum(String code, String msg, Integer result, String desc) {
        this.code = code;
        this.msg = msg;
        this.result = result;
        this.desc = desc;
    }
}

    @Autowired
    private ClamAVClient clamAVClient;

    @PostMapping(value = "/uploadFile")
	public R<?> uploadFile(@RequestParam("file") MultipartFile file) {
        // 判断文件是否存在病毒
        try {
            InputStream inputStream = file.getInputStream();
            byte[] scan = clamAVClient.scan(inputStream);
            String res = new String(scan, StandardCharsets.UTF_8);
            if(ClamAVEnum.OK.getCode().equals(res)){
                System.out.println(ClamAVEnum.OK.getDesc());
            }else if(ClamAVEnum.ALLOCATE_MEMORY.getCode().equals(res)){
                return R.failure(ClamAVEnum.ALLOCATE_MEMORY.getDesc());
            }else {
                return R.failure(ClamAVEnum.INFECTED.getDesc());
            }
        } catch (IOException e) {
            try {
                clamAVClient.ping();
            } catch (IOException ioException) {
                return R.failure("连接杀毒服务失败！");
            }
            return R.failure("连接杀毒服务异常或上传文件超过杀毒软件最大文件限制！");
        }

        return R.data("ok");
    }