libre.sh/keycloak-scim
3
0
Fork 0
Code Issues 15 Pull requests Releases Packages Activity Actions

Updated TotpGenerator tool to read secret from System.in instead of arg

Browse source
This commit is contained in:
Stian Thorgersen 2013-11-28 18:07:55 +00:00
parent 64396a16b8
commit 46f7e1f7fc
3 changed files with 35 additions and 19 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
testsuite/integration/README.md
View file

@ -30,11 +30,11 @@ TOTP codes
To generate totp codes without Google authenticator run:
mvn exec:java -Ptotp -Dsecret='PJBX GURY NZIT C2JX I44T S3D2 JBKD G6SB'
mvn exec:java -Ptotp
or run org.keycloak.testutils.TotpGenerator from your favourite IDE!
Replace value of -Dsecret with the secret from the totp configuration page (remember quotes!)
Once started copy/paste the totp secret and press enter. To use a new secret just copy/paste and press enter again.
Mail server
-----------

3
testsuite/integration/pom.xml
View file

@ -268,9 +268,6 @@
<artifactId>exec-maven-plugin</artifactId>
<configuration>
<mainClass>org.keycloak.testutils.TotpGenerator</mainClass>
<arguments>
<argument>${secret}</argument>
</arguments>
</configuration>
</plugin>
</plugins>

47
testsuite/integration/src/main/java/org/keycloak/testutils/TotpGenerator.java
View file

@ -3,6 +3,10 @@ package org.keycloak.testutils;
import org.keycloak.models.utils.Base32;
import org.keycloak.models.utils.TimeBasedOTP;
import java.io.BufferedInputStream;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Timer;
import java.util.TimerTask;
import java.util.concurrent.TimeUnit;
@ -10,23 +14,38 @@ import java.util.concurrent.TimeUnit;
public class TotpGenerator {
public static void main(String[] args) {
String totp = "";
for (String a : args) {
totp += a.trim();
}
totp = totp.replace(" ", "");
public static void main(String[] args) throws IOException {
final String google = new String(Base32.decode(totp));
final TimeBasedOTP otp = new TimeBasedOTP();
Timer timer = new Timer();
TotpTask task = null;
Timer t = new Timer();
t.schedule(new TimerTask() {
@Override
public void run() {
System.out.println(otp.generate(google));
BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
System.out.print("Insert secret: ");
for (String l = br.readLine(); true; l = br.readLine()) {
if (task != null) {
task.cancel();
}
}, 0, TimeUnit.SECONDS.toMillis(TimeBasedOTP.DEFAULT_INTERVAL_SECONDS));
System.out.println("Secret: " + l);
task = new TotpTask(l);
timer.schedule(task, 0, TimeUnit.SECONDS.toMillis(TimeBasedOTP.DEFAULT_INTERVAL_SECONDS));
}
}
private static class TotpTask extends TimerTask {
private String secret;
private TotpTask(String secret) {
this.secret = secret;
}
@Override
public void run() {
String google = new String(Base32.decode(secret));
TimeBasedOTP otp = new TimeBasedOTP();
System.out.println(otp.generate(google));
}
}
}