Make sure the streams are closed properly-adapter
[vfc/nfvo/driver/vnfm/svnfm.git] / huawei / vnfmadapter / VnfmadapterService / service / src / main / java / org / onap / vfc / nfvo / vnfm / svnfm / vnfmadapter / service / csm / connect / AbstractSslContext.java
1 /*
2  * Copyright 2016-2017 Huawei Technologies Co., Ltd.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  *     http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16
17 package org.onap.vfc.nfvo.vnfm.svnfm.vnfmadapter.service.csm.connect;
18
19 import java.io.BufferedInputStream;
20 import java.io.FileInputStream;
21 import java.io.FileNotFoundException;
22 import java.io.IOException;
23 import java.io.InputStream;
24 import java.security.GeneralSecurityException;
25 import java.security.KeyStore;
26 import java.security.NoSuchAlgorithmException;
27 import java.security.SecureRandom;
28 import java.security.cert.X509Certificate;
29
30 import javax.net.ssl.KeyManager;
31 import javax.net.ssl.KeyManagerFactory;
32 import javax.net.ssl.SSLContext;
33 import javax.net.ssl.TrustManager;
34 import javax.net.ssl.TrustManagerFactory;
35 import javax.net.ssl.X509TrustManager;
36
37 import org.onap.vfc.nfvo.vnfm.svnfm.vnfmadapter.common.restclient.SystemEnvVariablesFactory;
38 import org.onap.vfc.nfvo.vnfm.svnfm.vnfmadapter.service.constant.Constant;
39 import org.slf4j.Logger;
40 import org.slf4j.LoggerFactory;
41
42 import net.sf.json.JSONObject;
43
44 /**
45  * SSL context
46  * .</br>
47  *
48  * @author
49  * @version VFC 1.0 Sep 14, 2016
50  */
51 public class AbstractSslContext {
52
53     private static final Logger LOG = LoggerFactory.getLogger(AbstractSslContext.class);
54
55     protected AbstractSslContext() {
56         // constructor
57     }
58
59     private static SSLContext getSSLContext() throws NoSuchAlgorithmException {
60         return SSLContext.getInstance("TLSv1.2");
61     }
62
63     protected static SSLContext getAnonymousSSLContext() throws GeneralSecurityException {
64         SSLContext sslContext = getSSLContext();
65         sslContext.init(null, new TrustManager[] {new TrustAnyTrustManager()}, new SecureRandom());
66         return sslContext;
67     }
68
69     protected static SSLContext getCertificateSSLContext() throws GeneralSecurityException {
70         SSLContext sslContext = getSSLContext();
71         JSONObject sslConf = null;
72         try {
73             sslConf = readSSLConfToJson();
74         } catch(Exception e) {
75             LOG.error("readSSLConfToJson error", e);
76         }
77         sslContext.init(createKeyManager(sslConf), createTrustManager(sslConf), new SecureRandom());
78         return sslContext;
79     }
80
81     protected static KeyManager[] createKeyManager(JSONObject sslConf) {
82         KeyManager[] kms = null;
83         try {
84             String CERT_STORE = "etc/conf/server.p12";
85             String CERT_STORE_PASSWORD = "Changeme_123";  // NOSONAR
86             String KEY_STORE_TYPE = "PKCS12";
87             if(sslConf != null) {
88                 CERT_STORE = sslConf.getString("keyStore");
89                 CERT_STORE_PASSWORD = sslConf.getString("keyStorePass");
90                 KEY_STORE_TYPE = sslConf.getString("keyStoreType");
91             }
92             // load jks file
93             try(FileInputStream f_certStore = new FileInputStream(CERT_STORE)) {
94                 KeyStore ks = KeyStore.getInstance(KEY_STORE_TYPE);
95                 ks.load(f_certStore, CERT_STORE_PASSWORD.toCharArray());
96
97                 // init and create
98                 String alg = KeyManagerFactory.getDefaultAlgorithm();
99                 KeyManagerFactory kmFact = KeyManagerFactory.getInstance(alg);
100                 kmFact.init(ks, CERT_STORE_PASSWORD.toCharArray());
101
102                 kms = kmFact.getKeyManagers();
103             }
104         } catch(Exception e) {
105             LOG.error("create KeyManager fail!", e);
106         }
107         return kms;
108     }
109
110     protected static TrustManager[] createTrustManager(JSONObject sslConf) {
111         TrustManager[] tms = null;
112         try {
113
114             String TRUST_STORE = "etc/conf/trust.jks";
115             String TRUST_STORE_PASSWORD = "Changeme_123";  // NOSONAR
116             String TRUST_STORE_TYPE = "jks";
117             if(sslConf != null) {
118                 TRUST_STORE = sslConf.getString("trustStore");
119                 TRUST_STORE_PASSWORD = sslConf.getString("trustStorePass");
120                 TRUST_STORE_TYPE = sslConf.getString("trustStoreType");
121             }
122             try(FileInputStream f_trustStore = new FileInputStream(TRUST_STORE)) {
123                 KeyStore ks = KeyStore.getInstance(TRUST_STORE_TYPE);
124                 ks.load(f_trustStore, TRUST_STORE_PASSWORD.toCharArray());
125
126                 String alg = TrustManagerFactory.getDefaultAlgorithm();
127                 TrustManagerFactory tmFact = TrustManagerFactory.getInstance(alg);
128                 tmFact.init(ks);
129                 tms = tmFact.getTrustManagers();
130             }
131         } catch(Exception e) {
132             LOG.error("create TrustManager fail!", e);
133         }
134         return tms;
135     }
136
137     /**
138      * readSSLConfToJson
139      *
140      * @return
141      * @throws IOException
142      * @since VFC 1.0
143      */
144     public static JSONObject readSSLConfToJson() throws IOException {
145         JSONObject sslJson = null;
146
147         String fileContent = "";
148
149         String fileName = SystemEnvVariablesFactory.getInstance().getAppRoot()
150                 + System.getProperty(Constant.FILE_SEPARATOR) + "etc" + System.getProperty(Constant.FILE_SEPARATOR)
151                 + "conf" + System.getProperty(Constant.FILE_SEPARATOR) + "sslconf.json";
152
153         try (InputStream ins = new FileInputStream(fileName)) {
154             try(BufferedInputStream bins = new BufferedInputStream(ins)) {
155
156                 byte[] contentByte = new byte[ins.available()];
157                 int num = bins.read(contentByte);
158
159                 if(num > 0) {
160                     fileContent = new String(contentByte);
161                 }
162                 sslJson = JSONObject.fromObject(fileContent);
163             }
164         } catch(FileNotFoundException e) {
165             LOG.error(fileName + "is not found!", e);
166         } catch(Exception e) {
167             LOG.error("read sslconf file fail.please check if the 'sslconf.json' is exist.", e);
168         }
169
170         return sslJson;
171     }
172
173     private static class TrustAnyTrustManager implements X509TrustManager {
174
175         @Override
176         public X509Certificate[] getAcceptedIssuers() {
177             return new X509Certificate[] {};
178         }
179
180         @Override
181         public void checkServerTrusted(X509Certificate[] certs, String authType) {
182             // NOSONAR
183         }
184
185         @Override
186         public void checkClientTrusted(X509Certificate[] certs, String authType) {
187             // NOSONAR
188         }
189     }
190 }