File tree 1 file changed +5
-2
lines changed
1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change @@ -16,6 +16,8 @@ pub enum Error {
16
16
ResponseErr ( String ) ,
17
17
#[ fail( display = "Error parsing URL protocol: {}" , _0) ]
18
18
ProtocolErr ( & ' static str ) ,
19
+ #[ fail( display = "Error altering URL scheme" ) ]
20
+ SchemeErr ( ) ,
19
21
#[ fail( display = "Error parsing URL: {}" , _0) ]
20
22
UrlErr ( url:: ParseError ) ,
21
23
#[ fail( display = "Error connecting to WebSocket host: {}" , _0) ]
@@ -144,8 +146,9 @@ impl NodeClient<'_> {
144
146
_ => Err ( Error :: ProtocolErr ( "Unexpected protocol" ) ) ,
145
147
} ?;
146
148
147
- // The scheme has already been sanitized so this should always succeed
148
- url. set_scheme ( scheme) . expect ( "Could not alter URL scheme" ) ;
149
+ if url. set_scheme ( scheme) . is_err ( ) {
150
+ return Err ( Error :: SchemeErr ( ) ) ;
151
+ } ;
149
152
150
153
let mut request: Request = url. into ( ) ;
151
154
request. add_header (
You can’t perform that action at this time.
0 commit comments