com.esri.gpt.control.webharvest.validator.WAFValidator.java Source code

Introduction

Here is the source code for com.esri.gpt.control.webharvest.validator.WAFValidator.java

Source

/*
 * Copyright 2013 Esri.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.esri.gpt.control.webharvest.validator;

import com.esri.gpt.catalog.harvest.protocols.HarvestProtocolWaf;
import com.esri.gpt.framework.http.CredentialProvider;
import java.io.IOException;
import java.net.MalformedURLException;
import java.net.SocketException;
import java.net.URL;
import org.apache.commons.net.ftp.FTPClient;

/**
 * WAF protocol validator.
 */
class WAFValidator extends GenericValidator {
    private HarvestProtocolWaf protocol;

    /**
     * Creates instance of the validator.
     *
     * @param url host URL
     * @param protocol protocol
     */
    public WAFValidator(String url, HarvestProtocolWaf protocol) {
        super(url);
        this.protocol = protocol;
    }

    @Override
    public boolean validate(IMessageCollector mb) {
        boolean _valid = true;

        if (url.isEmpty()) {
            mb.addErrorMessage("catalog.harvest.manage.edit.err.hostUrlReq");
            _valid = false;
        }

        return _valid;
    }

    @Override
    public boolean checkConnection(IMessageCollector mb) {
        if (url.toLowerCase().startsWith("ftp://")) {
            FTPClient client = null;
            try {
                URL u = new URL(url);
                String host = u.getHost();
                int port = u.getPort() >= 0 ? u.getPort() : 21;
                client = new FTPClient();
                client.connect(host, port);
                CredentialProvider cp = getCredentialProvider();
                if (cp == null) {
                    client.login("anonymous", "anonymous");
                } else {
                    client.login(cp.getUsername(), cp.getPassword());
                }
                return true;
            } catch (MalformedURLException ex) {
                mb.addErrorMessage("catalog.harvest.manage.test.err.HarvestInvalidUrl");
                return false;
            } catch (IOException ex) {
                mb.addErrorMessage("catalog.harvest.manage.test.err.HarvestConnectionException");
                return false;
            } finally {
                if (client != null) {
                    try {
                        client.disconnect();
                    } catch (IOException ex) {
                    }
                }
            }
        } else {
            return super.checkConnection(mb);
        }
    }

    @Override
    protected CredentialProvider getCredentialProvider() {
        String userName = protocol.getUserName();
        String userPassword = protocol.getUserPassword();
        if (!userName.isEmpty() && !userPassword.isEmpty()) {
            return new CredentialProvider(userName, userPassword);
        }
        return null;
    }
}