X-Git-Url: http://git.pjr.cc/?p=ga4php.git;a=blobdiff_plain;f=lib%2Fga4php.php;h=22a7b2c56a856ce7e91ee3a798c413e1c77dbd00;hp=3435d03c812c1d5dec3e6df299f34ac3d4499244;hb=HEAD;hpb=1065be060c1b237c4fcdfdef6e3e2ee663d1ca33

diff --git a/lib/ga4php.php b/lib/ga4php.php
index 3435d03..22a7b2c 100644
--- a/lib/ga4php.php
+++ b/lib/ga4php.php
@@ -37,6 +37,20 @@ abstract class GoogleAuthenticator {
 		return $data;
 	}
 	
+	// custom data field manipulation bits
+	function setCustomData($username, $data) {
+		$data = $this->internalGetData($username);
+		$data["user"] = $key;
+		$this->internalPutData($username, $data);
+	}
+	
+	function getCustomData($username) {
+		$data = $this->internalGetData($username);
+		$custom = $data["user"];
+		return $custom;
+		
+	}
+	
 	// an internal funciton to get data from the overloaded functions
 	// and turn them into php arrays.
 	function internalGetData($username) {
@@ -112,7 +126,10 @@ abstract class GoogleAuthenticator {
 		// consider scrapping this
 		$token = $this->internalGetData($username);
 		$token["tokenkey"] = $key;
-		$this->internalPutData($username, $token);		
+		$this->internalPutData($username, $token);
+		
+		// TODO error checking
+		return true;
 	}
 	
 	
@@ -122,6 +139,8 @@ abstract class GoogleAuthenticator {
 		$this->internalPutData($username, "");		
 	}
 	
+
+	
 	// user has input their user name and some code, authenticate
 	// it
 	function authenticateUser($username, $code) {
@@ -151,7 +170,7 @@ abstract class GoogleAuthenticator {
 				$en = $tlid+$this->hotpSkew;
 				for($i=$st; $i<$en; $i++) {
 					$stest = $this->oath_hotp($tkey, $i);
-					error_log("testing code: $code, $stest, $tkey, $tid");
+					//error_log("testing code: $code, $stest, $tkey, $tid");
 					if($code == $stest) {
 						$tokendata["tokencounter"] = $i;
 						$this->internalPutData($username, $tokendata);
@@ -196,7 +215,7 @@ abstract class GoogleAuthenticator {
 		// 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)');
-		$tokendata = internalGetData($username);
+		$tokendata = $this->internalGetData($username);
 		
 		// TODO: check return value
 		$ttype = $tokendata["tokentype"];
@@ -219,7 +238,7 @@ abstract class GoogleAuthenticator {
 						$stest2 = $this->oath_hotp($tkey, $i+1);
 						if($code2 == $stest2) {
 							$tokendata["tokencounter"] = $i+1;
-							internalPutData($username, $tokendata);						
+							$this->internalPutData($username, $tokendata);						
 							return true;
 						}
 					}
@@ -247,7 +266,10 @@ abstract class GoogleAuthenticator {
 		$data = $this->internalGetData($user);
 		$toktype = $data["tokentype"];
 		$key = $this->helperhex2b32($data["tokenkey"]);
-		$counter = $data["tokencounter"];
+
+		// token counter should be one more then current token value, otherwise
+		// it gets confused
+		$counter = $data["tokencounter"]+1;
 		$toktype = strtolower($toktype);
 		if($toktype == "hotp") {
 			$url = "otpauth://$toktype/$user?secret=$key&counter=$counter";