Handle UnauthenticatedResponseException internally

This commit is contained in:
AsamK 2021-10-31 20:53:03 +01:00
parent 782f96b580
commit 31dec5a666
7 changed files with 16 additions and 24 deletions

View file

@ -30,7 +30,6 @@ import org.whispersystems.signalservice.api.messages.SignalServiceContent;
import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope; import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope;
import org.whispersystems.signalservice.api.push.SignalServiceAddress; import org.whispersystems.signalservice.api.push.SignalServiceAddress;
import org.whispersystems.signalservice.api.util.PhoneNumberFormatter; import org.whispersystems.signalservice.api.util.PhoneNumberFormatter;
import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
import java.io.Closeable; import java.io.Closeable;
import java.io.File; import java.io.File;
@ -118,7 +117,7 @@ public interface Manager extends Closeable {
void addDeviceLink(URI linkUri) throws IOException, InvalidDeviceLinkException; void addDeviceLink(URI linkUri) throws IOException, InvalidDeviceLinkException;
void setRegistrationLockPin(Optional<String> pin) throws IOException, UnauthenticatedResponseException; void setRegistrationLockPin(Optional<String> pin) throws IOException;
Profile getRecipientProfile(RecipientIdentifier.Single recipient) throws IOException; Profile getRecipientProfile(RecipientIdentifier.Single recipient) throws IOException;

View file

@ -382,7 +382,7 @@ public class ManagerImpl implements Manager {
public void deleteAccount() throws IOException { public void deleteAccount() throws IOException {
try { try {
pinHelper.removeRegistrationLockPin(); pinHelper.removeRegistrationLockPin();
} catch (UnauthenticatedResponseException e) { } catch (IOException e) {
logger.warn("Failed to remove registration lock pin"); logger.warn("Failed to remove registration lock pin");
} }
account.setRegistrationLockPin(null, null); account.setRegistrationLockPin(null, null);
@ -453,7 +453,7 @@ public class ManagerImpl implements Manager {
} }
@Override @Override
public void setRegistrationLockPin(java.util.Optional<String> pin) throws IOException, UnauthenticatedResponseException { public void setRegistrationLockPin(java.util.Optional<String> pin) throws IOException {
if (!account.isMasterDevice()) { if (!account.isMasterDevice()) {
throw new RuntimeException("Only master device can set a PIN"); throw new RuntimeException("Only master device can set a PIN");
} }

View file

@ -23,18 +23,26 @@ public class PinHelper {
public void setRegistrationLockPin( public void setRegistrationLockPin(
String pin, MasterKey masterKey String pin, MasterKey masterKey
) throws IOException, UnauthenticatedResponseException { ) throws IOException {
final var pinChangeSession = keyBackupService.newPinChangeSession(); final var pinChangeSession = keyBackupService.newPinChangeSession();
final var hashedPin = PinHashing.hashPin(pin, pinChangeSession); final var hashedPin = PinHashing.hashPin(pin, pinChangeSession);
pinChangeSession.setPin(hashedPin, masterKey); try {
pinChangeSession.setPin(hashedPin, masterKey);
} catch (UnauthenticatedResponseException e) {
throw new IOException(e);
}
pinChangeSession.enableRegistrationLock(masterKey); pinChangeSession.enableRegistrationLock(masterKey);
} }
public void removeRegistrationLockPin() throws IOException, UnauthenticatedResponseException { public void removeRegistrationLockPin() throws IOException {
final var pinChangeSession = keyBackupService.newPinChangeSession(); final var pinChangeSession = keyBackupService.newPinChangeSession();
pinChangeSession.disableRegistrationLock(); pinChangeSession.disableRegistrationLock();
pinChangeSession.removePin(); try {
pinChangeSession.removePin();
} catch (UnauthenticatedResponseException e) {
throw new IOException(e);
}
} }
public KbsPinData getRegistrationLockData( public KbsPinData getRegistrationLockData(

View file

@ -6,9 +6,7 @@ import net.sourceforge.argparse4j.inf.Subparser;
import org.asamk.signal.OutputWriter; import org.asamk.signal.OutputWriter;
import org.asamk.signal.commands.exceptions.CommandException; import org.asamk.signal.commands.exceptions.CommandException;
import org.asamk.signal.commands.exceptions.IOErrorException; import org.asamk.signal.commands.exceptions.IOErrorException;
import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
import org.asamk.signal.manager.Manager; import org.asamk.signal.manager.Manager;
import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
import java.io.IOException; import java.io.IOException;
import java.util.Optional; import java.util.Optional;
@ -31,8 +29,6 @@ public class RemovePinCommand implements JsonRpcLocalCommand {
) throws CommandException { ) throws CommandException {
try { try {
m.setRegistrationLockPin(Optional.empty()); m.setRegistrationLockPin(Optional.empty());
} catch (UnauthenticatedResponseException e) {
throw new UnexpectedErrorException("Remove pin failed with unauthenticated response: " + e.getMessage(), e);
} catch (IOException e) { } catch (IOException e) {
throw new IOErrorException("Remove pin error: " + e.getMessage(), e); throw new IOErrorException("Remove pin error: " + e.getMessage(), e);
} }

View file

@ -6,9 +6,7 @@ import net.sourceforge.argparse4j.inf.Subparser;
import org.asamk.signal.OutputWriter; import org.asamk.signal.OutputWriter;
import org.asamk.signal.commands.exceptions.CommandException; import org.asamk.signal.commands.exceptions.CommandException;
import org.asamk.signal.commands.exceptions.IOErrorException; import org.asamk.signal.commands.exceptions.IOErrorException;
import org.asamk.signal.commands.exceptions.UnexpectedErrorException;
import org.asamk.signal.manager.Manager; import org.asamk.signal.manager.Manager;
import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
import java.io.IOException; import java.io.IOException;
import java.util.Optional; import java.util.Optional;
@ -34,9 +32,6 @@ public class SetPinCommand implements JsonRpcLocalCommand {
try { try {
var registrationLockPin = ns.getString("pin"); var registrationLockPin = ns.getString("pin");
m.setRegistrationLockPin(Optional.of(registrationLockPin)); m.setRegistrationLockPin(Optional.of(registrationLockPin));
} catch (UnauthenticatedResponseException e) {
throw new UnexpectedErrorException("Set pin error failed with unauthenticated response: " + e.getMessage(),
e);
} catch (IOException e) { } catch (IOException e) {
throw new IOErrorException("Set pin error: " + e.getMessage(), e); throw new IOErrorException("Set pin error: " + e.getMessage(), e);
} }

View file

@ -37,7 +37,6 @@ import org.whispersystems.signalservice.api.messages.SignalServiceAttachmentRemo
import org.whispersystems.signalservice.api.push.SignalServiceAddress; import org.whispersystems.signalservice.api.push.SignalServiceAddress;
import org.whispersystems.signalservice.api.push.exceptions.UnregisteredUserException; import org.whispersystems.signalservice.api.push.exceptions.UnregisteredUserException;
import org.whispersystems.signalservice.api.util.UuidUtil; import org.whispersystems.signalservice.api.util.UuidUtil;
import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
import java.io.File; import java.io.File;
import java.io.IOException; import java.io.IOException;
@ -167,7 +166,7 @@ public class DbusManagerImpl implements Manager {
} }
@Override @Override
public void setRegistrationLockPin(final Optional<String> pin) throws IOException, UnauthenticatedResponseException { public void setRegistrationLockPin(final Optional<String> pin) throws IOException {
if (pin.isPresent()) { if (pin.isPresent()) {
signal.setPin(pin.get()); signal.setPin(pin.get());
} else { } else {

View file

@ -34,7 +34,6 @@ import org.freedesktop.dbus.exceptions.DBusExecutionException;
import org.freedesktop.dbus.types.Variant; import org.freedesktop.dbus.types.Variant;
import org.whispersystems.signalservice.api.messages.SendMessageResult; import org.whispersystems.signalservice.api.messages.SendMessageResult;
import org.whispersystems.signalservice.api.util.InvalidNumberException; import org.whispersystems.signalservice.api.util.InvalidNumberException;
import org.whispersystems.signalservice.internal.contacts.crypto.UnauthenticatedResponseException;
import java.io.File; import java.io.File;
import java.io.IOException; import java.io.IOException;
@ -583,8 +582,6 @@ public class DbusSignalImpl implements Signal {
public void removePin() { public void removePin() {
try { try {
m.setRegistrationLockPin(Optional.empty()); m.setRegistrationLockPin(Optional.empty());
} catch (UnauthenticatedResponseException e) {
throw new Error.Failure("Remove pin failed with unauthenticated response: " + e.getMessage());
} catch (IOException e) { } catch (IOException e) {
throw new Error.Failure("Remove pin error: " + e.getMessage()); throw new Error.Failure("Remove pin error: " + e.getMessage());
} }
@ -594,8 +591,6 @@ public class DbusSignalImpl implements Signal {
public void setPin(String registrationLockPin) { public void setPin(String registrationLockPin) {
try { try {
m.setRegistrationLockPin(Optional.of(registrationLockPin)); m.setRegistrationLockPin(Optional.of(registrationLockPin));
} catch (UnauthenticatedResponseException e) {
throw new Error.Failure("Set pin error failed with unauthenticated response: " + e.getMessage());
} catch (IOException e) { } catch (IOException e) {
throw new Error.Failure("Set pin error: " + e.getMessage()); throw new Error.Failure("Set pin error: " + e.getMessage());
} }