Signal-iOS/SignalServiceKit/tests/Messages/TestProtocolRunnerTest.swift
Evan Hahn 370ff654e7
Change license to AGPL
Change license to AGPL

This commit:

- Updates the `LICENSE` file

- Start every file with something like:

      // Copyright YEAR_FIRST_PUBLISHED Signal Messenger, LLC
      // SPDX-License-Identifier: AGPL-3.0-only

---

First, I removed existing license headers with this Ruby 3.1.2 script:

    require 'set'

    EXTENSIONS_TO_CHECK = Set['.h', '.hpp', '.cpp', '.m', '.mm', '.pch', '.swift']

    same = 0
    different = 0

    all_files = `git ls-files`.lines.map { |line| line.strip }
    all_files.each do |relative_path|
      if relative_path == 'Pods'
        next
      end

      unless EXTENSIONS_TO_CHECK.include? File.extname(relative_path)
        next
      end

      path = File.expand_path(relative_path)

      contents = File.read(path)
      new_contents = contents.sub(/\/\/\n\/\/  Copyright .*\n\/\/\n\n/, '')

      if contents == new_contents
        same += 1
      else
        different += 1
      end

      File.write(path, new_contents)
    end

    puts "updated #{different} file(s), left #{same} untouched"

I'm sure this script could be improved, but it worked well enough.

Then, I created `Scripts/lint/lint-license-headers` and ran it to auto-
fix a lot of files. This changed the mode of some files, but I think
that's actually desirable. For example,
`SignalServiceKit/src/Util/AppContext.m` previously had a mode of
`0755/-rwxr-xr-x`, and it's now `0644/-rw-r--r--`.

Then I fixed some stragglers and updated the precommit script.

See [a similar change in the Desktop app][0].

[0]: 8bfaf598af
2022-10-13 08:25:37 -05:00

171 lines
8.6 KiB
Swift

