@@ -27,10 +27,12 @@
 @class IRCUser;
 
 @protocol IRCConnectionDelegate <OFObject>
 @optional
 - (void)connection: (IRCConnection*)connection
+   didCreateSocket: (OF_KINDOF(OFTCPSocket)*)socket;
+- (void)connection: (IRCConnection*)connection
     didReceiveLine: (OFString*)line;
 - (void)connection: (IRCConnection*)connection
        didSendLine: (OFString*)line;
 - (void)connectionWasEstablished: (IRCConnection*)connection;
 - (void)connection: (IRCConnection*)connection
@@ -70,18 +72,20 @@
 - (void)connectionWasClosed: (IRCConnection*)connection;
 @end
 
 @interface IRCConnection: OFObject
 {
-	OFTCPSocket *_socket;
+	Class _socketClass;
+	OF_KINDOF(OFTCPSocket) *_socket;
 	OFString *_server;
 	uint16_t _port;
 	OFString *_nickname, *_username, *_realname;
 	OFMutableDictionary *_channels;
 	id <IRCConnectionDelegate> _delegate;
 }
 
+@property (assign) Class socketClass;
 @property (copy) OFString *server;
 @property (assign) uint16_t port;
 @property (copy) OFString *nickname, *username, *realname;
 @property (assign) id <IRCConnectionDelegate> delegate;
 @property (readonly, retain) OFTCPSocket *socket;