mirror of https://github.com/beemdevelopment/Aegis
Merge pull request #304 from alexbakker/microsoft-auth
Add support for importing from Microsoft Authenticatorpull/319/head
Michael Schättgen
6 years ago
committed by
GitHub
commit
48d22fc06d
@ -0,0 +1,126 @@
|
|||||||
|
package com.beemdevelopment.aegis.importers;
|
||||||
|
|
||||||
|
import android.content.Context;
|
||||||
|
import android.database.Cursor;
|
||||||
|
|
||||||
|
import com.beemdevelopment.aegis.encoding.Base32;
|
||||||
|
import com.beemdevelopment.aegis.encoding.Base64;
|
||||||
|
import com.beemdevelopment.aegis.encoding.EncodingException;
|
||||||
|
import com.beemdevelopment.aegis.otp.OtpInfo;
|
||||||
|
import com.beemdevelopment.aegis.otp.OtpInfoException;
|
||||||
|
import com.beemdevelopment.aegis.otp.TotpInfo;
|
||||||
|
import com.beemdevelopment.aegis.vault.VaultEntry;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
public class MicrosoftAuthImporter extends DatabaseImporter {
|
||||||
|
private static final String _subPath = "databases/PhoneFactor";
|
||||||
|
private static final String _pkgName = "com.azure.authenticator";
|
||||||
|
|
||||||
|
private static final int TYPE_TOTP = 0;
|
||||||
|
private static final int TYPE_MICROSOFT = 1;
|
||||||
|
|
||||||
|
public MicrosoftAuthImporter(Context context) {
|
||||||
|
super(context);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String getAppPkgName() {
|
||||||
|
return _pkgName;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
protected String getAppSubPath() {
|
||||||
|
return _subPath;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public State read(FileReader reader) throws DatabaseImporterException {
|
||||||
|
SqlImporterHelper helper = new SqlImporterHelper(getContext());
|
||||||
|
List<Entry> entries = helper.read(Entry.class, reader.getStream(), "accounts");
|
||||||
|
return new State(entries);
|
||||||
|
}
|
||||||
|
|
||||||
|
public static class State extends DatabaseImporter.State {
|
||||||
|
private List<Entry> _entries;
|
||||||
|
|
||||||
|
private State(List<Entry> entries) {
|
||||||
|
super(false);
|
||||||
|
_entries = entries;
|
||||||
|
}
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public Result convert() {
|
||||||
|
Result result = new Result();
|
||||||
|
|
||||||
|
for (Entry sqlEntry : _entries) {
|
||||||
|
try {
|
||||||
|
int type = sqlEntry.getType();
|
||||||
|
if (type == TYPE_TOTP || type == TYPE_MICROSOFT) {
|
||||||
|
VaultEntry entry = convertEntry(sqlEntry);
|
||||||
|
result.addEntry(entry);
|
||||||
|
}
|
||||||
|
} catch (DatabaseImporterEntryException e) {
|
||||||
|
result.addError(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static VaultEntry convertEntry(Entry entry) throws DatabaseImporterEntryException {
|
||||||
|
try {
|
||||||
|
byte[] secret;
|
||||||
|
int digits = 6;
|
||||||
|
|
||||||
|
switch (entry.getType()) {
|
||||||
|
case TYPE_TOTP:
|
||||||
|
secret = Base32.decode(entry.getSecret());
|
||||||
|
break;
|
||||||
|
case TYPE_MICROSOFT:
|
||||||
|
digits = 8;
|
||||||
|
secret = Base64.decode(entry.getSecret());
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
throw new DatabaseImporterEntryException(String.format("Unsupported OTP type: %d", entry.getType()), entry.toString());
|
||||||
|
}
|
||||||
|
|
||||||
|
OtpInfo info = new TotpInfo(secret, "SHA1", digits, 30);
|
||||||
|
return new VaultEntry(info, entry.getUserName(), entry.getIssuer());
|
||||||
|
} catch (EncodingException | OtpInfoException e) {
|
||||||
|
throw new DatabaseImporterEntryException(e, entry.toString());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static class Entry extends SqlImporterHelper.Entry {
|
||||||
|
private int _type;
|
||||||
|
private String _secret;
|
||||||
|
private String _issuer;
|
||||||
|
private String _userName;
|
||||||
|
|
||||||
|
public Entry(Cursor cursor) {
|
||||||
|
super(cursor);
|
||||||
|
_type = SqlImporterHelper.getInt(cursor, "account_type");
|
||||||
|
_secret = SqlImporterHelper.getString(cursor, "oath_secret_key");
|
||||||
|
_issuer = SqlImporterHelper.getString(cursor, "name");
|
||||||
|
_userName = SqlImporterHelper.getString(cursor, "username");
|
||||||
|
}
|
||||||
|
|
||||||
|
public int getType() {
|
||||||
|
return _type;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getSecret() {
|
||||||
|
return _secret;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getIssuer() {
|
||||||
|
return _issuer;
|
||||||
|
}
|
||||||
|
|
||||||
|
public String getUserName() {
|
||||||
|
return _userName;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@ -0,0 +1,89 @@
|
|||||||
|
package com.beemdevelopment.aegis.importers;
|
||||||
|
|
||||||
|
import android.content.Context;
|
||||||
|
import android.database.Cursor;
|
||||||
|
import android.database.sqlite.SQLiteDatabase;
|
||||||
|
import android.database.sqlite.SQLiteException;
|
||||||
|
|
||||||
|
import com.topjohnwu.superuser.ShellUtils;
|
||||||
|
|
||||||
|
import java.io.File;
|
||||||
|
import java.io.FileOutputStream;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStream;
|
||||||
|
import java.lang.reflect.InvocationTargetException;
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import static android.database.sqlite.SQLiteDatabase.OPEN_READONLY;
|
||||||
|
|
||||||
|
public class SqlImporterHelper {
|
||||||
|
private Context _context;
|
||||||
|
|
||||||
|
public SqlImporterHelper(Context context) {
|
||||||
|
_context = context;
|
||||||
|
}
|
||||||
|
|
||||||
|
public <T extends Entry> List<T> read(Class<T> type, InputStream inStream, String table) throws DatabaseImporterException {
|
||||||
|
File file;
|
||||||
|
|
||||||
|
try {
|
||||||
|
// create a temporary copy of the database so that SQLiteDatabase can open it
|
||||||
|
file = File.createTempFile("db-import-", "", _context.getCacheDir());
|
||||||
|
try (FileOutputStream out = new FileOutputStream(file)) {
|
||||||
|
ShellUtils.pump(inStream, out);
|
||||||
|
}
|
||||||
|
} catch (IOException e) {
|
||||||
|
throw new DatabaseImporterException(e);
|
||||||
|
}
|
||||||
|
|
||||||
|
try (SQLiteDatabase db = SQLiteDatabase.openDatabase(file.getAbsolutePath(), null, OPEN_READONLY, null)) {
|
||||||
|
try (Cursor cursor = db.rawQuery(String.format("SELECT * FROM %s", table), null)) {
|
||||||
|
List<T> entries = new ArrayList<>();
|
||||||
|
|
||||||
|
if (cursor.moveToFirst()) {
|
||||||
|
do {
|
||||||
|
T entry = type.getDeclaredConstructor(Cursor.class).newInstance(cursor);
|
||||||
|
entries.add(entry);
|
||||||
|
} while (cursor.moveToNext());
|
||||||
|
}
|
||||||
|
|
||||||
|
return entries;
|
||||||
|
} catch (InstantiationException | IllegalAccessException
|
||||||
|
| NoSuchMethodException | InvocationTargetException e) {
|
||||||
|
throw new RuntimeException(e);
|
||||||
|
}
|
||||||
|
} catch (SQLiteException e) {
|
||||||
|
throw new DatabaseImporterException(e);
|
||||||
|
} finally {
|
||||||
|
// always delete the temporary file
|
||||||
|
file.delete();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
public static String getString(Cursor cursor, String columnName) {
|
||||||
|
return cursor.getString(cursor.getColumnIndex(columnName));
|
||||||
|
}
|
||||||
|
|
||||||
|
public static String getString(Cursor cursor, String columnName, String def) {
|
||||||
|
String res = cursor.getString(cursor.getColumnIndex(columnName));
|
||||||
|
if (res == null) {
|
||||||
|
return def;
|
||||||
|
}
|
||||||
|
return res;
|
||||||
|
}
|
||||||
|
|
||||||
|
public static int getInt(Cursor cursor, String columnName) {
|
||||||
|
return cursor.getInt(cursor.getColumnIndex(columnName));
|
||||||
|
}
|
||||||
|
|
||||||
|
public static long getLong(Cursor cursor, String columnName) {
|
||||||
|
return cursor.getLong(cursor.getColumnIndex(columnName));
|
||||||
|
}
|
||||||
|
|
||||||
|
public static abstract class Entry {
|
||||||
|
public Entry (Cursor cursor) {
|
||||||
|
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue