69 lines
2.0 KiB
JavaScript
69 lines
2.0 KiB
JavaScript
import { decodeProto } from "../../lib/protobuf-decoder/protobufDecoder";
|
|
import { TOTP } from "../../lib/totp-quickjs";
|
|
import { base64decode } from "../../lib/totp-quickjs/base32decoder";
|
|
|
|
const otpauthScheme = "otpauth:/";
|
|
const googleMigrationScheme = "otpauth-migration:/";
|
|
|
|
export function getTOTPByLink(link) {
|
|
if(link.includes(otpauthScheme))
|
|
return getByOtpauthScheme(link)
|
|
if(link.includes(googleMigrationScheme))
|
|
return getByGoogleMigrationScheme(link)
|
|
|
|
return null;
|
|
}
|
|
|
|
function getHashType(algorithm) {
|
|
if (algorithm == "SHA1") return "SHA-1";
|
|
if (algorithm == "SHA256") return "SHA-256";
|
|
if (algorithm == "SHA512") return "SHA-512";
|
|
else return "SHA-1";
|
|
}
|
|
|
|
function getByOtpauthScheme(link){
|
|
try {
|
|
let args = link.split("/", otpauthScheme.length);
|
|
let type = args[2]; //Returns 'hotp' or 'totp'
|
|
let issuer = args[3].split(":")[0]?.split("?")[0]; //Returns issuer
|
|
let client =
|
|
args[3].split(":")[1]?.split("?")[0] ??
|
|
args[3].split(":")[0]?.split("?")[0]; //Returns client
|
|
let secret = args[3].split("secret=")[1]?.split("&")[0]; //Returns secret
|
|
let period = args[3].split("period=")[1]?.split("&")[0]; //Returns period
|
|
let digits = args[3].split("digits=")[1]?.split("&")[0]; //Returns digits
|
|
let algorithm = args[3].split("algorithm=")[1]?.split("&")[0]; //Returns algorithm
|
|
|
|
if (type.toLowerCase() != "totp")
|
|
throw new Error("Type is not valid, requires 'TOTP'");
|
|
|
|
if (secret === undefined) throw new Error("Secret not defined");
|
|
|
|
if(issuer == client){
|
|
issuer = args[3].split("issuer=")[1]?.split("&")[0]
|
|
}
|
|
|
|
issuer = decodeURIComponent(issuer);
|
|
client = decodeURIComponent(client);
|
|
|
|
return new TOTP(
|
|
secret,
|
|
issuer,
|
|
client,
|
|
digits,
|
|
period,
|
|
0,
|
|
getHashType(algorithm)
|
|
);
|
|
} catch (err) {
|
|
console.log(err)
|
|
return null;
|
|
}
|
|
}
|
|
|
|
function getByGoogleMigrationScheme(link){
|
|
console.log("Hello")
|
|
let data = link.split("data=")[1]; //Returns secret
|
|
let decodedProto = decodeProto(base64decode(data));
|
|
console.log(decodedProto)
|
|
} |