123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086 |
- /*
- * Copyright (c) 2006-2015 DMDirc Developers
- *
- * Permission is hereby granted, free of charge, to any person obtaining a copy
- * of this software and associated documentation files (the "Software"), to deal
- * in the Software without restriction, including without limitation the rights
- * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
- * copies of the Software, and to permit persons to whom the Software is
- * furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice shall be included in
- * all copies or substantial portions of the Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
- * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
- * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
- * SOFTWARE.
- */
-
- package com.dmdirc;
-
- import com.dmdirc.commandparser.CommandType;
- import com.dmdirc.config.profiles.Profile;
- import com.dmdirc.events.ServerConnectErrorEvent;
- import com.dmdirc.events.ServerConnectedEvent;
- import com.dmdirc.events.ServerConnectingEvent;
- import com.dmdirc.events.ServerDisconnectedEvent;
- import com.dmdirc.events.ServerReconnectScheduledEvent;
- import com.dmdirc.events.ServerUnknownProtocolEvent;
- import com.dmdirc.interfaces.Connection;
- import com.dmdirc.interfaces.GroupChatManager;
- import com.dmdirc.interfaces.InviteManager;
- import com.dmdirc.interfaces.User;
- import com.dmdirc.interfaces.WindowModel;
- import com.dmdirc.interfaces.config.ConfigChangeListener;
- import com.dmdirc.interfaces.config.ConfigProvider;
- import com.dmdirc.interfaces.config.ConfigProviderMigrator;
- import com.dmdirc.interfaces.config.IdentityFactory;
- import com.dmdirc.parser.common.DefaultStringConverter;
- import com.dmdirc.parser.common.IgnoreList;
- import com.dmdirc.parser.common.ParserError;
- import com.dmdirc.parser.common.ThreadedParser;
- import com.dmdirc.parser.interfaces.ClientInfo;
- import com.dmdirc.parser.interfaces.EncodingParser;
- import com.dmdirc.parser.interfaces.Parser;
- import com.dmdirc.parser.interfaces.ProtocolDescription;
- import com.dmdirc.parser.interfaces.SecureParser;
- import com.dmdirc.parser.interfaces.StringConverter;
- import com.dmdirc.tls.CertificateManager;
- import com.dmdirc.ui.core.components.WindowComponent;
- import com.dmdirc.ui.input.TabCompleterFactory;
- import com.dmdirc.ui.input.TabCompletionType;
- import com.dmdirc.ui.messages.BackBufferFactory;
- import com.dmdirc.ui.messages.ColourManager;
- import com.dmdirc.ui.messages.Formatter;
- import com.dmdirc.ui.messages.HighlightManager;
-
- import java.net.NoRouteToHostException;
- import java.net.SocketException;
- import java.net.SocketTimeoutException;
- import java.net.URI;
- import java.net.URISyntaxException;
- import java.net.UnknownHostException;
- import java.util.ArrayList;
- import java.util.Arrays;
- import java.util.Collection;
- import java.util.Collections;
- import java.util.Date;
- import java.util.Map;
- import java.util.Optional;
- import java.util.concurrent.ConcurrentSkipListMap;
- import java.util.concurrent.ScheduledExecutorService;
- import java.util.concurrent.ScheduledFuture;
- import java.util.concurrent.TimeUnit;
- import java.util.concurrent.locks.ReadWriteLock;
- import java.util.concurrent.locks.ReentrantReadWriteLock;
- import java.util.function.Function;
-
- import javax.annotation.Nonnull;
- import javax.annotation.Nullable;
- import javax.net.ssl.SSLException;
-
- import org.slf4j.Logger;
- import org.slf4j.LoggerFactory;
-
- import static com.dmdirc.util.LogUtils.APP_ERROR;
- import static com.google.common.base.Preconditions.checkNotNull;
-
- /**
- * The Server class represents the client's view of a server. It maintains a list of all channels,
- * queries, etc, and handles parser callbacks pertaining to the server.
- */
- public class Server extends FrameContainer implements Connection {
-
- private static final Logger LOG = LoggerFactory.getLogger(Server.class);
- /** The name of the general domain. */
- private static final String DOMAIN_GENERAL = "general";
-
- /** Manager of group chats. */
- private final GroupChatManagerImpl groupChatManager;
- /** Manager of invites. */
- private final InviteManager inviteManager;
- /** Open query windows on the server. */
- private final Map<String, Query> queries = new ConcurrentSkipListMap<>();
- /** The user manager to retrieve users from. */
- private final UserManager userManager;
-
- /** The Parser instance handling this server. */
- @Nonnull
- private Optional<Parser> parser = Optional.empty();
- /** The Parser instance that used to be handling this server. */
- @Nonnull
- private Optional<Parser> oldParser = Optional.empty();
- /** The parser-supplied protocol description object. */
- @Nonnull
- private Optional<ProtocolDescription> protocolDescription = Optional.empty();
-
- /**
- * Object used to synchronise access to parser. This object should be locked by anything
- * requiring that the parser reference remains the same for a duration of time, or by anything
- * which is updating the parser reference.
- *
- * If used in conjunction with myStateLock, the parserLock must always be locked INSIDE the
- * myStateLock to prevent deadlocks.
- */
- private final ReadWriteLock parserLock = new ReentrantReadWriteLock();
-
- /** Object used to synchronise access to myState. */
- private final Object myStateLock = new Object();
- /** The current state of this server. */
- private final ServerStatus myState = new ServerStatus(this, myStateLock);
-
- /** The address of the server we're connecting to. */
- @Nonnull
- private URI address;
- /** The profile we're using. */
- @Nonnull
- private Profile profile;
-
- /** Our reason for being away, if any. */
- private Optional<String> awayMessage;
- /** Our event handler. */
- private final ServerEventHandler eventHandler;
- /** Our ignore list. */
- private final IgnoreList ignoreList = new IgnoreList();
- /** Our string convertor. */
- private StringConverter converter = new DefaultStringConverter();
- /** ParserFactory we use for creating parsers. */
- private final ParserFactory parserFactory;
- /** Factory to use to create new identities. */
- private final IdentityFactory identityFactory;
- /** The migrator to use to change our config provider. */
- private final ConfigProviderMigrator configMigrator;
- /** Factory to use for creating queries. */
- private final QueryFactory queryFactory;
- /** The config provider to write user settings to. */
- private final ConfigProvider userSettings;
- /** Executor service to use to schedule repeated events. */
- private final ScheduledExecutorService executorService;
- /** The message encoder factory to create a message encoder with. */
- private final MessageEncoderFactory messageEncoderFactory;
- /** The manager to use for highlighting. */
- private final HighlightManager highlightManager;
- /** Listener to use for config changes. */
- private final ConfigChangeListener configListener = (domain, key) -> updateTitle();
- /** The future used when a reconnect timer is scheduled. */
- private ScheduledFuture<?> reconnectTimerFuture;
-
- /**
- * Creates a new server which will connect to the specified URL with the specified profile.
- */
- public Server(
- final ConfigProviderMigrator configMigrator,
- final ParserFactory parserFactory,
- final TabCompleterFactory tabCompleterFactory,
- final IdentityFactory identityFactory,
- final QueryFactory queryFactory,
- final DMDircMBassador eventBus,
- final MessageEncoderFactory messageEncoderFactory,
- final ConfigProvider userSettings,
- final GroupChatManagerImplFactory groupChatManagerFactory,
- final ScheduledExecutorService executorService,
- @Nonnull final URI uri,
- @Nonnull final Profile profile,
- final BackBufferFactory backBufferFactory,
- final UserManager userManager) {
- super(null, "server-disconnected",
- getHost(uri),
- getHost(uri),
- configMigrator.getConfigProvider(),
- backBufferFactory,
- tabCompleterFactory.getTabCompleter(configMigrator.getConfigProvider(),
- CommandType.TYPE_SERVER, CommandType.TYPE_GLOBAL),
- eventBus,
- Arrays.asList(
- WindowComponent.TEXTAREA.getIdentifier(),
- WindowComponent.INPUTFIELD.getIdentifier(),
- WindowComponent.CERTIFICATE_VIEWER.getIdentifier()));
-
- this.parserFactory = parserFactory;
- this.identityFactory = identityFactory;
- this.configMigrator = configMigrator;
- this.queryFactory = queryFactory;
- this.executorService = executorService;
- this.userSettings = userSettings;
- this.messageEncoderFactory = messageEncoderFactory;
- this.userManager = userManager;
- this.groupChatManager = groupChatManagerFactory.create(this);
- this.inviteManager = new InviteManagerImpl(this);
-
- awayMessage = Optional.empty();
- eventHandler = new ServerEventHandler(this, groupChatManager, getEventBus());
-
- this.address = uri;
- this.profile = profile;
- setConnectionDetails(uri, profile);
-
- updateIcon();
-
- getConfigManager().addChangeListener("formatter", "serverName", configListener);
- getConfigManager().addChangeListener("formatter", "serverTitle", configListener);
-
- initBackBuffer();
- this.highlightManager = new HighlightManager(getConfigManager(),
- new ColourManager(getConfigManager()));
- highlightManager.init();
- getEventBus().subscribe(highlightManager);
- }
-
- /**
- * Updates the connection details for this server. If the specified URI does not define a port,
- * the default port from the protocol description will be used.
- *
- * @param uri The new URI that this server should connect to
- * @param profile The profile that this server should use
- */
- private void setConnectionDetails(final URI uri, final Profile profile) {
- this.address = checkNotNull(uri);
- this.protocolDescription = Optional.ofNullable(parserFactory.getDescription(uri));
- this.profile = profile;
-
- if (uri.getPort() == -1) {
- protocolDescription.ifPresent(pd -> {
- try {
- this.address = new URI(uri.getScheme(), uri.getUserInfo(), uri.getHost(),
- pd.getDefaultPort(), uri.getPath(), uri.getQuery(), uri.getFragment());
- } catch (URISyntaxException ex) {
- LOG.warn(APP_ERROR, "Unable to construct URI", ex);
- }
- });
- }
- }
-
- @Override
- public void connect() {
- connect(address, profile);
- }
-
- @Override
- @Precondition({
- "The current parser is null or not connected",
- "The specified profile is not null"
- })
- @SuppressWarnings("fallthrough")
- public void connect(final URI address, final Profile profile) {
- checkNotNull(address);
- checkNotNull(profile);
-
- synchronized (myStateLock) {
- LOG.info("Connecting to {}, current state is {}", address, myState.getState());
-
- switch (myState.getState()) {
- case RECONNECT_WAIT:
- LOG.debug("Cancelling reconnection timer");
- if (reconnectTimerFuture != null) {
- reconnectTimerFuture.cancel(false);
- }
- break;
- case CLOSING:
- // Ignore the connection attempt
- return;
- case CONNECTED:
- case CONNECTING:
- disconnect(getConfigManager().getOption(DOMAIN_GENERAL, "quitmessage"));
- case DISCONNECTING:
- while (!myState.getState().isDisconnected()) {
- try {
- myStateLock.wait();
- } catch (InterruptedException ex) {
- return;
- }
- }
- break;
- default:
- // Do nothing
- break;
- }
-
- final URI connectAddress;
- final Parser newParser;
-
- try {
- parserLock.writeLock().lock();
- if (parser.isPresent()) {
- throw new IllegalArgumentException("Connection attempt while parser "
- + "is still connected.\n\nMy state:" + getState());
- }
-
- configMigrator.migrate(address.getScheme(), "", "", address.getHost());
-
- setConnectionDetails(address, profile);
-
- updateTitle();
- updateIcon();
-
- parser = Optional.ofNullable(buildParser());
-
- if (!parser.isPresent()) {
- getEventBus().publishAsync(
- new ServerUnknownProtocolEvent(this, address.getScheme()));
- return;
- }
-
- newParser = parser.get();
- connectAddress = newParser.getURI();
- } finally {
- parserLock.writeLock().unlock();
- }
-
- myState.transition(ServerState.CONNECTING);
-
- doCallbacks();
-
- updateAwayState(Optional.empty());
- inviteManager.removeInvites();
-
- newParser.connect();
- if (newParser instanceof ThreadedParser) {
- ((ThreadedParser) newParser).getControlThread()
- .setName("Parser - " + connectAddress.getHost());
- }
- }
-
- getEventBus().publish(new ServerConnectingEvent(this, address));
- }
-
- @Override
- public void reconnect(final String reason) {
- synchronized (myStateLock) {
- if (myState.getState() == ServerState.CLOSING) {
- return;
- }
-
- disconnect(reason);
-
- connect(address, profile);
- }
- }
-
- @Override
- public void reconnect() {
- reconnect(getConfigManager().getOption(DOMAIN_GENERAL, "reconnectmessage"));
- }
-
- @Override
- public void disconnect() {
- disconnect(getConfigManager().getOption(DOMAIN_GENERAL, "quitmessage"));
- }
-
- @Override
- public void disconnect(final String reason) {
- synchronized (myStateLock) {
- LOG.info("Disconnecting. Current state: {}", myState.getState());
-
- switch (myState.getState()) {
- case CLOSING:
- case DISCONNECTING:
- case DISCONNECTED:
- case TRANSIENTLY_DISCONNECTED:
- return;
- case RECONNECT_WAIT:
- LOG.debug("Cancelling reconnection timer");
- if (reconnectTimerFuture != null) {
- reconnectTimerFuture.cancel(false);
- }
- break;
- default:
- break;
- }
-
- groupChatManager.handleDisconnect();
-
- try {
- parserLock.readLock().lock();
- if (parser.isPresent()) {
- myState.transition(ServerState.DISCONNECTING);
-
- inviteManager.removeInvites();
- updateIcon();
-
- parser.get().disconnect(reason);
- } else {
- myState.transition(ServerState.DISCONNECTED);
- }
- } finally {
- parserLock.readLock().unlock();
- }
-
- if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "closequeriesonquit")) {
- closeQueries();
- }
- }
- }
-
- /**
- * Schedules a reconnect attempt to be performed after a user-defined delay.
- */
- @Precondition("The server state is transiently disconnected")
- private void doDelayedReconnect() {
- synchronized (myStateLock) {
- LOG.info("Performing delayed reconnect. State: {}", myState.getState());
-
- if (myState.getState() != ServerState.TRANSIENTLY_DISCONNECTED) {
- throw new IllegalStateException("doDelayedReconnect when not "
- + "transiently disconnected\n\nState: " + myState);
- }
-
- final int delay = Math.max(1000,
- getConfigManager().getOptionInt(DOMAIN_GENERAL, "reconnectdelay"));
-
- getEventBus().publishAsync(new ServerReconnectScheduledEvent(this, delay / 1000));
-
- reconnectTimerFuture = executorService.schedule(() -> {
- synchronized (myStateLock) {
- LOG.debug("Reconnect task executing, state: {}", myState.getState());
- if (myState.getState() == ServerState.RECONNECT_WAIT) {
- myState.transition(ServerState.TRANSIENTLY_DISCONNECTED);
- reconnect();
- }
- }
- }, delay, TimeUnit.MILLISECONDS);
-
- LOG.info("Scheduling reconnect task for delay of {}", delay);
-
- myState.transition(ServerState.RECONNECT_WAIT);
- updateIcon();
- }
- }
-
- @Override
- public boolean hasQuery(final String host) {
- return queries.containsKey(converter.toLowerCase(getUser(host).getNickname()));
- }
-
- @Override
- public Optional<User> getLocalUser() {
- return parser.map(Parser::getLocalClient)
- .map(client -> userManager.getUserFromClientInfo(client, this));
- }
-
- @Override
- public User getUser(final String details) {
- return parser.map(p -> p.getClient(details))
- .map(client -> userManager.getUserFromClientInfo(client, this)).get();
- }
-
- @Override
- public Query getQuery(final String host) {
- return getQuery(host, false);
- }
-
- @Override
- public Query getQuery(final String host, final boolean focus) {
- synchronized (myStateLock) {
- if (myState.getState() == ServerState.CLOSING) {
- // Can't open queries while the server is closing
- return null;
- }
- }
-
- final String nick = getUser(host).getNickname();
- final String lnick = converter.toLowerCase(nick);
-
- if (!queries.containsKey(lnick)) {
- final Query newQuery = queryFactory.getQuery(this, getUser(host));
- if (!getState().isDisconnected()) {
- newQuery.reregister();
- }
- getTabCompleter().addEntry(TabCompletionType.QUERY_NICK, nick);
- queries.put(lnick, newQuery);
- }
-
- return queries.get(lnick);
- }
-
- @Override
- public void updateQuery(final Query query, final String oldNick, final String newNick) {
- getTabCompleter().removeEntry(TabCompletionType.QUERY_NICK, oldNick);
- getTabCompleter().addEntry(TabCompletionType.QUERY_NICK, newNick);
-
- queries.put(converter.toLowerCase(newNick), query);
- queries.remove(converter.toLowerCase(oldNick));
- }
-
- @Override
- public Collection<Query> getQueries() {
- return Collections.unmodifiableCollection(queries.values());
- }
-
- @Override
- public void delQuery(final Query query) {
- getTabCompleter().removeEntry(TabCompletionType.QUERY_NICK, query.getNickname());
- queries.remove(converter.toLowerCase(query.getNickname()));
- }
-
- /**
- * Closes all open query windows associated with this server.
- */
- private void closeQueries() {
- new ArrayList<>(queries.values()).forEach(Query::close);
- }
-
- /**
- * Retrieves the host component of the specified URI, or throws a relevant exception if this is
- * not possible.
- *
- * @param uri The URI to be processed
- *
- * @return The URI's host component, as returned by {@link URI#getHost()}.
- *
- * @throws NullPointerException If <code>uri</code> is null
- * @throws IllegalArgumentException If the specified URI has no host
- * @since 0.6.4
- */
- private static String getHost(final URI uri) {
- if (uri.getHost() == null) {
- throw new IllegalArgumentException("URIs must have hosts");
- }
-
- return uri.getHost();
- }
-
- /**
- * Builds an appropriately configured {@link Parser} for this server.
- *
- * @return A configured parser.
- */
- @Nullable
- private Parser buildParser() {
- final Parser myParser = parserFactory.getParser(profile, address, getConfigManager())
- .orElse(null);
-
- if (myParser != null) {
- myParser.setIgnoreList(ignoreList);
- }
-
- if (myParser instanceof SecureParser) {
- final CertificateManager certificateManager =
- new CertificateManager(this, address.getHost(), getConfigManager(),
- userSettings, getEventBus());
- final SecureParser secureParser = (SecureParser) myParser;
- secureParser.setTrustManagers(certificateManager);
- secureParser.setKeyManagers(certificateManager.getKeyManager());
- }
-
- if (myParser instanceof EncodingParser) {
- final EncodingParser encodingParser = (EncodingParser) myParser;
- encodingParser.setEncoder(messageEncoderFactory.getMessageEncoder(this, myParser));
- }
-
- return myParser;
- }
-
- @Override
- public boolean compareURI(final URI uri) {
- return parser.map(p -> p.compareURI(uri)).orElse(
- oldParser.map(op -> op.compareURI(uri)).orElse(false));
- }
-
- /**
- * Updates this server's icon.
- */
- private void updateIcon() {
- final String icon = myState.getState() == ServerState.CONNECTED
- ? protocolDescription.get().isSecure(address)
- ? "secure-server" : "server" : "server-disconnected";
- setIcon(icon);
- }
-
- /**
- * Registers callbacks.
- */
- private void doCallbacks() {
- eventHandler.registerCallbacks();
- queries.values().forEach(Query::reregister);
- }
-
- @Override
- public void sendLine(final String line) {
- synchronized (myStateLock) {
- try {
- parserLock.readLock().lock();
- parser.ifPresent(p -> {
- if (!line.isEmpty() && myState.getState() == ServerState.CONNECTED) {
- p.sendRawMessage(line);
- }
- });
- } finally {
- parserLock.readLock().unlock();
- }
- }
- }
-
- @Override
- public void sendMessage(final String target, final String message) {
- if (!message.isEmpty()) {
- parser.ifPresent(p -> p.sendMessage(target, message));
- }
- }
-
- @Override
- public int getMaxLineLength() {
- return withParserReadLock(Parser::getMaxLength, -1);
- }
-
- @Override
- @Nonnull
- public Optional<Parser> getParser() {
- return parser;
- }
-
- @Nonnull
- @Override
- public Profile getProfile() {
- return profile;
- }
-
- @Override
- public String getAddress() {
- return withParserReadLock(Parser::getServerName, address.getHost());
- }
-
- @Override
- public String getNetwork() {
- try {
- parserLock.readLock().lock();
- return parser.map(p -> p.getNetworkName().isEmpty()
- ? getNetworkFromServerName(p.getServerName()) : p.getNetworkName())
- .orElseThrow(() -> new IllegalStateException(
- "getNetwork called when parser is null (state: " + getState() + ')'));
- } finally {
- parserLock.readLock().unlock();
- }
- }
-
- @Override
- public boolean isNetwork(final String target) {
- synchronized (myStateLock) {
- return withParserReadLock(p -> getNetwork().equalsIgnoreCase(target), false);
- }
- }
-
- /**
- * Calculates a network name from the specified server name. This method implements parts 2-4 of
- * the procedure documented at getNetwork().
- *
- * @param serverName The server name to parse
- *
- * @return A network name for the specified server
- */
- protected static String getNetworkFromServerName(final String serverName) {
- final String[] parts = serverName.split("\\.");
- final String[] tlds = {"biz", "com", "info", "net", "org"};
- boolean isTLD = false;
-
- for (String tld : tlds) {
- if (serverName.endsWith('.' + tld)) {
- isTLD = true;
- break;
- }
- }
-
- if (isTLD && parts.length > 2) {
- return parts[parts.length - 2] + '.' + parts[parts.length - 1];
- } else if (parts.length > 2) {
- final StringBuilder network = new StringBuilder();
-
- for (int i = 1; i < parts.length; i++) {
- if (network.length() > 0) {
- network.append('.');
- }
-
- network.append(parts[i]);
- }
-
- return network.toString();
- } else {
- return serverName;
- }
- }
-
- @Override
- public String getIrcd() {
- return parser.get().getServerSoftwareType();
- }
-
- @Override
- public String getProtocol() {
- return address.getScheme();
- }
-
- @Override
- public boolean isAway() {
- return awayMessage.isPresent();
- }
-
- @Override
- public String getAwayMessage() {
- return awayMessage.orElse(null);
- }
-
- @Override
- public ServerState getState() {
- return myState.getState();
- }
-
- @Override
- public ServerStatus getStatus() {
- return myState;
- }
-
- @Override
- public void close() {
- synchronized (myStateLock) {
- eventHandler.unregisterCallbacks();
- getConfigManager().removeListener(configListener);
- highlightManager.stop();
- getEventBus().unsubscribe(highlightManager);
- executorService.shutdown();
-
- disconnect();
-
- myState.transition(ServerState.CLOSING);
- }
-
- groupChatManager.closeAll();
- closeQueries();
- inviteManager.removeInvites();
-
- super.close();
- }
-
- @Override
- public WindowModel getWindowModel() {
- return this;
- }
-
- @Override
- @Deprecated
- public void addLineToAll(final String messageType, final Date date,
- final Object... args) {
- groupChatManager.addLineToAll(messageType, date, args);
-
- for (Query query : queries.values()) {
- query.addLine(messageType, date, args);
- }
-
- addLine(messageType, date, args);
- }
-
- @Override
- public void sendCTCPReply(final String source, final String type, final String args) {
- if ("VERSION".equalsIgnoreCase(type)) {
- parser.get().sendCTCPReply(source, "VERSION",
- "DMDirc " + getConfigManager().getOption("version", "version") +
- " - https://www.dmdirc.com/");
- } else if ("PING".equalsIgnoreCase(type)) {
- parser.get().sendCTCPReply(source, "PING", args);
- } else if ("CLIENTINFO".equalsIgnoreCase(type)) {
- parser.get().sendCTCPReply(source, "CLIENTINFO", "VERSION PING CLIENTINFO");
- }
- }
-
- @Override
- public Optional<Connection> getConnection() {
- return Optional.of(this);
- }
-
- @Override
- public void updateTitle() {
- synchronized (myStateLock) {
- if (myState.getState() == ServerState.CLOSING) {
- return;
- }
-
- try {
- parserLock.readLock().lock();
- final Object[] arguments = {
- address.getHost(), parser.map(Parser::getServerName).orElse("Unknown"),
- address.getPort(), parser.map(p -> getNetwork()).orElse("Unknown"),
- getLocalUser().map(User::getNickname).orElse("Unknown")
- };
-
- setName(Formatter.formatMessage(getConfigManager(),
- "serverName", arguments));
- setTitle(Formatter.formatMessage(getConfigManager(),
- "serverTitle", arguments));
- } finally {
- parserLock.readLock().unlock();
- }
- }
- }
-
- /**
- * Called when the socket has been closed.
- */
- public void onSocketClosed() {
- LOG.info("Received socket closed event, state: {}", myState.getState());
-
- if (Thread.holdsLock(myStateLock)) {
- LOG.info("State lock contended: rerunning on a new thread");
-
- executorService.schedule(this::onSocketClosed, 0, TimeUnit.SECONDS);
- return;
- }
-
- getEventBus().publish(new ServerDisconnectedEvent(this));
-
- eventHandler.unregisterCallbacks();
-
- synchronized (myStateLock) {
- if (myState.getState() == ServerState.CLOSING
- || myState.getState() == ServerState.DISCONNECTED) {
- // This has been triggered via .disconnect()
- return;
- }
-
- if (myState.getState() == ServerState.DISCONNECTING) {
- myState.transition(ServerState.DISCONNECTED);
- } else {
- myState.transition(ServerState.TRANSIENTLY_DISCONNECTED);
- }
-
- groupChatManager.handleSocketClosed();
-
- try {
- parserLock.writeLock().lock();
- oldParser = parser;
- parser = Optional.empty();
- } finally {
- parserLock.writeLock().unlock();
- }
-
- updateIcon();
-
- if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "closequeriesondisconnect")) {
- closeQueries();
- }
-
- inviteManager.removeInvites();
- updateAwayState(Optional.empty());
-
- if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "reconnectondisconnect")
- && myState.getState() == ServerState.TRANSIENTLY_DISCONNECTED) {
- doDelayedReconnect();
- }
- }
- }
-
- /**
- * Called when an error was encountered while connecting.
- *
- * @param errorInfo The parser's error information
- */
- @Precondition("The current server state is CONNECTING")
- public void onConnectError(final ParserError errorInfo) {
- synchronized (myStateLock) {
- LOG.info("Received connect error event, state: {}; error: {}", myState.getState(),
- errorInfo);
-
- if (myState.getState() == ServerState.CLOSING
- || myState.getState() == ServerState.DISCONNECTING) {
- // Do nothing
- return;
- } else if (myState.getState() != ServerState.CONNECTING) {
- // Shouldn't happen
- throw new IllegalStateException("Connect error when not "
- + "connecting\n\n" + getStatus().getTransitionHistory());
- }
-
- myState.transition(ServerState.TRANSIENTLY_DISCONNECTED);
-
- try {
- parserLock.writeLock().lock();
- oldParser = parser;
- parser = Optional.empty();
- } finally {
- parserLock.writeLock().unlock();
- }
-
- updateIcon();
-
- getEventBus().publish(new ServerConnectErrorEvent(this, getErrorDescription
- (errorInfo)));
-
- if (getConfigManager().getOptionBool(DOMAIN_GENERAL, "reconnectonconnectfailure")) {
- doDelayedReconnect();
- }
- }
- }
-
- /**
- * Gets a user-readable description of the specified error.
- *
- * @param errorInfo The parser error to get a description for.
- * @return A user-readable error description.
- */
- private static String getErrorDescription(final ParserError errorInfo) {
- final String description;
- if (errorInfo.getException() == null) {
- description = errorInfo.getData();
- } else {
- final Exception exception = errorInfo.getException();
-
- if (exception instanceof UnknownHostException) {
- description = "Unknown host (unable to resolve)";
- } else if (exception instanceof NoRouteToHostException) {
- description = "No route to host";
- } else if (exception instanceof SocketTimeoutException) {
- description = "Connection attempt timed out";
- } else if (exception instanceof SocketException
- || exception instanceof SSLException) {
- description = exception.getMessage();
- } else {
- LOG.info(APP_ERROR, "Unknown socket error: {}",
- exception.getClass().getCanonicalName(), exception);
- description = "Unknown error: " + exception.getMessage();
- }
- }
- return description;
- }
-
- /**
- * Called after the parser receives the 005 headers from the server.
- */
- @Precondition("State is CONNECTING")
- public void onPost005() {
- synchronized (myStateLock) {
- if (myState.getState() != ServerState.CONNECTING) {
- // Shouldn't happen
- throw new IllegalStateException("Received onPost005 while not "
- + "connecting\n\n" + myState.getTransitionHistory());
- }
-
- myState.transition(ServerState.CONNECTED);
-
- configMigrator.migrate(address.getScheme(),
- parser.get().getServerSoftwareType(), getNetwork(), parser.get().getServerName());
-
- updateIcon();
- updateTitle();
- updateIgnoreList();
-
- converter = parser.get().getStringConverter();
- groupChatManager.handleConnected();
- }
-
- getEventBus().publish(new ServerConnectedEvent(this));
- }
-
- @Override
- public IgnoreList getIgnoreList() {
- return ignoreList;
- }
-
- @Override
- public void updateIgnoreList() {
- ignoreList.clear();
- ignoreList.addAll(getConfigManager().getOptionList("network", "ignorelist"));
- }
-
- @Override
- public void saveIgnoreList() {
- getNetworkIdentity().setOption("network", "ignorelist", ignoreList.getRegexList());
- }
-
- @Override
- public ConfigProvider getServerIdentity() {
- return identityFactory.createServerConfig(parser.get().getServerName());
- }
-
- @Override
- public ConfigProvider getNetworkIdentity() {
- return identityFactory.createNetworkConfig(getNetwork());
- }
-
- @Override
- public void updateAwayState(final Optional<String> message) {
- checkNotNull(message);
- if (awayMessage.equals(message)) {
- return;
- }
-
- awayMessage = message;
- }
-
- @Override
- public String getUserModes() {
- return getParser().map(Parser::getChannelUserModes).orElse("");
- }
-
- @Override
- public String getBooleanModes() {
- return getParser().map(Parser::getBooleanChannelModes).orElse("");
- }
-
- @Override
- public String getListModes() {
- return getParser().map(Parser::getListChannelModes).orElse("");
- }
-
- @Override
- public String getParameterModes() {
- return getParser().map(Parser::getParameterChannelModes).orElse("");
- }
-
- @Override
- public String getDoubleParameterModes() {
- return getParser().map(Parser::getDoubleParameterChannelModes).orElse("");
- }
-
- @Override
- public int getMaxListModes(final char mode) {
- return getParser().map(p -> p.getMaxListModes(mode)).orElse(-1);
- }
-
- @Override
- public GroupChatManager getGroupChatManager() {
- return groupChatManager;
- }
-
- @Override
- public InviteManager getInviteManager() {
- return inviteManager;
- }
-
- @Override
- public void setNickname(final String nickname) {
- parser.map(Parser::getLocalClient).ifPresent(c -> c.setNickname(nickname));
- }
-
- @Override
- public Optional<String> getNickname() {
- return parser.map(Parser::getLocalClient).map(ClientInfo::getNickname);
- }
-
- @Override
- public void requestUserInfo(final User user) {
- parser.ifPresent(p -> p.sendWhois(user.getNickname()));
- }
-
- /**
- * Utility method to get a result from the parser while holding the {@link #parserLock}
- * read lock.
- *
- * @param func The function to use to retrieve information from the parser.
- * @param orElse The value to return if the parser is otherwise not present.
- * @param <T> The type of result returned.
- * @return The value returned by {@code func}, if the parser is present, otherwise the
- * {@code orElse} value.
- */
- private <T> T withParserReadLock(
- final Function<? super Parser, ? extends T> func,
- final T orElse) {
- try {
- parserLock.readLock().lock();
- return parser.map(func).orElse(orElse);
- } finally {
- parserLock.readLock().unlock();
- }
- }
-
- }
|