1/*
2 * Copyright (c) 2007, 2015, Oracle and/or its affiliates. All rights reserved.
3 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
4 *
5 * This code is free software; you can redistribute it and/or modify it
6 * under the terms of the GNU General Public License version 2 only, as
7 * published by the Free Software Foundation.
8 *
9 * This code is distributed in the hope that it will be useful, but WITHOUT
10 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
11 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
12 * version 2 for more details (a copy is included in the LICENSE file that
13 * accompanied this code).
14 *
15 * You should have received a copy of the GNU General Public License version
16 * 2 along with this work; if not, write to the Free Software Foundation,
17 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
18 *
19 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
20 * or visit www.oracle.com if you need additional information or have any
21 * questions.
22 */
23import java.security.GeneralSecurityException;
24import java.security.Provider;
25import java.security.Security;
26import javax.crypto.SecretKey;
27import javax.crypto.Cipher;
28import javax.crypto.SecretKeyFactory;
29import javax.crypto.spec.PBEKeySpec;
30import javax.crypto.spec.PBEParameterSpec;
31
32/**
33 * Default wrapper for a password based encryption Cipher.
34 */
35
36public class DefaultPBEWrapper extends AbstractPBEWrapper {
37    /**
38     * Define default SALT size as 8.
39     */
40    private static final int PBE_SALT_SIZE = 8;
41
42    /**
43     * Default PBE wrapper constructor.
44     *
45     * @param algo PGE algorithm to wrap.
46     * @param passwd password phrase
47     */
48    public DefaultPBEWrapper(PBEAlgorithm algo, String passwd) {
49        super(algo, passwd, PBE_SALT_SIZE);
50    }
51
52    /**
53     * Instantiate Cipher for the PBE algorithm.
54     *
55     * @param mode Cipher mode: encrypt or decrypt.
56     * @return Cipher in accordance to the PBE algorithm
57     * @throws java.security.GeneralSecurityException
58     */
59    @Override
60    protected Cipher initCipher(int mode) throws  GeneralSecurityException {
61        Provider provider = Security.getProvider("SunJCE");
62        if (provider == null) {
63            throw new RuntimeException("SunJCE provider does not exist.");
64        }
65        SecretKey key = SecretKeyFactory.getInstance(baseAlgo)
66                .generateSecret(new PBEKeySpec(password.toCharArray()));
67        Cipher ci = Cipher.getInstance(transformation, provider);
68        ci.init(mode, key, new PBEParameterSpec(salt, DEFAULT_ITERATION));
69        return ci;
70    }
71}
72