Nickname active, but id provided should result in valid refid

[mailer.git] / ref.php

diff --git a/ref.php b/ref.php
index 95e6400a434320925e45b232f6372166ff0f4954..b308c84198ac346f0e8c761192c84e442634d31f 100644 (file)
--- a/ref.php
+++ b/ref.php
@@ -74,8 +74,24 @@ if (!empty($ref)) {
                $result = SQL_QUERY_ESC("SELECT `userid` FROM `{?_MYSQL_PREFIX?}_user_data` WHERE `nickname`='%s' LIMIT 1",
                        array($ref), __FILE__, __LINE__);
 
-               // Load userid
-               list($ref) = SQL_FETCHROW($result);
+               // Do we have an entry?
+               if (SQL_NUMROWS($result) == 1) {
+                       // Load userid
+                       list($ref) = SQL_FETCHROW($result);
+               } else {
+                       // So do we have a userid?
+                       $result2 = SQL_QUERY_ESC("SELECT `userid` FROM `{?_MYSQL_PREFIX?}_user_data` WHERE `userid`='%s' LIMIT 1",
+                               array($ref), __FILE__, __LINE__);
+
+                       // Do we have an entry?
+                       if (SQL_NUMROWS($result2) == 0) {
+                               // No entry, so no referal id
+                               $ref = 0;
+                       } // END - if
+
+                       // Free result
+                       SQL_FREERESULT($result2);
+               }
 
                // Free result
                SQL_FREERESULT($result);