moving to an overload function
[ga4php.git] / lib / lib.php
index dc10d56..42bc599 100644 (file)
@@ -1,4 +1,373 @@
 <?php
 
 
+/*
+ * TODO's:
+ * Implement TOTP fully
+ * Error checking, lots of error checking
+ * have a way of encapsulating token data stright into a single field so it could be added
+ *    in some way to a preexisting app without modifying the DB as such... or by just adding
+ *    a single field to a user table...
+ * Remove all reliance on the SQLite database. Data should come from the encasultating application
+ *    which will be expected to provde two function calls where it can get/store data - DONE
+ */
+
+/*
+ * I am an idiot.
+ * For some reason i had it in my head to pass static functions into the construct
+ * of the class in order to hand data going in/out
+ * when in reality i should be making the data in/out func's abstract
+ * and getting implementors to extend the class
+ * 
+ * For now im going to keep implementing it this way and thus my class will
+ * forever be an example of poor design choices. It'll change it very shortly though 
+ */
+
+/*
+ * The way we should really be doing things is to have an array that encapsulates "normal" data (or a class?)
+ * and then just manipulate it, then use a checkin function to push the data base into the db...
+ */
+class GoogleAuthenticator {
+       
+       // first we init google authenticator by passing it a filename
+       // for its sqlite database.
+       // $getDataFunction expects 1 argument which defines what data it wants
+       // and can be "userlist" or "usertoken:username"
+       // $putDataFunciton expects 2 arguments, $1 is data type, $2 is the data
+       // $1 can be "changetoken:username", "removetoken:username", $2 is the token
+       // data in some encoded form
+       // tokenDATA will be like HOTP;KEY;CID where CID is the current counter value
+       // why encode like this? i cant think of a good reason tbh, i should probably just
+       // use php's arry encoder functions  
+       function __construct($getDataFunction, $putDataFunction) {
+               $this->putDataFunction = $putDataFunction;
+               $this->getDataFunction = $getDataFunction;
+       }
+       
+       abstract function getData($username);
+       abstract function putData($username, $data);
+       abstract function getUsers();
+       
+       // a function to create an empty data structure
+       function createEmptyData() {
+               $data["tokenkey"] = ""; // the token key
+               $data["tokentype"] = ""; // the token type
+               $data["tokentimer"] = ""; // the token timer (For totp) and not supported by ga yet             
+               $data["tokencounter"] = ""; // the token counter for hotp
+               $data["tokenalgorithm"] = ""; // the token algorithm (not supported by ga yet)
+               
+               return $data;
+       }
+       
+       // an internal funciton to get 
+       function internalGetData($username) {
+               $data = getData($username);
+               $deco = unserialize(base64_decode($data));
+               
+               if(!$deco) {
+                       $deco = createEmptyData();
+               }
+               
+               return $deco;
+       }
+       
+
+       function internalPutData($username, $data) {
+               $enco = base64_encode(serialize($data));
+               
+               return putData($username, $enco);
+       }
+       
+
+       // set the token type the user it going to use.
+       // this defaults to HOTP - we only do 30s token
+       // so lets not be able to set that yet
+       function setupTokenType($username, $tokentype) {
+               if($tokentype!="HOTP" and $tokentype!="TOTP") {
+                       $errorText = "Invalid Token Type";
+                       return false;
+               }
+               
+               $data = getData($username);
+               $data["tokentype"] = $tokentype;
+               $func = $this->putDataFunction;
+               $func("settokentype", $put);
+               
+               return true;    
+       }
+       
+       
+       // create "user" with insert
+       function setUser($username, $key = "", $ttype="HOTP") {
+               if($key == "") $key = $this->createBase32Key();
+               $hkey = $this->helperb322hex($key);
+               
+               $token["username"] = $username;
+               $token["tokenkey"] = $hkey;
+               $token["tokentype"] = $ttype;
+               
+               $func = $this->putDataFunction;
+               $func("setusertoken", $token);
+               
+               return $key;
+       }
+       
+       
+       // sets the key for a user - this is assuming you dont want
+       // to use one created by the application. returns false
+       // if the key is invalid or the user doesn't exist.
+       function setUserKey($username, $key) {
+               // consider scrapping this
+       }
+       
+       
+       // have user?
+       function userExists($username) {
+               // need to think about this
+       }
+       
+       
+       // self explanitory?
+       function deleteUser($username) {
+               $func = $this->putDataFunction;
+               $func("deleteusertoken", $username);
+       }
+       
+       // user has input their user name and some code, authenticate
+       // it
+       function authenticateUser($username, $code) {
+               
+               $func = $this->getDataFunction;
+               $tokendata = $func("gettoken", $username);
+               
+               // TODO: check return value
+               $ttype = $tokendata["tokentype"];
+               $tlid = $tokendata["tokencounter"];
+               $tkey = $tokendata["tokenkey"];
+               switch($ttype) {
+                       case "HOTP":
+                               $st = $tlid;
+                               $en = $tlid+20;
+                               for($i=$st; $i<$en; $i++) {
+                                       $stest = $this->oath_hotp($tkey, $i);
+                                       //error_log("code: $code, $stest, $tkey, $tid");
+                                       if($code == $stest) {
+                                               $tokenset["username"] = $username;
+                                               $tokenset["tokencounter"] = $i;
+                                               
+                                               $func = $this->putDataFunction;
+                                               $func("settokencounter", $tokenset);
+                                               return true;
+                                       }
+                               }
+                               return false;
+                               break;
+                       case "TOTP":
+                               $t_now = time();
+                               $t_ear = $t_now - 45;
+                               $t_lat = $t_now + 60;
+                               $t_st = ((int)($t_ear/30));
+                               $t_en = ((int)($t_lat/30));
+                               //error_log("kmac: $t_now, $t_ear, $t_lat, $t_st, $t_en");
+                               for($i=$t_st; $i<=$t_en; $i++) {
+                                       $stest = $this->oath_hotp($tkey, $i);
+                                       //error_log("code: $code, $stest, $tkey\n");
+                                       if($code == $stest) {
+                                               return true;
+                                       }
+                               }
+                               break;
+                       default:
+                               echo "how the frig did i end up here?";
+               }
+               
+               return false;
+
+       }
+       
+       // this function allows a user to resync their key. If too
+       // many codes are called, we only check up to 20 codes in the future
+       // so if the user is at 21, they'll always fail. 
+       function resyncCode($username, $code1, $code2) {
+               // here we'll go from 0 all the way thru to 200k.. if we cant find the code, so be it, they'll need a new one
+               // for HOTP tokens we start at x and go to x+20
+               
+               // for TOTP we go +/-1min TODO = remember that +/- 1min should
+               // be changed based on stepping if we change the expiration time
+               // for keys
+               
+               //              $this->dbConnector->query('CREATE TABLE "tokens" ("token_id" INTEGER PRIMARY KEY AUTOINCREMENT,"token_key" TEXT NOT NULL, "token_type" TEXT NOT NULL, "token_lastid" INTEGER NOT NULL)');
+               
+               $func = $this->getDataFunction;
+               $tokendata = $func("gettoken", $username);
+               
+               // TODO: check return value
+               $ttype = $tokendata["tokentype"];
+               $tlid = $tokendata["tokencounter"];
+               $tkey = $tokendata["tokenkey"];
+               
+               switch($ttype) {
+                       case "HOTP":
+                               $st = 0;
+                               $en = 200000;
+                               for($i=$st; $i<$en; $i++) {
+                                       $stest = $this->oath_hotp($tkey, $i);
+                                       //echo "code: $code, $stest, $tkey\n";
+                                       if($code1 == $stest) {
+                                               $stest2 = $this->oath_hotp($tkey, $i+1);
+                                               if($code2 == $stest2) {
+                                                       $tokenset["username"] = $username;
+                                                       $tokenset["tokencounter"] = $i+1;
+                                               
+                                                       $func = $this->putDataFunction;
+                                                       $func("settokencounter", $tokenset);
+                                                       return true;
+                                               }
+                                       }
+                               }
+                               return false;
+                               break;
+                       case "TOTP":
+                               break;
+                       default:
+                               echo "how the frig did i end up here?";
+               }
+               
+               return false;
+               
+       }
+       
+       // gets the error text associated with the last error
+       function getErrorText() {
+               return $this->errorText;
+       }
+       
+       // create a url compatibile with google authenticator.
+       function createURL($user, $key,$toktype = "HOTP") {
+               // oddity in the google authenticator... hotp needs to be lowercase.
+               $toktype = strtolower($toktype);
+               if($toktype == "hotp") {
+                       $url = "otpauth://$toktype/$user?secret=$key&counter=1";
+               } else {
+                       $url = "otpauth://$toktype/$user?secret=$key";
+               }
+               //echo "url: $url\n";
+               return $url;
+       }
+       
+       // creeates a base 32 key (random)
+       function createBase32Key() {
+               $alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567";
+               $key = "";
+               for($i=0; $i<16; $i++) {
+                       $offset = rand(0,strlen($alphabet)-1);
+                       //echo "$i off is $offset\n";
+                       $key .= $alphabet[$offset];
+               }
+               
+               return $key;
+       }
+               
+       
+       function helperb322hex($b32) {
+        $alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567";
+
+        $out = "";
+        $dous = "";
+
+        for($i = 0; $i < strlen($b32); $i++) {
+               $in = strrpos($alphabet, $b32[$i]);
+               $b = str_pad(base_convert($in, 10, 2), 5, "0", STR_PAD_LEFT);
+            $out .= $b;
+            $dous .= $b.".";
+        }
+
+        $ar = str_split($out,20);
+
+        //echo "$dous, $b\n";
+
+        //print_r($ar);
+        $out2 = "";
+        foreach($ar as $val) {
+                $rv = str_pad(base_convert($val, 2, 16), 5, "0", STR_PAD_LEFT);
+                //echo "rv: $rv from $val\n";
+                $out2 .= $rv;
+
+        }
+        //echo "$out2\n";
+
+        return $out2;
+       }
+       
+       function helperhex2b32($hex) {
+        $alphabet = "ABCDEFGHIJKLMNOPQRSTUVWXYZ234567";
+
+        $ar = str_split($hex, 5);
+
+        $out = "";
+        foreach($ar as $var) {
+                $bc = base_convert($var, 16, 2);
+                $bin = str_pad($bc, 20, "0", STR_PAD_LEFT);
+                $out .= $bin;
+                //echo "$bc was, $var is, $bin are\n";
+        }
+
+        $out2 = "";
+        $ar2 = str_split($out, 5);
+        foreach($ar2 as $var2) {
+                $bc = base_convert($var2, 2, 10);
+                $out2 .= $alphabet[$bc];
+        }
+
+        return $out2;
+       }
+       
+       function oath_hotp($key, $counter)
+       {
+               $key = pack("H*", $key);
+           $cur_counter = array(0,0,0,0,0,0,0,0);
+           for($i=7;$i>=0;$i--)
+           {
+               $cur_counter[$i] = pack ('C*', $counter);
+               $counter = $counter >> 8;
+           }
+           $bin_counter = implode($cur_counter);
+           // Pad to 8 chars
+           if (strlen ($bin_counter) < 8)
+           {
+               $bin_counter = str_repeat (chr(0), 8 - strlen ($bin_counter)) . $bin_counter;
+           }
+       
+           // HMAC
+           $hash = hash_hmac ('sha1', $bin_counter, $key);
+           return str_pad($this->oath_truncate($hash), 6, "0", STR_PAD_LEFT);
+       }
+       
+       function oath_truncate($hash, $length = 6)
+       {
+           // Convert to dec
+           foreach(str_split($hash,2) as $hex)
+           {
+               $hmac_result[]=hexdec($hex);
+           }
+       
+           // Find offset
+           $offset = $hmac_result[19] & 0xf;
+       
+           // Algorithm from RFC
+           return
+           (
+               (($hmac_result[$offset+0] & 0x7f) << 24 ) |
+               (($hmac_result[$offset+1] & 0xff) << 16 ) |
+               (($hmac_result[$offset+2] & 0xff) << 8 ) |
+               ($hmac_result[$offset+3] & 0xff)
+           ) % pow(10,$length);
+       }
+       
+       
+       // some private data bits.
+       private $getDatafunction;
+       private $putDatafunction;
+       private $errorText;
+}
 ?>
\ No newline at end of file