/**
* Copyright (c) 2013 - 2019 LEAP Encryption Access Project and contributors
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program. If not, see .
*/
package se.leap.bitmaskclient.eip;
import android.content.Context;
import com.google.gson.Gson;
import com.google.gson.reflect.TypeToken;
import org.json.JSONArray;
import org.json.JSONException;
import org.json.JSONObject;
import java.io.IOException;
import java.lang.reflect.Type;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import de.blinkt.openvpn.VpnProfile;
import de.blinkt.openvpn.core.ConfigParser;
import de.blinkt.openvpn.core.VpnStatus;
import de.blinkt.openvpn.core.connection.Connection;
import se.leap.bitmaskclient.Provider;
import se.leap.bitmaskclient.ProviderObservable;
import static de.blinkt.openvpn.core.connection.Connection.TransportType.OBFS4;
import static de.blinkt.openvpn.core.connection.Connection.TransportType.OPENVPN;
import static se.leap.bitmaskclient.Constants.GATEWAYS;
import static se.leap.bitmaskclient.Constants.PROVIDER_PRIVATE_KEY;
import static se.leap.bitmaskclient.Constants.PROVIDER_VPN_CERTIFICATE;
import static se.leap.bitmaskclient.utils.PreferenceHelper.getUsePluggableTransports;
/**
* @author parmegv
*/
public class GatewaysManager {
private static final String TAG = GatewaysManager.class.getSimpleName();
private Context context;
private LinkedHashMap gateways = new LinkedHashMap<>();
private Type listType = new TypeToken>() {}.getType();
private ArrayList presortedList = new ArrayList<>();
public GatewaysManager(Context context) {
configureFromCurrentProvider();
this.context = context;
}
/**
* select closest Gateway
* @return the n closest Gateway
*/
public Gateway select(int nClosest) {
Connection.TransportType transportType = getUsePluggableTransports(context) ? OBFS4 : OPENVPN;
if (presortedList.size() > 0) {
return getGatewayFromPresortedList(nClosest, transportType);
}
return getGatewayFromTimezoneCalculation(nClosest, transportType);
}
private Gateway getGatewayFromTimezoneCalculation(int nClosest, Connection.TransportType transportType) {
List list = new ArrayList<>(gateways.values());
GatewaySelector gatewaySelector = new GatewaySelector(list);
Gateway gateway;
int found = 0;
int i = 0;
while ((gateway = gatewaySelector.select(i)) != null) {
if (gateway.suppoortsTransport(transportType)) {
if (found == nClosest) {
return gateway;
}
found++;
}
i++;
}
return null;
}
private Gateway getGatewayFromPresortedList(int nClosest, Connection.TransportType transportType) {
int found = 0;
for (Gateway gateway : presortedList) {
if (gateway.suppoortsTransport(transportType)) {
if (found == nClosest) {
return gateway;
}
found++;
}
}
return null;
}
/**
* Get position of the gateway from a sorted set (along the distance of the gw to your time zone)
* @param profile profile belonging to a gateway
* @return position of the gateway owning to the profile
*/
public int getPosition(VpnProfile profile) {
if (presortedList.size() > 0) {
return getPositionFromPresortedList(profile);
}
return getPositionFromTimezoneCalculatedList(profile);
}
private int getPositionFromPresortedList(VpnProfile profile) {
Connection.TransportType transportType = profile.mUsePluggableTransports ? OBFS4 : OPENVPN;
int nClosest = 0;
for (Gateway gateway : presortedList) {
if (gateway.suppoortsTransport(transportType)) {
if (profile.equals(gateway.getProfile(transportType))) {
return nClosest;
}
nClosest++;
}
}
return -1;
}
private int getPositionFromTimezoneCalculatedList(VpnProfile profile) {
Connection.TransportType transportType = profile.mUsePluggableTransports ? OBFS4 : OPENVPN;
GatewaySelector gatewaySelector = new GatewaySelector(new ArrayList<>(gateways.values()));
Gateway gateway;
int nClosest = 0;
int i = 0;
while ((gateway = gatewaySelector.select(i)) != null) {
if (gateway.suppoortsTransport(transportType)) {
if (profile.equals(gateway.getProfile(transportType))) {
return nClosest;
}
nClosest++;
}
i++;
}
return -1;
}
/**
* check if there are no gateways defined
* @return true if no gateways defined else false
*/
public boolean isEmpty() {
return gateways.isEmpty();
}
/**
* @return number of gateways defined in the GatewaysManager
*/
public int size() {
return gateways.size();
}
@Override
public String toString() {
return new Gson().toJson(gateways, listType);
}
/**
* parse gateways from Provider's eip service
* @param provider
*/
private void parseDefaultGateways(Provider provider) {
try {
JSONObject eipDefinition = provider.getEipServiceJson();
JSONObject secrets = secretsConfigurationFromCurrentProvider();
JSONArray gatewaysDefined = new JSONArray();
try {
gatewaysDefined = eipDefinition.getJSONArray(GATEWAYS);
} catch (Exception e) {
e.printStackTrace();
}
for (int i = 0; i < gatewaysDefined.length(); i++) {
try {
JSONObject gw = gatewaysDefined.getJSONObject(i);
Gateway aux = new Gateway(eipDefinition, secrets, gw, this.context);
if (gateways.get(aux.getHost()) == null) {
addGateway(aux);
}
} catch (JSONException | ConfigParser.ConfigParseError | IOException e) {
e.printStackTrace();
VpnStatus.logError("Unable to parse gateway config!");
}
}
} catch (NullPointerException npe) {
npe.printStackTrace();
}
}
private void parseGatewaysFromGeoIpServiceJson(Provider provider) {
try {
JSONObject geoIpJson = provider.getGeoIpJson();
JSONArray gatewaylist = geoIpJson.getJSONArray(GATEWAYS);
for (int i = 0; i < gatewaylist.length(); i++) {
try {
String key = gatewaylist.getString(i);
if (gateways.containsKey(key)) {
presortedList.add(gateways.get(key));
}
} catch (JSONException e) {
e.printStackTrace();
}
}
} catch (NullPointerException | JSONException npe) {
npe.printStackTrace();
}
}
private JSONObject secretsConfigurationFromCurrentProvider() {
JSONObject result = new JSONObject();
Provider provider = ProviderObservable.getInstance().getCurrentProvider();
try {
result.put(Provider.CA_CERT, provider.getCaCert());
result.put(PROVIDER_PRIVATE_KEY, provider.getPrivateKey());
result.put(PROVIDER_VPN_CERTIFICATE, provider.getVpnCertificate());
} catch (JSONException e) {
e.printStackTrace();
}
return result;
}
private void addGateway(Gateway gateway) {
gateways.put(gateway.getHost(), gateway);
}
private void configureFromCurrentProvider() {
Provider provider = ProviderObservable.getInstance().getCurrentProvider();
parseDefaultGateways(provider);
parseGatewaysFromGeoIpServiceJson(provider);
}
}