Don't fail on unsupported auth methods

Author: sfackler

src/lib.rs

@@ -19,9 +19,13 @@ use std::rt::io::net::ip::SocketAddr;
 use std::rt::io::net::tcp::TcpStream;
 
 use message::{BackendMessage,
+              AuthenticationOk,
+              AuthenticationKerberosV5,
               AuthenticationCleartextPassword,
               AuthenticationMD5Password,
-              AuthenticationOk,
+              AuthenticationSCMCredential,
+              AuthenticationGSS,
+              AuthenticationSSPI,
               BackendKeyData,
               BindComplete,
               CommandComplete,
@@ -180,7 +184,8 @@ impl PostgresConnection {
         let conn = PostgresConnection {
             stream: Cell::new(stream),
             next_stmt_id: Cell::new(0),
-            notice_handler: Cell::new(~DefaultNoticeHandler as ~PostgresNoticeHandler)
+            notice_handler: Cell::new(~DefaultNoticeHandler
+                                      as ~PostgresNoticeHandler)
         };
 
         args.push((~"client_encoding", ~"UTF8"));
@@ -299,6 +304,10 @@ impl PostgresConnection {
                     password: output.as_slice()
                 });
             }
+            AuthenticationKerberosV5
+            | AuthenticationSCMCredential
+            | AuthenticationGSS
+            | AuthenticationSSPI => return Some(UnsupportedAuthentication),
             _ => fail!()
         }
 

src/message.rs

@@ -12,11 +12,15 @@ pub static PROTOCOL_VERSION: i32 = 0x0003_0000;
 
 #[deriving(ToStr)]
 pub enum BackendMessage {
+    AuthenticationOk,
+    AuthenticationKerberosV5,
     AuthenticationCleartextPassword,
     AuthenticationMD5Password {
         salt: ~[u8]
     },
-    AuthenticationOk,
+    AuthenticationSCMCredential,
+    AuthenticationGSS,
+    AuthenticationSSPI,
     BackendKeyData {
         process_id: i32,
         secret_key: i32
@@ -310,9 +314,13 @@ fn read_data_row(buf: &mut MemReader) -> BackendMessage {
 fn read_auth_message(buf: &mut MemReader) -> BackendMessage {
     match buf.read_be_i32_() {
         0 => AuthenticationOk,
+        2 => AuthenticationKerberosV5,
         3 => AuthenticationCleartextPassword,
         5 => AuthenticationMD5Password { salt: buf.read_bytes(4) },
-        val => fail2!("Unknown Authentication identifier `{}`", val)
+        6 => AuthenticationSCMCredential,
+        7 => AuthenticationGSS,
+        9 => AuthenticationSSPI,
+        val => fail2!("Invalid authentication identifier `{}`", val)
     }
 }