X-Git-Url: https://git.mxchange.org/?p=mailer.git;a=blobdiff_plain;f=ref.php;h=b308c84198ac346f0e8c761192c84e442634d31f;hp=451bb7eaa36c123940826f0db573e6b44c6062b6;hb=1ebe6e5a94096e8a88e28b31e83a16d13679e743;hpb=64c8349613addc3da2242c5cd6b99d64e3fb5f8e

diff --git a/ref.php b/ref.php
index 451bb7eaa3..b308c84198 100644
--- 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);