Signal/Signal-Server
1
0
Fork 0
mirror of https://github.com/signalapp/Signal-Server synced 2026-04-20 21:38:06 +01:00
Code Issues Packages Projects Releases Wiki Activity

Use the S3 object monitor to retrieve Tor exit node lists.

Browse Source
This commit is contained in:
Jon Chambers
2021-05-17 17:58:13 -04:00
committed by Jon Chambers
parent cfa8cbedc1
commit fbaf4a09e2
4 changed files with 79 additions and 183 deletions
Download Patch File Download Diff File Expand all files Collapse all files

87
service/src/test/java/org/whispersystems/textsecuregcm/util/TorExitNodeManagerTest.java
View File

@@ -5,89 +5,32 @@
package org.whispersystems.textsecuregcm.util;
import com.github.tomakehurst.wiremock.junit.WireMockRule;
import com.github.tomakehurst.wiremock.matching.StringValuePattern;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import java.io.ByteArrayInputStream;
import java.nio.charset.StandardCharsets;
import java.util.concurrent.ScheduledExecutorService;
import org.junit.Test;
import org.whispersystems.textsecuregcm.configuration.TorExitNodeConfiguration;
import org.whispersystems.textsecuregcm.redis.AbstractRedisClusterTest;
import java.util.UUID;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.ScheduledExecutorService;
import java.util.concurrent.TimeUnit;
import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
import static com.github.tomakehurst.wiremock.client.WireMock.equalTo;
import static com.github.tomakehurst.wiremock.client.WireMock.get;
import static com.github.tomakehurst.wiremock.client.WireMock.getRequestedFor;
import static com.github.tomakehurst.wiremock.client.WireMock.post;
import static com.github.tomakehurst.wiremock.client.WireMock.urlEqualTo;
import static com.github.tomakehurst.wiremock.client.WireMock.verify;
import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.options;
import static org.junit.Assert.*;
public class TorExitNodeManagerTest extends AbstractRedisClusterTest {
private ScheduledExecutorService scheduledExecutorService;
private ExecutorService clientExecutorService;
private TorExitNodeManager torExitNodeManager;
private static final String LIST_PATH = "/list";
@Rule
public WireMockRule wireMockRule = new WireMockRule(options().dynamicPort().dynamicHttpsPort());
@Before
public void setUp() throws Exception {
final TorExitNodeConfiguration configuration = new TorExitNodeConfiguration();
configuration.setListUrl("http://localhost:" + wireMockRule.port() + LIST_PATH);
scheduledExecutorService = Executors.newSingleThreadScheduledExecutor();
clientExecutorService = Executors.newSingleThreadExecutor();
torExitNodeManager = new TorExitNodeManager(scheduledExecutorService, clientExecutorService, configuration);
}
@After
public void tearDown() throws Exception {
scheduledExecutorService.shutdown();
scheduledExecutorService.awaitTermination(1, TimeUnit.SECONDS);
clientExecutorService.shutdown();
clientExecutorService.awaitTermination(1, TimeUnit.SECONDS);
}
@Test
public void testIsTorExitNode() {
final TorExitNodeConfiguration configuration = new TorExitNodeConfiguration();
configuration.setS3Region("ap-northeast-3");
final TorExitNodeManager torExitNodeManager =
new TorExitNodeManager(mock(ScheduledExecutorService.class), configuration);
assertFalse(torExitNodeManager.isTorExitNode("10.0.0.1"));
assertFalse(torExitNodeManager.isTorExitNode("10.0.0.2"));
final String etag = UUID.randomUUID().toString();
wireMockRule.stubFor(get(urlEqualTo(LIST_PATH))
.willReturn(aResponse()
.withHeader("ETag", etag)
.withBody("10.0.0.1\n10.0.0.2")));
torExitNodeManager.refresh().join();
verify(getRequestedFor(urlEqualTo(LIST_PATH)).withoutHeader("If-None-Match"));
assertTrue(torExitNodeManager.isTorExitNode("10.0.0.1"));
assertTrue(torExitNodeManager.isTorExitNode("10.0.0.2"));
assertFalse(torExitNodeManager.isTorExitNode("10.0.0.3"));
wireMockRule.stubFor(get(urlEqualTo(LIST_PATH)).withHeader("If-None-Match", equalTo(etag))
.willReturn(aResponse().withStatus(304)));
torExitNodeManager.refresh().join();
verify(getRequestedFor(urlEqualTo(LIST_PATH)).withHeader("If-None-Match", equalTo(etag)));
torExitNodeManager.handleExitListChanged(
new ByteArrayInputStream("10.0.0.1\n10.0.0.2".getBytes(StandardCharsets.UTF_8)));
assertTrue(torExitNodeManager.isTorExitNode("10.0.0.1"));
assertTrue(torExitNodeManager.isTorExitNode("10.0.0.2"));