001    /**
002     * Copyright (c) 2000-2013 Liferay, Inc. All rights reserved.
003     *
004     * This library is free software; you can redistribute it and/or modify it under
005     * the terms of the GNU Lesser General Public License as published by the Free
006     * Software Foundation; either version 2.1 of the License, or (at your option)
007     * any later version.
008     *
009     * This library is distributed in the hope that it will be useful, but WITHOUT
010     * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
011     * FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more
012     * details.
013     */
014    
015    package com.liferay.portlet.documentlibrary.antivirus;
016    
017    import com.liferay.portal.kernel.exception.SystemException;
018    
019    import java.io.File;
020    import java.io.InputStream;
021    
022    /**
023     * @author Brian Wing Shun Chan
024     */
025    public class AntivirusScannerWrapper implements AntivirusScanner {
026    
027            public AntivirusScannerWrapper(AntivirusScanner antivirusScanner) {
028                    _originalAntivirusScanner = antivirusScanner;
029                    _antivirusScanner = antivirusScanner;
030            }
031    
032            @Override
033            public boolean isActive() {
034                    return _antivirusScanner.isActive();
035            }
036    
037            @Override
038            public void scan(byte[] bytes)
039                    throws AntivirusScannerException, SystemException {
040    
041                    _antivirusScanner.scan(bytes);
042            }
043    
044            @Override
045            public void scan(File file)
046                    throws AntivirusScannerException, SystemException {
047    
048                    _antivirusScanner.scan(file);
049            }
050    
051            @Override
052            public void scan(InputStream inputStream)
053                    throws AntivirusScannerException, SystemException {
054    
055                    _antivirusScanner.scan(inputStream);
056            }
057    
058            public void setAntivirusScanner(AntivirusScanner antivirusScanner) {
059                    if (antivirusScanner == null) {
060                            _antivirusScanner = _originalAntivirusScanner;
061                    }
062                    else {
063                            _antivirusScanner = antivirusScanner;
064                    }
065            }
066    
067            private AntivirusScanner _antivirusScanner;
068            private AntivirusScanner _originalAntivirusScanner;
069    
070    }