Search in sources :

Example 1 with I18N

use of org.zaproxy.zap.utils.I18N in project zaproxy by zaproxy.

the class Constant method initializeFilesAndDirectories.

public void initializeFilesAndDirectories() {
    FileCopier copier = new FileCopier();
    File f = null;
    // Set up the version from the manifest
    PROGRAM_VERSION = getVersionFromManifest();
    PROGRAM_TITLE = PROGRAM_NAME + " " + PROGRAM_VERSION;
    if (zapHome == null) {
        zapHome = getDefaultHomeDirectory(true);
    }
    zapHome = getAbsolutePath(zapHome);
    f = new File(zapHome);
    FILE_CONFIG = zapHome + FILE_CONFIG;
    FOLDER_SESSION = zapHome + FOLDER_SESSION;
    DBNAME_UNTITLED = zapHome + DBNAME_UNTITLED;
    ACCEPTED_LICENSE = zapHome + ACCEPTED_LICENSE;
    DIRBUSTER_CUSTOM_DIR = zapHome + DIRBUSTER_DIR;
    FUZZER_DIR = zapHome + FUZZER_DIR;
    FOLDER_LOCAL_PLUGIN = zapHome + FOLDER_LOCAL_PLUGIN;
    try {
        System.setProperty(SYSTEM_PAROS_USER_LOG, zapHome);
        if (!f.isDirectory()) {
            if (!f.mkdir()) {
                // ZAP: report failure to create directory
                System.out.println("Failed to create directory " + f.getAbsolutePath());
            }
        }
        // Setup the logging
        File logFile = new File(zapHome + "/log4j.properties");
        if (!logFile.exists()) {
            copier.copy(new File(zapInstall, "xml/log4j.properties"), logFile);
        }
        System.setProperty("log4j.configuration", logFile.getAbsolutePath());
        PropertyConfigurator.configure(logFile.getAbsolutePath());
        f = new File(FILE_CONFIG);
        if (!f.isFile()) {
            this.copyDefaultConfigs(f, false);
        }
        f = new File(FOLDER_SESSION);
        if (!f.isDirectory()) {
            LOG.info("Creating directory " + FOLDER_SESSION);
            if (!f.mkdir()) {
                // ZAP: report failure to create directory
                System.out.println("Failed to create directory " + f.getAbsolutePath());
            }
        }
        f = new File(DIRBUSTER_CUSTOM_DIR);
        if (!f.isDirectory()) {
            LOG.info("Creating directory " + DIRBUSTER_CUSTOM_DIR);
            if (!f.mkdir()) {
                // ZAP: report failure to create directory
                System.out.println("Failed to create directory " + f.getAbsolutePath());
            }
        }
        f = new File(FUZZER_DIR);
        if (!f.isDirectory()) {
            LOG.info("Creating directory " + FUZZER_DIR);
            if (!f.mkdir()) {
                // ZAP: report failure to create directory
                System.out.println("Failed to create directory " + f.getAbsolutePath());
            }
        }
        f = new File(FOLDER_LOCAL_PLUGIN);
        if (!f.isDirectory()) {
            LOG.info("Creating directory " + FOLDER_LOCAL_PLUGIN);
            if (!f.mkdir()) {
                // ZAP: report failure to create directory
                System.out.println("Failed to create directory " + f.getAbsolutePath());
            }
        }
        f = new File(zapHome, FOLDER_FILTER);
        if (!f.isDirectory()) {
            LOG.info("Creating directory: " + f.getAbsolutePath());
            if (!f.mkdir()) {
                System.out.println("Failed to create directory " + f.getAbsolutePath());
            }
        }
    } catch (Exception e) {
        System.err.println("Unable to initialize home directory! " + e.getMessage());
        e.printStackTrace(System.err);
        System.exit(1);
    }
    // Upgrade actions
    try {
        try {
            // ZAP: Changed to use ZapXmlConfiguration, to enforce the same character encoding when reading/writing configurations.
            XMLConfiguration config = new ZapXmlConfiguration(FILE_CONFIG);
            config.setAutoSave(false);
            long ver = config.getLong("version");
            if (ver == VERSION_TAG) {
            // Nothing to do
            } else if (isDevBuild() || isDailyBuild()) {
            // Nothing to do
            } else {
                // Backup the old one
                LOG.info("Backing up config file to " + FILE_CONFIG + ".bak");
                f = new File(FILE_CONFIG);
                try {
                    copier.copy(f, new File(FILE_CONFIG + ".bak"));
                } catch (IOException e) {
                    String msg = "Failed to backup config file " + FILE_CONFIG + " to " + FILE_CONFIG + ".bak " + e.getMessage();
                    System.err.println(msg);
                    LOG.error(msg, e);
                }
                if (ver == V_PAROS_TAG) {
                    upgradeFrom1_1_0(config);
                    upgradeFrom1_2_0(config);
                }
                if (ver <= V_1_0_0_TAG) {
                // Nothing to do
                }
                if (ver <= V_1_1_0_TAG) {
                    upgradeFrom1_1_0(config);
                }
                if (ver <= V_1_2_0_TAG) {
                    upgradeFrom1_2_0(config);
                }
                if (ver <= V_1_2_1_TAG) {
                // Nothing to do
                }
                if (ver <= V_1_3_0_TAG) {
                // Nothing to do
                }
                if (ver <= V_1_3_1_TAG) {
                // Nothing to do
                }
                if (ver <= V_1_4_1_TAG) {
                    upgradeFrom1_4_1(config);
                }
                if (ver <= V_2_0_0_TAG) {
                    upgradeFrom2_0_0(config);
                }
                if (ver <= V_2_1_0_TAG) {
                // Nothing to do
                }
                if (ver <= V_2_2_0_TAG) {
                    upgradeFrom2_2_0(config);
                }
                if (ver <= V_2_3_1_TAG) {
                    upgradeFrom2_3_1(config);
                }
                if (ver <= V_2_4_3_TAG) {
                    upgradeFrom2_4_3(config);
                }
                if (ver <= V_2_5_0_TAG) {
                    upgradeFrom2_5_0(config);
                }
                LOG.info("Upgraded from " + ver);
                // Update the version
                config.setProperty("version", VERSION_TAG);
                config.save();
            }
        } catch (ConfigurationException | ConversionException | NoSuchElementException e) {
            //  if there is any error in config file (eg config file not exist, corrupted),
            //  overwrite previous configuration file 
            // ZAP: changed to use the correct file
            copier.copy(getPathDefaultConfigFile().toFile(), new File(FILE_CONFIG));
        }
    } catch (Exception e) {
        System.err.println("Unable to upgrade config file " + FILE_CONFIG + " " + e.getMessage());
        e.printStackTrace(System.err);
        System.exit(1);
    }
    // ZAP: Init i18n
    String lang;
    Locale locale = Locale.ENGLISH;
    try {
        // Select the correct locale
        // ZAP: Changed to use ZapXmlConfiguration, to enforce the same character encoding when reading/writing configurations.
        XMLConfiguration config = new ZapXmlConfiguration(FILE_CONFIG);
        config.setAutoSave(false);
        lang = config.getString(OptionsParamView.LOCALE, OptionsParamView.DEFAULT_LOCALE);
        if (lang.length() == 0) {
            lang = OptionsParamView.DEFAULT_LOCALE;
        }
        String[] langArray = lang.split("_");
        locale = new Locale(langArray[0], langArray[1]);
    } catch (Exception e) {
        System.out.println("Failed to initialise locale " + e);
    }
    Locale.setDefault(locale);
    messages = new I18N(locale);
}
Also used : ConversionException(org.apache.commons.configuration.ConversionException) Locale(java.util.Locale) IOException(java.io.IOException) FileCopier(org.parosproxy.paros.model.FileCopier) ConversionException(org.apache.commons.configuration.ConversionException) InvalidParameterException(java.security.InvalidParameterException) ParseException(java.text.ParseException) NoSuchElementException(java.util.NoSuchElementException) IOException(java.io.IOException) ConfigurationException(org.apache.commons.configuration.ConfigurationException) XMLConfiguration(org.apache.commons.configuration.XMLConfiguration) ConfigurationException(org.apache.commons.configuration.ConfigurationException) ZapXmlConfiguration(org.zaproxy.zap.utils.ZapXmlConfiguration) File(java.io.File) NoSuchElementException(java.util.NoSuchElementException) I18N(org.zaproxy.zap.utils.I18N)