//
// Copyright 2019 Signal Messenger, LLC
// SPDX-License-Identifier: AGPL-3.0-only
//
import XCTest
import LibSignalClient
import SignalServiceKit
class TestProtocolRunnerTest: SSKBaseTestSwift {
var aliceClient: FakeSignalClient!
var bobClient: FakeSignalClient!
override func setUp() {
super.setUp()
aliceClient = FakeSignalClient.generate(e164Identifier: "+122233alice")
bobClient = FakeSignalClient.generate(e164Identifier: "+12223334bob")
tsAccountManager.registerForTests(withLocalNumber: "+13235551234", uuid: UUID())
}
let runner = TestProtocolRunner()
func test_roundtrip() {
write { transaction in
try! self.runner.initialize(senderClient: self.aliceClient, recipientClient: self.bobClient, transaction: transaction)
let plaintext = "Those who stands for nothing will fall for anything"
let cipherMessage = try! self.runner.encrypt(plaintext.data(using: .utf8)!,
senderClient: self.aliceClient,
recipient: self.bobClient.protocolAddress,
context: NullContext())
let decrypted = try! self.runner.decrypt(cipherMessage,
recipientClient: self.bobClient,
sender: self.aliceClient.protocolAddress,
context: NullContext())
let decryptedText = String(data: decrypted, encoding: .utf8)!
XCTAssertEqual(plaintext, decryptedText)
}
}
func test_multiple() {
write { transaction in
try! self.runner.initialize(senderClient: self.aliceClient, recipientClient: self.bobClient, transaction: transaction)
// two encrypts
let plaintext1 = "Those who stands for nothing will fall for anything"
let cipherMessage1 = try! self.runner.encrypt(plaintext1.data(using: .utf8)!,
senderClient: self.aliceClient,
recipient: self.bobClient.protocolAddress,
context: NullContext())
let plaintext2 = "Do not despair when your enemy attacks you."
let cipherMessage2 = try! self.runner.encrypt(plaintext2.data(using: .utf8)!,
senderClient: self.aliceClient,
recipient: self.bobClient.protocolAddress,
context: NullContext())
// two decrypts
let decrypted1 = try! self.runner.decrypt(cipherMessage1,
recipientClient: self.bobClient,
sender: self.aliceClient.protocolAddress,
context: NullContext())
let decrypted2 = try! self.runner.decrypt(cipherMessage2,
recipientClient: self.bobClient,
sender: self.aliceClient.protocolAddress,
context: NullContext())
let decryptedText1 = String(data: decrypted1, encoding: .utf8)!
XCTAssertEqual(plaintext1, decryptedText1)
let decryptedText2 = String(data: decrypted2, encoding: .utf8)!
XCTAssertEqual(plaintext2, decryptedText2)
}
}
func test_outOfOrder() {
write { transaction in
try! self.runner.initialize(senderClient: self.aliceClient, recipientClient: self.bobClient, transaction: transaction)
// two encrypts
let plaintext1 = "Those who stands for nothing will fall for anything"
let cipherMessage1 = try! self.runner.encrypt(plaintext1.data(using: .utf8)!,
senderClient: self.aliceClient,
recipient: self.bobClient.protocolAddress,
context: NullContext())
let plaintext2 = "Do not despair when your enemy attacks you."
let cipherMessage2 = try! self.runner.encrypt(plaintext2.data(using: .utf8)!,
senderClient: self.aliceClient,
recipient: self.bobClient.protocolAddress,
context: NullContext())
// decrypt second message first
let decrypted2 = try! self.runner.decrypt(cipherMessage2,
recipientClient: self.bobClient,
sender: self.aliceClient.protocolAddress,
context: NullContext())
let decrypted1 = try! self.runner.decrypt(cipherMessage1,
recipientClient: self.bobClient,
sender: self.aliceClient.protocolAddress,
context: NullContext())
let decryptedText1 = String(data: decrypted1, encoding: .utf8)!
XCTAssertEqual(plaintext1, decryptedText1)
let decryptedText2 = String(data: decrypted2, encoding: .utf8)!
XCTAssertEqual(plaintext2, decryptedText2)
}
}
func test_localClient_receives() {
Self.identityManager.generateNewIdentityKey(for: .aci)
Self.tsAccountManager.registerForTests(withLocalNumber: "+13235551234",
uuid: UUID())
let localClient = LocalSignalClient()
write { transaction in
try! self.runner.initialize(senderClient: self.bobClient,
recipientClient: localClient,
transaction: transaction)
let plaintext = "Those who stands for nothing will fall for anything"
let cipherMessage = try! self.runner.encrypt(plaintext.data(using: .utf8)!,
senderClient: self.bobClient,
recipient: localClient.protocolAddress,
context: transaction)
let decrypted = try! self.runner.decrypt(cipherMessage,
recipientClient: localClient,
sender: self.bobClient.protocolAddress,
context: transaction)
let decryptedText = String(data: decrypted, encoding: .utf8)!
XCTAssertEqual(plaintext, decryptedText)
}
}
func test_localClient_sends() {
Self.identityManager.generateNewIdentityKey(for: .aci)
Self.tsAccountManager.registerForTests(withLocalNumber: "+13235551234",
uuid: UUID())
let localClient = LocalSignalClient()
write { transaction in
try! self.runner.initialize(senderClient: localClient,
recipientClient: self.bobClient,
transaction: transaction)
let plaintext = "Those who stands for nothing will fall for anything"
let cipherMessage = try! self.runner.encrypt(plaintext.data(using: .utf8)!,
senderClient: localClient,
recipient: self.bobClient.protocolAddress,
context: transaction)
let decrypted = try! self.runner.decrypt(cipherMessage,
recipientClient: self.bobClient,
sender: localClient.protocolAddress,
context: transaction)
let decryptedText = String(data: decrypted, encoding: .utf8)!
XCTAssertEqual(plaintext, decryptedText)
}
}
}