[Resolvido] Loop até validar código

Galera, preciso gerar um código aleatório e grava-lo na tabela, mas antes, preciso saber se esse código já existe. Se existe, gera outro e faz a verificação de novo, até que seja gerado um código único.

Meu código está assim:

require('codigo.php');

$codigo = codigo(6, true, true, true);

// Verifica se o código já existe

$codigoSql = mysql_query("SELECT COUNT(*) AS total FROM `codigo` WHERE `codigo`='$codigo'");
$codigoTotal = mysql_result($codigoSql, 0, 'total');
if ( $codigoTotal == 0 ) {
 mysql_query("INSERT INTO `codigo` VALUES (NULL, '$codigo')");

 $cupom = codigo(6, true, true, true);
 mysql_query("INSERT INTO `codigo` VALUES (NULL, '$codigo')");
}

Porém, se já existir o código, no else ele não verifica se o novo código já existe.