Example 2 with I18N

use of org.zaproxy.zap.utils.I18N in project zaproxy by zaproxy.

the class WithConfigsTest method setUp.

@Before
public void setUp() throws Exception {
    /*
        // Useful if you need to get some info when debugging
        BasicConfigurator.configure();
        ConsoleAppender ca = new ConsoleAppender();
        ca.setWriter(new OutputStreamWriter(System.out));
        ca.setLayout(new PatternLayout("%-5p [%t]: %m%n"));
        Logger.getRootLogger().addAppender(ca);
        Logger.getRootLogger().setLevel(Level.DEBUG);
/**/
    Constant.setZapInstall(INSTALL_PATH);
    HOME_DIR.mkdirs();
    Constant.setZapHome(HOME_DIR.getAbsolutePath());
    File langDir = new File(Constant.getZapInstall(), "lang");
    ClassLoaderUtil.addFile(langDir.getAbsolutePath());
    ExtensionLoader extLoader = Mockito.mock(ExtensionLoader.class);
    Control control = Mockito.mock(Control.class);
    Mockito.when(control.getExtensionLoader()).thenReturn(extLoader);
    // Init all the things
    Constant.getInstance();
    I18N i18n = Mockito.mock(I18N.class);
    given(i18n.getString(anyString())).willReturn("");
    given(i18n.getString(anyString(), anyObject())).willReturn("");
    given(i18n.getLocal()).willReturn(Locale.getDefault());
    Constant.messages = i18n;
    Control.initSingletonForTesting(Model.getSingleton());
    Mockito.when(control.getExtensionLoader()).thenReturn(extLoader);
}
Also used : Control(org.parosproxy.paros.control.Control) File(java.io.File) ExtensionLoader(org.parosproxy.paros.extension.ExtensionLoader) I18N(org.zaproxy.zap.utils.I18N) Before(org.junit.Before)

