@@ -4,8 +4,7 @@ use std::io::net::tcp;
44use std:: io:: net:: pipe;
55use std:: io:: { Stream , IoResult } ;
66
7- use { ConnectParams , SslMode , NoSsl , PreferSsl , RequireSsl , ConnectTarget } ;
8- use error:: { ConnectError , NoSslSupport , SslError } ;
7+ use { ConnectParams , SslMode , ConnectTarget , ConnectError } ;
98use message;
109use message:: { SslRequest , WriteMessage } ;
1110
@@ -90,24 +89,24 @@ pub fn initialize_stream(params: &ConnectParams, ssl: &SslMode)
9089 let mut socket = try!( open_socket ( params) ) ;
9190
9291 let ( ssl_required, ctx) = match * ssl {
93- NoSsl => return Ok ( MaybeSslStream :: Normal ( socket) ) ,
94- PreferSsl ( ref ctx) => ( false , ctx) ,
95- RequireSsl ( ref ctx) => ( true , ctx)
92+ SslMode :: None => return Ok ( MaybeSslStream :: Normal ( socket) ) ,
93+ SslMode :: Prefer ( ref ctx) => ( false , ctx) ,
94+ SslMode :: Require ( ref ctx) => ( true , ctx)
9695 } ;
9796
9897 try!( socket. write_message ( & SslRequest { code : message:: SSL_CODE } ) ) ;
9998 try!( socket. flush ( ) ) ;
10099
101100 if try!( socket. read_u8 ( ) ) == 'N' as u8 {
102101 if ssl_required {
103- return Err ( NoSslSupport ) ;
102+ return Err ( ConnectError :: NoSslSupport ) ;
104103 } else {
105104 return Ok ( MaybeSslStream :: Normal ( socket) ) ;
106105 }
107106 }
108107
109108 match ssl:: SslStream :: new ( ctx, socket) {
110109 Ok ( stream) => Ok ( MaybeSslStream :: Ssl ( stream) ) ,
111- Err ( err) => Err ( SslError ( err) )
110+ Err ( err) => Err ( ConnectError :: SslError ( err) )
112111 }
113112}
0 commit comments