moparisthebest/filebot
1
0
Fork 0
mirror of https://github.com/mitb-archive/filebot synced 2024-12-23 00:08:51 -05:00
Code Issues Releases Wiki Activity

Refactor

Browse Source
This commit is contained in:
Reinhard Pointner 2018-07-09 15:45:18 +07:00
parent 749e614d50
commit 6f481c0102
1 changed files with 89 additions and 44 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

133
source/net/filebot/License.java
View File

@ -8,7 +8,6 @@ import static net.filebot.util.RegularExpressions.*;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.Serializable;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.StandardCopyOption;
@ -19,55 +18,118 @@ import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
import org.apache.commons.io.IOUtils;
import net.filebot.util.SystemProperty;
import net.filebot.web.WebRequest;
public class License implements Serializable {
public class License {
private String product;
private String id;
private Instant expires;
private final String product;
private final String id;
private final Instant expires;
private Exception error;
private final Exception error;
private License(String product, String id, Instant expires) {
this.product = product;
this.id = id;
this.expires = expires;
this.error = null;
}
private License(Exception error) {
this.product = null;
this.id = null;
this.expires = null;
this.error = error;
}
public boolean expires() {
return expires != null;
}
public boolean error() {
return error != null;
}
public License check() throws Exception {
if (error()) {
throw error;
}
if (expires()) {
checkExpirationDate(expires);
}
return this;
}
@Override
public String toString() {
// FileBot License T1000
StringBuilder s = new StringBuilder().append(product).append(" License ").append(id);
// Valid-Until: 2019-07-04
if (expires()) {
s.append(" (Valid-Until: ").append(formatExpirationDate(expires)).append(")");
}
return s.toString();
}
public static License parseLicenseFile(File file) {
if (!file.exists()) {
return new License(new FileNotFoundException("UNREGISTERED"));
}
public License(File file) {
try {
// read and verify license file
if (!file.exists()) {
throw new FileNotFoundException("UNREGISTERED");
}
byte[] bytes = Files.readAllBytes(file.toPath());
// verify and get clear signed content
Map<String, String> properties = getProperties(bytes);
this.product = properties.get("Product");
this.id = properties.get("Order");
this.expires = LocalDate.parse(properties.get("Valid-Until"), DateTimeFormatter.ISO_LOCAL_DATE).atStartOfDay(ZoneOffset.UTC).plusDays(1).minusSeconds(1).toInstant();
String product = properties.get("Product");
String id = properties.get("Order");
Instant expires = Optional.ofNullable(properties.get("Valid-Until")).map(License::parseExpirationDate).orElse(null);
// check validity
checkExpirationDate(expires);
// verify license online
verifyLicense(bytes);
} catch (Exception e) {
error = e;
verifyLicense(id, bytes);
return new License(product, id, expires);
} catch (Exception error) {
return new License(error);
}
}
private String getExpirationDate() {
return expires == null ? null : expires.atZone(ZoneOffset.UTC).format(DateTimeFormatter.ISO_LOCAL_DATE);
}
private Map<String, String> getProperties(byte[] bytes) throws Exception {
byte[] pub = IOUtils.toByteArray(getClass().getResource("license.key"));
private static Map<String, String> getProperties(byte[] bytes) throws Exception {
byte[] pub = IOUtils.toByteArray(License.class.getResource("license.key"));
String msg = verifyClearSignMessage(bytes, pub);
return NEWLINE.splitAsStream(msg).map(s -> s.split(": ", 2)).collect(toMap(a -> a[0], a -> a[1]));
}
private void verifyLicense(byte[] bytes) throws Exception {
private static Instant parseExpirationDate(String date) {
return LocalDate.parse(date, DateTimeFormatter.ISO_LOCAL_DATE).atStartOfDay(ZoneOffset.UTC).plusDays(1).minusSeconds(1).toInstant();
}
private static String formatExpirationDate(Instant expires) {
return expires.atZone(ZoneOffset.UTC).format(DateTimeFormatter.ISO_LOCAL_DATE);
}
private static void checkExpirationDate(Instant expires) throws Exception {
if (expires != null && Instant.now().isAfter(expires)) {
throw new IllegalStateException("EXPIRED since " + formatExpirationDate(expires));
}
}
private static void verifyLicense(String id, byte[] bytes) throws Exception {
Cache cache = CacheManager.getInstance().getCache("license", CacheType.Persistent);
Object json = cache.json(id, i -> new URL("https://license.filebot.net/verify/" + i)).fetch((url, modified) -> WebRequest.post(url, bytes, "application/octet-stream", getRequestParameters())).expire(Cache.ONE_MONTH).get();
@ -76,7 +138,7 @@ public class License implements Serializable {
}
}
private Map<String, String> getRequestParameters() {
private static Map<String, String> getRequestParameters() {
Map<String, String> parameters = new HashMap<String, String>(2);
// add standard HTTP headers
@ -89,31 +151,14 @@ public class License implements Serializable {
return parameters;
}
public License check() throws Exception {
if (error != null) {
throw error;
}
if (Instant.now().isAfter(expires)) {
throw new IllegalStateException("EXPIRED since " + getExpirationDate());
}
return this;
}
@Override
public String toString() {
return String.format("%s License %s (Valid-Until: %s)", product, id, getExpirationDate());
}
public static final SystemProperty<File> FILE = SystemProperty.of("net.filebot.license", File::new, ApplicationFolder.AppData.resolve("license.txt"));
public static final MemoizedResource<License> INSTANCE = Resource.lazy(() -> new License(FILE.get()));
public static final MemoizedResource<License> INSTANCE = Resource.lazy(() -> parseLicenseFile(FILE.get()));
public static License importLicenseFile(File file) throws Exception {
// lock memoized resource while validating and setting a new license
synchronized (License.INSTANCE) {
// check if license file is valid and not expired
License license = new License(file).check();
License license = parseLicenseFile(file).check();
// write to default license file path
Files.copy(file.toPath(), License.FILE.get().toPath(), StandardCopyOption.REPLACE_EXISTING);