Example 3 with I18N

use of org.zaproxy.zap.utils.I18N in project zaproxy by zaproxy.

the class PluginFactoryUnitTest method mockConstantClass.

private void mockConstantClass() {
    PluginFactory pf = PowerMockito.mock(PluginFactory.class);
    Whitebox.setInternalState(pf, "loadedPlugins", null);
    Constant constant = PowerMockito.mock(Constant.class);
    Whitebox.setInternalState(constant, "instance", constant);
    Whitebox.setInternalState(constant, "zapInstall", ".");
    I18N i18n = PowerMockito.mock(I18N.class);
    given(i18n.getString(anyString())).willReturn("");
    given(i18n.getString(anyString(), anyObject())).willReturn("");
    Whitebox.setInternalState(constant, "messages", i18n);
    Model.getSingleton().getOptionsParam().load(new ZapXmlConfiguration());
}
Also used : Constant(org.parosproxy.paros.Constant) ZapXmlConfiguration(org.zaproxy.zap.utils.ZapXmlConfiguration) I18N(org.zaproxy.zap.utils.I18N)

Example 4 with I18N

use of org.zaproxy.zap.utils.I18N in project zaproxy by zaproxy.

the class UsersListModelUnitTest method mockConstantClass.

private void mockConstantClass() {
    Constant constant = PowerMockito.mock(Constant.class);
    I18N i18n = PowerMockito.mock(I18N.class);
    given(i18n.getString(anyString())).willReturn("");
    given(i18n.getString(anyString(), anyObject())).willReturn("");
    Whitebox.setInternalState(constant, "messages", i18n);
}
Also used : Constant(org.parosproxy.paros.Constant) I18N(org.zaproxy.zap.utils.I18N)

Example 5 with I18N

use of org.zaproxy.zap.utils.I18N in project zaproxy by zaproxy.

the class Constant method setLocale.

public static void setLocale(String loc) {
    String[] langArray = loc.split("_");
    Locale locale = new Locale(langArray[0], langArray[1]);
    Locale.setDefault(locale);
    if (messages == null) {
        messages = new I18N(locale);
    } else {
        messages.setLocale(locale);
    }
}
Also used : Locale(java.util.Locale) I18N(org.zaproxy.zap.utils.I18N)

Aggregations

I18N (org.zaproxy.zap.utils.I18N)6 Constant (org.parosproxy.paros.Constant)3 File (java.io.File)2 Locale (java.util.Locale)2 ZapXmlConfiguration (org.zaproxy.zap.utils.ZapXmlConfiguration)2 IOException (java.io.IOException)1 InvalidParameterException (java.security.InvalidParameterException)1 ParseException (java.text.ParseException)1 NoSuchElementException (java.util.NoSuchElementException)1 ConfigurationException (org.apache.commons.configuration.ConfigurationException)1 ConversionException (org.apache.commons.configuration.ConversionException)1 XMLConfiguration (org.apache.commons.configuration.XMLConfiguration)1 Before (org.junit.Before)1 Control (org.parosproxy.paros.control.Control)1 ExtensionLoader (org.parosproxy.paros.extension.ExtensionLoader)1 FileCopier (org.parosproxy.paros.model.FileCopier)1