Continued:

[ctracker.git] / libs / lib_connect.php

diff --git a/libs/lib_connect.php b/libs/lib_connect.php
index 952445486d1b0429e97be130b96ddeb6c5e00daf..2a6a3669721004e60fc3b4dfdc476a56aa4bd30c 100644 (file)
--- a/libs/lib_connect.php
+++ b/libs/lib_connect.php
@@ -97,7 +97,7 @@ function crackerTrackerCloseDatabaseLink () {
 }
 
 // Inserts given array, if IP/check_get combination was not found
-function crackerTrackerInsertArray ($table, $rowData) {
+function crackerTrackerInsertArray ($table, array $rowData) {
        // Is there a link up?
        if (!isCrackerTrackerDatabaseLinkUp()) {
                // Abort silently here
@@ -124,9 +124,9 @@ function crackerTrackerInsertArray ($table, $rowData) {
 }
 
 // Updates a given entry by just counting it up
-function updateCrackerTrackerEntry (array $rowData) {
+function updateCrackerTrackerEntry (array $rowData, $countColumn = 'count') {
        // Construct the SELECT query
-       $sqlString = 'UPDATE `ctracker_data` SET `count`=`count`+1 WHERE (`remote_addr`="' . crackerTrackerEscapeString($rowData['remote_addr']) . '" OR `proxy_addr`="' . crackerTrackerEscapeString($rowData['proxy_addr']) . '") AND `` = "' . crackerTrackerEscapeString($rowData['']) . '" LIMIT 1';
+       $sqlString = 'UPDATE `ctracker_data` SET `' . $countColumn . '`=`' . $countColumn . '`+1 WHERE (`remote_addr`="' . crackerTrackerEscapeString($rowData['remote_addr']) . '" OR `proxy_addr`="' . crackerTrackerEscapeString($rowData['proxy_addr']) . '") LIMIT 1';
 
        // Run the SQL and check if we have one line
        runCrackerTrackerSql($sqlString, __FUNCTION__, __LINE__);