org.apache.kerby.has.client.HasInitClient Maven / Gradle / Ivy
/**
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.kerby.has.client;
import org.apache.kerby.has.common.HasConfig;
import org.apache.kerby.kerberos.kerb.KrbException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.MalformedURLException;
import java.net.URL;
/**
* HAS client API for applications to interact with HAS server
*/
public class HasInitClient {
public static final Logger LOG = LoggerFactory.getLogger(HasInitClient.class);
private HasConfig hasConfig;
private File confDir;
public HasInitClient(HasConfig hasConfig, File confDir) {
this.hasConfig = hasConfig;
this.confDir = confDir;
}
public File getConfDir() {
return confDir;
}
private String getInitBaseURL() throws KrbException {
return HasClientUtil.getBaseUrl(hasConfig, "init");
}
private String getConfigBaseURL() throws KrbException {
return HasClientUtil.getBaseUrl(hasConfig, "conf");
}
public String startKdc() throws KrbException {
HttpURLConnection httpConn;
URL url;
try {
url = new URL(getInitBaseURL() + "kdcstart");
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "PUT", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() == 200) {
String response = HasClientUtil.getResponse(httpConn);
LOG.info(response);
return response;
} else {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
public void initKdc(File keytab) throws KrbException {
HttpURLConnection httpConn;
URL url;
try {
url = new URL(getInitBaseURL() + "kdcinit");
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "GET", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() != 200) {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
try (FileOutputStream fos = new FileOutputStream(keytab);
InputStream in = httpConn.getInputStream()) {
byte[] buffer = new byte[3 * 1024];
int read;
while ((read = in.read(buffer)) > 0) {
fos.write(buffer, 0, read);
}
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
public void getKrb5conf(File file) throws KrbException {
HttpURLConnection httpConn;
URL url;
try {
url = new URL(getConfigBaseURL() + "getkrb5conf");
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "GET", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() != 200) {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
try (FileOutputStream fos = new FileOutputStream(file);
InputStream in = httpConn.getInputStream()) {
byte[] buffer = new byte[3 * 1024];
int read;
while ((read = in.read(buffer)) > 0) {
fos.write(buffer, 0, read);
}
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
public void getHasClientConf(File file) throws KrbException {
HttpURLConnection httpConn;
URL url;
try {
url = new URL(getConfigBaseURL() + "gethasclientconf");
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "GET", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() != 200) {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
try (FileOutputStream fos = new FileOutputStream(file);
InputStream in = httpConn.getInputStream()) {
byte[] buffer = new byte[3 * 1024];
int read;
while ((read = in.read(buffer)) > 0) {
fos.write(buffer, 0, read);
}
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
public String setPlugin(String plugin) throws KrbException {
HttpURLConnection httpConn;
URL url;
try {
url = new URL(getConfigBaseURL() + "setplugin?plugin=" + plugin);
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "PUT", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() == 200) {
String response = HasClientUtil.getResponse(httpConn);
LOG.info(response);
return response;
} else {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
public String configKdc(String port, String realm, String host) throws KrbException {
HttpURLConnection httpConn;
URL url;
try {
url = new URL(getConfigBaseURL() + "configkdc?port=" + port + "&realm="
+ realm + "&host=" + host);
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "PUT", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() == 200) {
String response = HasClientUtil.getResponse(httpConn);
LOG.info(response);
return response;
} else {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
public String configBackend(String backendType, String dir, String mysqlUrl, String user,
String password) throws KrbException {
HttpURLConnection httpConn;
URL url;
if (backendType.equals("json")) {
try {
url = new URL(getConfigBaseURL() + "configbackend?backendType=" + backendType
+ "&dir=" + dir);
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
} else if (backendType.equals("mysql")) {
try {
url = new URL(getConfigBaseURL() + "configbackend?backendType=" + backendType
+ "&url=" + mysqlUrl + "&user=" + user + "&password=" + password);
} catch (MalformedURLException e) {
throw new KrbException("Failed to create a URL object.", e);
}
} else {
throw new KrbException("Unsupported backend: " + backendType);
}
httpConn = HasClientUtil.createConnection(hasConfig, url, "PUT", false);
try {
httpConn.connect();
if (httpConn.getResponseCode() == 200) {
String response = HasClientUtil.getResponse(httpConn);
LOG.info(response);
return response;
} else {
throw new KrbException(HasClientUtil.getResponse(httpConn));
}
} catch (IOException e) {
throw new KrbException("IO error occurred. " + e.getMessage());
}
}
}