mirror of
https://github.com/signalapp/Signal-Server
synced 2026-04-25 10:08:02 +01:00
181 lines
9.7 KiB
Java
181 lines
9.7 KiB
Java
/**
|
|
* Copyright (C) 2013 Open WhisperSystems
|
|
*
|
|
* This program is free software: you can redistribute it and/or modify
|
|
* it under the terms of the GNU Affero General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU Affero General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
package org.whispersystems.textsecuregcm;
|
|
|
|
import com.google.common.base.Optional;
|
|
import com.yammer.dropwizard.Service;
|
|
import com.yammer.dropwizard.config.Bootstrap;
|
|
import com.yammer.dropwizard.config.Environment;
|
|
import com.yammer.dropwizard.config.HttpConfiguration;
|
|
import com.yammer.dropwizard.db.DatabaseConfiguration;
|
|
import com.yammer.dropwizard.jdbi.DBIFactory;
|
|
import com.yammer.dropwizard.migrations.MigrationsBundle;
|
|
import com.yammer.metrics.reporting.GraphiteReporter;
|
|
import net.spy.memcached.MemcachedClient;
|
|
import org.bouncycastle.jce.provider.BouncyCastleProvider;
|
|
import org.skife.jdbi.v2.DBI;
|
|
import org.whispersystems.textsecuregcm.auth.AccountAuthenticator;
|
|
import org.whispersystems.textsecuregcm.auth.FederatedPeerAuthenticator;
|
|
import org.whispersystems.textsecuregcm.auth.MultiBasicAuthProvider;
|
|
import org.whispersystems.textsecuregcm.configuration.NexmoConfiguration;
|
|
import org.whispersystems.textsecuregcm.controllers.AccountController;
|
|
import org.whispersystems.textsecuregcm.controllers.AttachmentController;
|
|
import org.whispersystems.textsecuregcm.controllers.DeviceController;
|
|
import org.whispersystems.textsecuregcm.controllers.DirectoryController;
|
|
import org.whispersystems.textsecuregcm.controllers.FederationController;
|
|
import org.whispersystems.textsecuregcm.controllers.KeysController;
|
|
import org.whispersystems.textsecuregcm.controllers.MessageController;
|
|
import org.whispersystems.textsecuregcm.controllers.WebsocketControllerFactory;
|
|
import org.whispersystems.textsecuregcm.federation.FederatedClientManager;
|
|
import org.whispersystems.textsecuregcm.federation.FederatedPeer;
|
|
import org.whispersystems.textsecuregcm.limits.RateLimiters;
|
|
import org.whispersystems.textsecuregcm.mappers.IOExceptionMapper;
|
|
import org.whispersystems.textsecuregcm.mappers.RateLimitExceededExceptionMapper;
|
|
import org.whispersystems.textsecuregcm.providers.MemcacheHealthCheck;
|
|
import org.whispersystems.textsecuregcm.providers.MemcachedClientFactory;
|
|
import org.whispersystems.textsecuregcm.providers.RedisClientFactory;
|
|
import org.whispersystems.textsecuregcm.providers.RedisHealthCheck;
|
|
import org.whispersystems.textsecuregcm.push.PushSender;
|
|
import org.whispersystems.textsecuregcm.sms.NexmoSmsSender;
|
|
import org.whispersystems.textsecuregcm.sms.SmsSender;
|
|
import org.whispersystems.textsecuregcm.sms.TwilioSmsSender;
|
|
import org.whispersystems.textsecuregcm.storage.Accounts;
|
|
import org.whispersystems.textsecuregcm.storage.AccountsManager;
|
|
import org.whispersystems.textsecuregcm.storage.Device;
|
|
import org.whispersystems.textsecuregcm.storage.DirectoryManager;
|
|
import org.whispersystems.textsecuregcm.storage.Keys;
|
|
import org.whispersystems.textsecuregcm.storage.PendingAccounts;
|
|
import org.whispersystems.textsecuregcm.storage.PendingAccountsManager;
|
|
import org.whispersystems.textsecuregcm.storage.PendingDevices;
|
|
import org.whispersystems.textsecuregcm.storage.PendingDevicesManager;
|
|
import org.whispersystems.textsecuregcm.storage.PubSubManager;
|
|
import org.whispersystems.textsecuregcm.storage.StoredMessageManager;
|
|
import org.whispersystems.textsecuregcm.storage.StoredMessages;
|
|
import org.whispersystems.textsecuregcm.util.CORSHeaderFilter;
|
|
import org.whispersystems.textsecuregcm.util.UrlSigner;
|
|
import org.whispersystems.textsecuregcm.workers.DirectoryCommand;
|
|
|
|
import java.security.Security;
|
|
import java.util.concurrent.TimeUnit;
|
|
|
|
import redis.clients.jedis.JedisPool;
|
|
|
|
public class WhisperServerService extends Service<WhisperServerConfiguration> {
|
|
|
|
static {
|
|
Security.addProvider(new BouncyCastleProvider());
|
|
}
|
|
|
|
@Override
|
|
public void initialize(Bootstrap<WhisperServerConfiguration> bootstrap) {
|
|
bootstrap.setName("whisper-server");
|
|
bootstrap.addCommand(new DirectoryCommand());
|
|
bootstrap.addBundle(new MigrationsBundle<WhisperServerConfiguration>() {
|
|
@Override
|
|
public DatabaseConfiguration getDatabaseConfiguration(WhisperServerConfiguration configuration) {
|
|
return configuration.getDatabaseConfiguration();
|
|
}
|
|
});
|
|
}
|
|
|
|
@Override
|
|
public void run(WhisperServerConfiguration config, Environment environment)
|
|
throws Exception
|
|
{
|
|
config.getHttpConfiguration().setConnectorType(HttpConfiguration.ConnectorType.NONBLOCKING);
|
|
|
|
DBIFactory dbiFactory = new DBIFactory();
|
|
DBI jdbi = dbiFactory.build(environment, config.getDatabaseConfiguration(), "postgresql");
|
|
|
|
Accounts accounts = jdbi.onDemand(Accounts.class);
|
|
PendingAccounts pendingAccounts = jdbi.onDemand(PendingAccounts.class);
|
|
PendingDevices pendingDevices = jdbi.onDemand(PendingDevices.class);
|
|
Keys keys = jdbi.onDemand(Keys.class);
|
|
StoredMessages storedMessages = jdbi.onDemand(StoredMessages.class );
|
|
|
|
MemcachedClient memcachedClient = new MemcachedClientFactory(config.getMemcacheConfiguration()).getClient();
|
|
JedisPool redisClient = new RedisClientFactory(config.getRedisConfiguration()).getRedisClientPool();
|
|
|
|
DirectoryManager directory = new DirectoryManager(redisClient);
|
|
PendingAccountsManager pendingAccountsManager = new PendingAccountsManager(pendingAccounts, memcachedClient);
|
|
PendingDevicesManager pendingDevicesManager = new PendingDevicesManager(pendingDevices, memcachedClient);
|
|
AccountsManager accountsManager = new AccountsManager(accounts, directory, memcachedClient);
|
|
FederatedClientManager federatedClientManager = new FederatedClientManager(config.getFederationConfiguration());
|
|
PubSubManager pubSubManager = new PubSubManager(redisClient);
|
|
StoredMessageManager storedMessageManager = new StoredMessageManager(storedMessages, pubSubManager);
|
|
|
|
AccountAuthenticator deviceAuthenticator = new AccountAuthenticator(accountsManager);
|
|
RateLimiters rateLimiters = new RateLimiters(config.getLimitsConfiguration(), memcachedClient);
|
|
|
|
TwilioSmsSender twilioSmsSender = new TwilioSmsSender(config.getTwilioConfiguration());
|
|
Optional<NexmoSmsSender> nexmoSmsSender = initializeNexmoSmsSender(config.getNexmoConfiguration());
|
|
SmsSender smsSender = new SmsSender(twilioSmsSender, nexmoSmsSender, config.getTwilioConfiguration().isInternational());
|
|
UrlSigner urlSigner = new UrlSigner(config.getS3Configuration());
|
|
PushSender pushSender = new PushSender(config.getGcmConfiguration(),
|
|
config.getApnConfiguration(),
|
|
storedMessageManager,
|
|
accountsManager);
|
|
|
|
AttachmentController attachmentController = new AttachmentController(rateLimiters, federatedClientManager, urlSigner);
|
|
KeysController keysController = new KeysController(rateLimiters, keys, federatedClientManager);
|
|
MessageController messageController = new MessageController(rateLimiters, pushSender, accountsManager, federatedClientManager);
|
|
|
|
environment.addProvider(new MultiBasicAuthProvider<>(new FederatedPeerAuthenticator(config.getFederationConfiguration()),
|
|
FederatedPeer.class,
|
|
deviceAuthenticator,
|
|
Device.class, "WhisperServer"));
|
|
|
|
environment.addResource(new AccountController(pendingAccountsManager, accountsManager, rateLimiters, smsSender));
|
|
environment.addResource(new DeviceController(pendingDevicesManager, accountsManager, rateLimiters));
|
|
environment.addResource(new DirectoryController(rateLimiters, directory));
|
|
environment.addResource(new FederationController(accountsManager, attachmentController, keysController, messageController));
|
|
environment.addResource(attachmentController);
|
|
environment.addResource(keysController);
|
|
environment.addResource(messageController);
|
|
|
|
environment.addServlet(new WebsocketControllerFactory(deviceAuthenticator, storedMessageManager, pubSubManager),
|
|
"/v1/websocket/");
|
|
|
|
environment.addHealthCheck(new RedisHealthCheck(redisClient));
|
|
environment.addHealthCheck(new MemcacheHealthCheck(memcachedClient));
|
|
|
|
environment.addProvider(new IOExceptionMapper());
|
|
environment.addProvider(new RateLimitExceededExceptionMapper());
|
|
|
|
environment.addFilter(new CORSHeaderFilter(), "/*");
|
|
|
|
if (config.getGraphiteConfiguration().isEnabled()) {
|
|
GraphiteReporter.enable(15, TimeUnit.SECONDS,
|
|
config.getGraphiteConfiguration().getHost(),
|
|
config.getGraphiteConfiguration().getPort());
|
|
}
|
|
}
|
|
|
|
private Optional<NexmoSmsSender> initializeNexmoSmsSender(NexmoConfiguration configuration) {
|
|
if (configuration == null) {
|
|
return Optional.absent();
|
|
} else {
|
|
return Optional.of(new NexmoSmsSender(configuration));
|
|
}
|
|
}
|
|
|
|
public static void main(String[] args) throws Exception {
|
|
new WhisperServerService().run(args);
|
|
}
|
|
|
|
}
|