HTML fix

[mailer.git] / ref.php

diff --git a/ref.php b/ref.php
index 451bb7eaa36c123940826f0db573e6b44c6062b6..b308c84198ac346f0e8c761192c84e442634d31f 100644 (file)
--- a/ref.php
+++ b/ref.php
@@ -46,6 +46,12 @@ $GLOBALS['output_mode'] = -1;
 // Load the required file(s)
 require('inc/config-global.php');
 
+// No refid by default
+$URL = 'index.php';
+
+// Invalid request by default
+$ref = 0;
+
 // Base URL for redirection
 switch (getConfig('refid_target')) {
        case 'register':
@@ -55,27 +61,37 @@ switch (getConfig('refid_target')) {
        case 'index':
                $URL = 'index.php?refid=';
                break;
-}
-
-// No refid and we add our refid (don't forget to set $def_refid!)
-$URL = 'index.php';
+} // END - switch
 
 // Get referal ID from ref or refid variable
 if (isGetRequestElementSet('ref'))        $ref = getRequestElement('ref');
  elseif (isGetRequestElementSet('refid')) $ref = getRequestElement('refid');
 
 if (!empty($ref)) {
-       // Invalid request by default
-       $ref = 0;
-
        // Test if nickname or numeric id
        if (isExtensionActive('nickname')) {
                // Nickname in URL, so load the ID
                $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);