includes/functions.php

   1 <?php\r
   2 /*\r
   3  * YOURLS\r
   4  * Function library\r
   5  */\r
   6 \r
   7 // Determine the allowed character set in short URLs\r
   8 function yourls_get_shorturl_charset() {\r
   9         static $charset = null;\r
  10         if( $charset !== null )\r
  11                 return $charset;\r
  12                 \r
  13         if( !defined('YOURLS_URL_CONVERT') ) {\r
  14                 $charset = '0123456789abcdefghijklmnopqrstuvwxyz-';\r
  15         } else {\r
  16                 switch( YOURLS_URL_CONVERT ) {\r
  17                         case 36:\r
  18                                 $charset = '0123456789abcdefghijklmnopqrstuvwxyz-';\r
  19                                 break;\r
  20                         case 62:\r
  21                         case 64: // just because some people get this wrong in their config.php\r
  22                                 $charset = '0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ-';\r
  23                                 break;\r
  24                 }\r
  25         }\r
  26         \r
  27         $charset = yourls_apply_filter( 'get_shorturl_charset', $charset );\r
  28         return $charset;\r
  29 }\r
  30  \r
  31 // function to convert an integer (1337) to a string (3jk).\r
  32 function yourls_int2string( $num, $chars = null ) {\r
  33         if( $chars == null )\r
  34                 $chars = yourls_get_shorturl_charset();\r
  35         $string = '';\r
  36         $len = strlen( $chars );\r
  37         while( $num >= $len ) {\r
  38                 $mod = bcmod( $num, $len );\r
  39                 $num = bcdiv( $num, $len );\r
  40                 $string = $chars[$mod] . $string;\r
  41         }\r
  42         $string = $chars[$num] . $string;\r
  43         \r
  44         return yourls_apply_filter( 'int2string', $string );\r
  45 }\r
  46 \r
  47 // function to convert a string (3jk) to an integer (1337)\r
  48 function yourls_string2int( $string, $chars = null ) {\r
  49         if( $chars == null )\r
  50                 $chars = yourls_get_shorturl_charset();\r
  51         $integer = 0;\r
  52         $string = strrev( $string  );\r
  53         $baselen = strlen( $chars );\r
  54         $inputlen = strlen( $string );\r
  55         for ($i = 0; $i < $inputlen; $i++) {\r
  56                 $index = strpos( $chars, $string[$i] );\r
  57                 $integer = bcadd( $integer, bcmul( $index, bcpow( $baselen, $i ) ) );\r
  58         }\r
  59         return yourls_apply_filter( 'string2int', $integer );\r
  60         \r
  61 }\r
  62 \r
  63 // Make sure a link keyword (ie "1fv" as in "site.com/1fv") is valid.\r
  64 function yourls_sanitize_string( $string ) {\r
  65         // make a regexp pattern with the shorturl charset, and remove everything but this\r
  66         $pattern = yourls_make_regexp_pattern( yourls_get_shorturl_charset() );\r
  67         $valid = substr(preg_replace('/[^'.$pattern.']/', '', $string ), 0, 199);\r
  68         \r
  69         return yourls_apply_filter( 'sanitize_string', $valid, $string );\r
  70 }\r
  71 \r
  72 // Make an optimized regexp pattern from a string of characters\r
  73 function yourls_make_regexp_pattern( $string ) {\r
  74         $pattern = preg_quote( $string, '-' ); // add - as an escaped characters -- this is fixed in PHP 5.3\r
  75         // TODO: replace char sequences by smart sequences such as 0-9, a-z, A-Z ... ?\r
  76         return $pattern;\r
  77 }\r
  78 \r
  79 // Alias function. I was always getting it wrong.\r
  80 function yourls_sanitize_keyword( $keyword ) {\r
  81         return yourls_sanitize_string( $keyword );\r
  82 }\r
  83 \r
  84 // Is an URL a short URL?\r
  85 function yourls_is_shorturl( $shorturl ) {\r
  86         // TODO: make sure this function evolves with the feature set.\r
  87         // A short URL might be, in the future:\r
  88         // - http://site.com/abc\r
  89         // - http://site.com/abc-bleh\r
  90         // Could allow site.com/abc+ and site.com/abc+all\r
  91         \r
  92         $is_short = false;\r
  93         $keyword = preg_replace( '!^'.YOURLS_SITE.'/!', '', $shorturl ); // accept either 'http://ozh.in/abc' or 'abc'\r
  94         if( $keyword && $keyword == yourls_sanitize_string( $keyword ) && yourls_keyword_is_taken( $keyword ) ) {\r
  95                 $is_short = true;\r
  96         }\r
  97         \r
  98         return yourls_apply_filter( 'is_shorturl', $is_short );\r
  99 }\r
 100 \r
 101 // A few sanity checks on the URL\r
 102 function yourls_sanitize_url($url) {\r
 103         // make sure there's only one 'http://' at the beginning (prevents pasting a URL right after the default 'http://')\r
 104         $url = str_replace('http://http://', 'http://', $url);\r
 105 \r
 106         // make sure there's a protocol, add http:// if not\r
 107         if ( !preg_match('!^([a-zA-Z]+://)!', $url ) )\r
 108                 $url = 'http://'.$url;\r
 109         \r
 110         $url = yourls_clean_url($url);\r
 111         \r
 112         return substr( $url, 0, 1999 );\r
 113 }\r
 114 \r
 115 // Function to filter all invalid characters from a URL. Stolen from WP's clean_url()\r
 116 function yourls_clean_url( $url ) {\r
 117         $url = preg_replace('|[^a-z0-9-~+_.?\[\]\^#=!&;,/:%@$\|*\'"()\\x80-\\xff]|i', '', $url );\r
 118         $strip = array('%0d', '%0a', '%0D', '%0A');\r
 119         $url = yourls_deep_replace($strip, $url);\r
 120         $url = str_replace(';//', '://', $url);\r
 121         $url = str_replace('&amp;', '&', $url); // Revert & not to break query strings\r
 122         \r
 123         return $url;\r
 124 }\r
 125 \r
 126 // Perform a replacement while a string is found, eg $subject = '%0%0%0DDD', $search ='%0D' -> $result =''\r
 127 // Stolen from WP's _deep_replace\r
 128 function yourls_deep_replace($search, $subject){\r
 129         $found = true;\r
 130         while($found) {\r
 131                 $found = false;\r
 132                 foreach( (array) $search as $val ) {\r
 133                         while(strpos($subject, $val) !== false) {\r
 134                                 $found = true;\r
 135                                 $subject = str_replace($val, '', $subject);\r
 136                         }\r
 137                 }\r
 138         }\r
 139         \r
 140         return $subject;\r
 141 }\r
 142 \r
 143 // Make sure an integer is a valid integer (PHP's intval() limits to too small numbers)\r
 144 // TODO FIXME FFS: unused ?\r
 145 function yourls_sanitize_int($in) {\r
 146         return ( substr(preg_replace('/[^0-9]/', '', strval($in) ), 0, 20) );\r
 147 }\r
 148 \r
 149 // Make sure a integer is safe\r
 150 // Note: this is not checking for integers, since integers on 32bits system are way too limited\r
 151 // TODO: find a way to validate as integer\r
 152 function yourls_intval($in) {\r
 153         return yourls_escape($in);\r
 154 }\r
 155 \r
 156 // Escape a string\r
 157 function yourls_escape( $in ) {\r
 158         return mysql_real_escape_string($in);\r
 159 }\r
 160 \r
 161 // Check to see if a given keyword is reserved (ie reserved URL or an existing page)\r
 162 // Returns bool\r
 163 function yourls_keyword_is_reserved( $keyword ) {\r
 164         global $yourls_reserved_URL;\r
 165         $keyword = yourls_sanitize_keyword( $keyword );\r
 166         $reserved = false;\r
 167         \r
 168         if ( in_array( $keyword, $yourls_reserved_URL)\r
 169                 or file_exists( YOURLS_ABSPATH ."/pages/$keyword.php" )\r
 170                 or is_dir( YOURLS_ABSPATH ."/$keyword" )\r
 171         )\r
 172                 $reserved = true;\r
 173         \r
 174         return yourls_apply_filter( 'keyword_is_reserved', $reserved, $keyword );\r
 175 }\r
 176 \r
 177 // Function: Get IP Address. Returns a DB safe string.\r
 178 function yourls_get_IP() {\r
 179         if( !empty( $_SERVER['REMOTE_ADDR'] ) ) {\r
 180                 $ip = $_SERVER['REMOTE_ADDR'];\r
 181         } else {\r
 182                 if(!empty($_SERVER['HTTP_CLIENT_IP'])) {\r
 183                         $ip = $_SERVER['HTTP_CLIENT_IP'];\r
 184                 } else if(!empty($_SERVER['HTTP_X_FORWARDED_FOR'])) {\r
 185                         $ip = $_SERVER['HTTP_X_FORWARDED_FOR'];\r
 186                 } else if(!empty($_SERVER['HTTP_VIA '])) {\r
 187                         $ip = $_SERVER['HTTP_VIA '];\r
 188                 }\r
 189         }\r
 190 \r
 191         return yourls_apply_filter( 'get_IP', yourls_sanitize_ip( $ip ) );\r
 192 }\r
 193 \r
 194 // Sanitize an IP address\r
 195 function yourls_sanitize_ip( $ip ) {\r
 196         return preg_replace( '/[^0-9a-fA-F:., ]/', '', $ip );\r
 197 }\r
 198 \r
 199 // Make sure a date is m(m)/d(d)/yyyy, return false otherwise\r
 200 function yourls_sanitize_date( $date ) {\r
 201         if( !preg_match( '!^\d{1,2}/\d{1,2}/\d{4}$!' , $date ) ) {\r
 202                 return false;\r
 203         }\r
 204         return $date;\r
 205 }\r
 206 \r
 207 // Sanitize a date for SQL search. Return false if malformed input.\r
 208 function yourls_sanitize_date_for_sql( $date ) {\r
 209         if( !yourls_sanitize_date( $date ) )\r
 210                 return false;\r
 211         return date('Y-m-d', strtotime( $date ) );\r
 212 }\r
 213 \r
 214 // Add the "Edit" row\r
 215 function yourls_table_edit_row( $keyword ) {\r
 216         global $ydb;\r
 217         \r
 218         $table = YOURLS_DB_TABLE_URL;\r
 219         $keyword = yourls_sanitize_string( $keyword );\r
 220         $id = yourls_string2int( $keyword ); // used as HTML #id\r
 221         $url = $ydb->get_row("SELECT `url` FROM `$table` WHERE `keyword` = '$keyword';");\r
 222         $safe_url = stripslashes( $url->url );\r
 223         $www = YOURLS_SITE;\r
 224         \r
 225         if( $url ) {\r
 226                 $return = <<<RETURN\r
 227 <tr id="edit-$id" class="edit-row"><td colspan="5"><strong>Original URL</strong>:<input type="text" id="edit-url-$id" name="edit-url-$id" value="$safe_url" class="text" size="70" /> <strong>Short URL</strong>: $www/<input type="text" id="edit-keyword-$id" name="edit-keyword-$id" value="$keyword" class="text" size="10" /></td><td colspan="1"><input type="button" id="edit-submit-$id" name="edit-submit-$id" value="Save" title="Save new values" class="button" onclick="edit_save('$id');" />&nbsp;<input type="button" id="edit-close-$id" name="edit-close-$id" value="X" title="Cancel editing" class="button" onclick="hide_edit('$id');" /><input type="hidden" id="old_keyword_$id" value="$keyword"/></td></tr>\r
 228 RETURN;\r
 229         } else {\r
 230                 $return = '<tr><td colspan="6">Error, URL not found</td></tr>';\r
 231         }\r
 232         \r
 233         $return = yourls_apply_filter( 'table_edit_row', $return, $keyword, $url );\r
 234 \r
 235         return $return;\r
 236 }\r
 237 \r
 238 // Add a link row\r
 239 function yourls_table_add_row( $keyword, $url, $ip, $clicks, $timestamp ) {\r
 240         $keyword = yourls_sanitize_string( $keyword );\r
 241         $id = yourls_string2int( $keyword ); // used as HTML #id\r
 242         $date = date( 'M d, Y H:i', $timestamp+( YOURLS_HOURS_OFFSET * 3600) );\r
 243         $clicks = number_format($clicks, 0, '', '');\r
 244         $shorturl = YOURLS_SITE.'/'.$keyword;\r
 245         $display_url = htmlentities( yourls_trim_long_string( $url ) );\r
 246         $statlink = $shorturl.'+';\r
 247         $url = htmlentities( $url );\r
 248         \r
 249         $actions = <<<ACTION\r
 250 <a href="$statlink" id="statlink-$id" class="button button_stats">&nbsp;&nbsp;&nbsp;</a>&nbsp;<input type="button" id="edit-button-$id" name="edit-button" value="" title="Edit" class="button button_edit" onclick="edit('$id');" />&nbsp;<input type="button" id="delete-button-$id" name="delete-button" value="" title="Delete" class="button button_delete" onclick="remove('$id');" />\r
 251 ACTION;\r
 252         $actions = yourls_apply_filter( 'action_links', $actions, $keyword, $url, $ip, $clicks, $timestamp );\r
 253         \r
 254         $row = <<<ROW\r
 255 <tr id="id-$id"><td id="keyword-$id"><a href="$shorturl">$keyword</a></td><td id="url-$id"><a href="$url" title="$url">$display_url</a></td><td id="timestamp-$id">$date</td><td id="ip-$id">$ip</td><td id="clicks-$id">$clicks</td><td class="actions" id="actions-$id">$actions<input type="hidden" id="keyword_$id" value="$keyword"/></td></tr>\r
 256 ROW;\r
 257         $row = yourls_apply_filter( 'table_add_row', $row, $keyword, $url, $ip, $clicks, $timestamp );\r
 258         \r
 259         return $row;\r
 260 }\r
 261 \r
 262 // Get next id a new link will have if no custom keyword provided\r
 263 function yourls_get_next_decimal() {\r
 264         return yourls_apply_filter( 'get_next_decimal', (int)yourls_get_option( 'next_id' ) );\r
 265 }\r
 266 \r
 267 // Update id for next link with no custom keyword\r
 268 function yourls_update_next_decimal( $int = '' ) {\r
 269         $int = ( $int == '' ) ? yourls_get_next_decimal() + 1 : (int)$int ;\r
 270         $update = yourls_update_option( 'next_id', $int );\r
 271         yourls_do_action( 'update_next_decimal', $int, $update );\r
 272         return $update;\r
 273 }\r
 274 \r
 275 // Delete a link in the DB\r
 276 function yourls_delete_link_by_keyword( $keyword ) {\r
 277         global $ydb;\r
 278 \r
 279         $table = YOURLS_DB_TABLE_URL;\r
 280         $keyword = yourls_sanitize_string( $keyword );\r
 281         $delete = $ydb->query("DELETE FROM `$table` WHERE `keyword` = '$keyword';");\r
 282         yourls_do_action( 'delete_link', $keyword, $delete );\r
 283         return $delete;\r
 284 }\r
 285 \r
 286 // SQL query to insert a new link in the DB. Needs sanitized data. Returns boolean for success or failure of the inserting\r
 287 function yourls_insert_link_in_db($url, $keyword) {\r
 288         global $ydb;\r
 289 \r
 290         $table = YOURLS_DB_TABLE_URL;\r
 291         $timestamp = date('Y-m-d H:i:s');\r
 292         $ip = yourls_get_IP();\r
 293         $insert = $ydb->query("INSERT INTO `$table` VALUES('$keyword', '$url', '$timestamp', '$ip', 0);");\r
 294         \r
 295         yourls_do_action( 'insert_link', (bool)$insert, $url, $keyword, $timestamp, $ip );\r
 296         \r
 297         return (bool)$insert;\r
 298 }\r
 299 \r
 300 // Add a new link in the DB, either with custom keyword, or find one\r
 301 function yourls_add_new_link( $url, $keyword = '' ) {\r
 302         global $ydb;\r
 303 \r
 304         if ( !$url || $url == 'http://' || $url == 'https://' ) {\r
 305                 $return['status'] = 'fail';\r
 306                 $return['code'] = 'error:nourl';\r
 307                 $return['message'] = 'Missing URL input';\r
 308                 $return['errorCode'] = '400';\r
 309                 yourls_do_action( 'add_new_link_fail_nourl' );\r
 310                 return $return;\r
 311         }\r
 312         \r
 313         // Prevent DB flood\r
 314         $ip = yourls_get_IP();\r
 315         yourls_check_IP_flood( $ip );\r
 316         \r
 317         // Prevent internal redirection loops: cannot shorten a shortened URL\r
 318         $url = yourls_escape( yourls_sanitize_url($url) );\r
 319         if( preg_match( '!^'.YOURLS_SITE.'/!', $url ) ) {\r
 320                 if( yourls_is_shorturl( $url ) ) {\r
 321                         $return['status'] = 'fail';\r
 322                         $return['code'] = 'error:noloop';\r
 323                         $return['message'] = 'URL is a short URL';\r
 324                         $return['errorCode'] = '400';\r
 325                         yourls_do_action( 'add_new_link_fail_noloop' );\r
 326                         return $return;\r
 327                 }\r
 328         }\r
 329 \r
 330         yourls_do_action( 'pre_add_new_link', $url, $keyword );\r
 331         \r
 332         $table = YOURLS_DB_TABLE_URL;\r
 333         $strip_url = stripslashes($url);\r
 334         $url_exists = $ydb->get_row("SELECT keyword,url FROM `$table` WHERE `url` = '".$strip_url."';");\r
 335         $return = array();\r
 336 \r
 337         // New URL : store it -- or: URL exists, but duplicates allowed\r
 338         if( !$url_exists || yourls_allow_duplicate_longurls() ) {\r
 339 \r
 340                 // Custom keyword provided\r
 341                 if ( $keyword ) {\r
 342                         $keyword = yourls_escape( yourls_sanitize_string($keyword) );\r
 343                         $keyword = yourls_apply_filter( 'custom_keyword', $keyword );\r
 344                         if ( !yourls_keyword_is_free($keyword) ) {\r
 345                                 // This shorturl either reserved or taken already\r
 346                                 $return['status'] = 'fail';\r
 347                                 $return['code'] = 'error:keyword';\r
 348                                 $return['message'] = 'Short URL '.$keyword.' already exists in database or is reserved';\r
 349                         } else {\r
 350                                 // all clear, store !\r
 351                                 yourls_insert_link_in_db($url, $keyword);\r
 352                                 $return['url'] = array('keyword' => $keyword, 'url' => $strip_url, 'date' => date('Y-m-d H:i:s'), 'ip' => $ip );\r
 353                                 $return['status'] = 'success';\r
 354                                 $return['message'] = $strip_url.' added to database';\r
 355                                 $return['html'] = yourls_table_add_row( $keyword, $url, $ip, 0, time() );\r
 356                                 $return['shorturl'] = YOURLS_SITE .'/'. $keyword;\r
 357                         }\r
 358 \r
 359                 // Create random keyword        \r
 360                 } else {\r
 361                         $timestamp = date('Y-m-d H:i:s');\r
 362                         $id = yourls_get_next_decimal();\r
 363                         $ok = false;\r
 364                         do {\r
 365                                 $keyword = yourls_int2string( $id );\r
 366                                 $keyword = yourls_apply_filter( 'random_keyword', $keyword );\r
 367                                 $free = yourls_keyword_is_free($keyword);\r
 368                                 $add_url = @yourls_insert_link_in_db($url, $keyword);\r
 369                                 $ok = ($free && $add_url);\r
 370                                 if ( $ok === false && $add_url === 1 ) {\r
 371                                         // we stored something, but shouldn't have (ie reserved id)\r
 372                                         $delete = yourls_delete_link_by_keyword( $keyword );\r
 373                                         $return['extra_info'] .= '(deleted '.$keyword.')';\r
 374                                 } else {\r
 375                                         // everything ok, populate needed vars\r
 376                                         $return['url'] = array('keyword' => $keyword, 'url' => $strip_url, 'date' => $timestamp, 'ip' => $ip );\r
 377                                         $return['status'] = 'success';\r
 378                                         $return['message'] = $strip_url.' added to database';\r
 379                                         $return['html'] = yourls_table_add_row( $keyword, $url, $ip, 0, time() );\r
 380                                         $return['shorturl'] = YOURLS_SITE .'/'. $keyword;\r
 381                                 }\r
 382                                 $id++;\r
 383                         } while (!$ok);\r
 384                         @yourls_update_next_decimal($id);\r
 385                 }\r
 386         } else {\r
 387                 // URL was already stored\r
 388                 $return['status'] = 'fail';\r
 389                 $return['code'] = 'error:url';\r
 390                 $return['message'] = $strip_url.' already exists in database';\r
 391                 $return['shorturl'] = YOURLS_SITE .'/'. $url_exists->keyword;\r
 392         }\r
 393         \r
 394         yourls_do_action( 'post_add_new_link', $url, $keyword );\r
 395 \r
 396         $return['statusCode'] = 200; // regardless of result, this is still a valid request\r
 397         return $return;\r
 398 }\r
 399 \r
 400 \r
 401 // Edit a link\r
 402 function yourls_edit_link($url, $keyword, $newkeyword='') {\r
 403         global $ydb;\r
 404 \r
 405         $table = YOURLS_DB_TABLE_URL;\r
 406         $url = yourls_escape(yourls_sanitize_url($url));\r
 407         $keyword = yourls_sanitize_string( $keyword );\r
 408         $newkeyword = yourls_sanitize_string( $newkeyword );\r
 409         $strip_url = stripslashes($url);\r
 410         $old_url = $ydb->get_var("SELECT `url` FROM `$table` WHERE `keyword` = '$keyword';");\r
 411         $old_id = $id = yourls_string2int( $keyword );\r
 412         $new_id = ( $newkeyword == '' ? $old_id : yourls_string2int( $newkeyword ) );\r
 413         \r
 414         // Check if new URL is not here already\r
 415         if ( $old_url != $url && !yourls_allow_duplicate_longurls() ) {\r
 416                 $new_url_already_there = intval($ydb->get_var("SELECT COUNT(keyword) FROM `$table` WHERE `url` = '$strip_url';"));\r
 417         } else {\r
 418                 $new_url_already_there = false;\r
 419         }\r
 420         \r
 421         // Check if the new keyword is not here already\r
 422         if ( $newkeyword != $keyword ) {\r
 423                 $keyword_is_ok = yourls_keyword_is_free( $newkeyword );\r
 424         } else {\r
 425                 $keyword_is_ok = true;\r
 426         }\r
 427         \r
 428         yourls_do_action( 'pre_edit_link', $url, $keyword, $newkeyword, $new_url_already_there, $keyword_is_ok );\r
 429         \r
 430         // All clear, update\r
 431         if ( ( !$new_url_already_there || yourls_allow_duplicate_longurls() ) && $keyword_is_ok ) {\r
 432                         $update_url = $ydb->query("UPDATE `$table` SET `url` = '$url', `keyword` = '$newkeyword' WHERE `keyword` = '$keyword';");\r
 433                 if( $update_url ) {\r
 434                         $return['url'] = array( 'keyword' => $newkeyword, 'shorturl' => YOURLS_SITE.'/'.$newkeyword, 'url' => $strip_url, 'display_url' => yourls_trim_long_string( $strip_url ), 'new_id' => $new_id );\r
 435                         $return['status'] = 'success';\r
 436                         $return['message'] = 'Link updated in database';\r
 437                 } else {\r
 438                         $return['status'] = 'fail';\r
 439                         $return['message'] = 'Error updating '.$strip_url.' (Short URL: '.$keyword.') to database';\r
 440                 }\r
 441         \r
 442         // Nope\r
 443         } else {\r
 444                 $return['status'] = 'fail';\r
 445                 $return['message'] = 'URL or keyword already exists in database';\r
 446         }\r
 447         \r
 448         return yourls_apply_filter( 'edit_link', $return, $url, $keyword, $newkeyword, $new_url_already_there, $keyword_is_ok );\r
 449 }\r
 450 \r
 451 \r
 452 // Check if keyword id is free (ie not already taken, and not reserved). Return bool.\r
 453 function yourls_keyword_is_free( $keyword ) {\r
 454         $free = true;\r
 455         if ( yourls_keyword_is_reserved( $keyword ) or yourls_keyword_is_taken( $keyword ) )\r
 456                 $free = false;\r
 457                 \r
 458         return yourls_apply_filter( 'keyword_is_free', $free, $keyword );\r
 459 }\r
 460 \r
 461 // Check if a keyword is taken (ie there is already a short URL with this id). Return bool.             \r
 462 function yourls_keyword_is_taken( $keyword ) {\r
 463         global $ydb;\r
 464         $keyword = yourls_sanitize_keyword( $keyword );\r
 465         $taken = false;\r
 466         $table = YOURLS_DB_TABLE_URL;\r
 467         $already_exists = $ydb->get_var("SELECT COUNT(`keyword`) FROM `$table` WHERE `keyword` = '$keyword';");\r
 468         if ( $already_exists )\r
 469                 $taken = true;\r
 470 \r
 471         return yourls_apply_filter( 'keyword_is_taken', $taken );\r
 472 }\r
 473 \r
 474 \r
 475 // Display a page\r
 476 function yourls_page( $page ) {\r
 477         $include = YOURLS_ABSPATH . "/pages/$page.php";\r
 478         if (!file_exists($include)) {\r
 479                 yourls_die("Page '$page' not found", 'Not found', 404);\r
 480         }\r
 481         yourls_do_action( 'pre_page', $page );\r
 482         include($include);\r
 483         yourls_do_action( 'post_page', $page );\r
 484         die();  \r
 485 }\r
 486 \r
 487 // Connect to DB\r
 488 function yourls_db_connect() {\r
 489         global $ydb;\r
 490 \r
 491         if (!defined('YOURLS_DB_USER')\r
 492                 or !defined('YOURLS_DB_PASS')\r
 493                 or !defined('YOURLS_DB_NAME')\r
 494                 or !defined('YOURLS_DB_HOST')\r
 495                 or !class_exists('ezSQL_mysql')\r
 496         ) yourls_die ('DB config missing, or could not find DB class', 'Fatal error', 503);\r
 497         \r
 498         // Are we standalone or in the WordPress environment?\r
 499         if ( class_exists('wpdb') ) {\r
 500                 $ydb =  new wpdb(YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST);\r
 501         } else {\r
 502                 $ydb =  new ezSQL_mysql(YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST);\r
 503         }\r
 504         if ( $ydb->last_error )\r
 505                 yourls_die( $ydb->last_error, 'Fatal error', 503 );\r
 506         \r
 507         if ( defined('YOURLS_DEBUG') && YOURLS_DEBUG === true )\r
 508                 $ydb->show_errors = true;\r
 509         \r
 510         return $ydb;\r
 511 }\r
 512 \r
 513 // Return XML output.\r
 514 function yourls_xml_encode($array) {\r
 515         require_once(YOURLS_INC.'/functions-xml.php');\r
 516         $converter= new yourls_array2xml;\r
 517         return $converter->array2xml($array);\r
 518 }\r
 519 \r
 520 // Return array of all informations associated with keyword. Returns false if keyword not found. Set optional $use_cache to false to force fetching from DB\r
 521 function yourls_get_keyword_infos( $keyword, $use_cache = true ) {\r
 522         global $ydb;\r
 523         $keyword = yourls_sanitize_string( $keyword );\r
 524 \r
 525         if( isset( $ydb->infos[$keyword] ) && $use_cache == true ) {\r
 526                 return yourls_apply_filter( 'get_keyword_infos', $ydb->infos[$keyword], $keyword );\r
 527         }\r
 528         \r
 529         $table = YOURLS_DB_TABLE_URL;\r
 530         $infos = $ydb->get_row("SELECT * FROM `$table` WHERE `keyword` = '$keyword'");\r
 531         \r
 532         if( $infos ) {\r
 533                 $infos = (array)$infos;\r
 534                 $ydb->infos[$keyword] = $infos;\r
 535         } else {\r
 536                 $ydb->infos[$keyword] = false;\r
 537         }\r
 538                 \r
 539         return yourls_apply_filter( 'get_keyword_infos', $ydb->infos[$keyword], $keyword );\r
 540 }\r
 541 \r
 542 // Return (string) selected information associated with a keyword. Optional $notfound = string default message if nothing found\r
 543 function yourls_get_keyword_info( $keyword, $field, $notfound = false ) {\r
 544         $keyword = yourls_sanitize_string( $keyword );\r
 545         $infos = yourls_get_keyword_infos( $keyword );\r
 546         \r
 547         $return = $notfound;\r
 548         if ( isset($infos[$field]) && $infos[$field] !== false )\r
 549                 $return = $infos[$field];\r
 550 \r
 551         return yourls_apply_filter( 'get_keyword_info', $return, $keyword, $field, $notfound ); \r
 552 }\r
 553 \r
 554 // Return long URL associated with keyword. Optional $notfound = string default message if nothing found\r
 555 function yourls_get_keyword_longurl( $keyword, $notfound = false ) {\r
 556         return yourls_get_keyword_info( $keyword, 'url', $notfound );\r
 557 }\r
 558 \r
 559 // Return number of clicks on a keyword. Optional $notfound = string default message if nothing found\r
 560 function yourls_get_keyword_clicks( $keyword, $notfound = false ) {\r
 561         return yourls_get_keyword_info( $keyword, 'clicks', $notfound );\r
 562 }\r
 563 \r
 564 // Return IP that added a keyword. Optional $notfound = string default message if nothing found\r
 565 function yourls_get_keyword_IP( $keyword, $notfound = false ) {\r
 566         return yourls_get_keyword_info( $keyword, 'ip', $notfound );\r
 567 }\r
 568 \r
 569 // Return timestamp associated with a keyword. Optional $notfound = string default message if nothing found\r
 570 function yourls_get_keyword_timestamp( $keyword, $notfound = false ) {\r
 571         return yourls_get_keyword_info( $keyword, 'timestamp', $notfound );\r
 572 }\r
 573 \r
 574 // Update click count on a short URL. Return 0/1 for error/success.\r
 575 function yourls_update_clicks( $keyword ) {\r
 576         global $ydb;\r
 577         $keyword = yourls_sanitize_string( $keyword );\r
 578         $table = YOURLS_DB_TABLE_URL;\r
 579         $update = $ydb->query("UPDATE `$table` SET `clicks` = clicks + 1 WHERE `keyword` = '$keyword'");\r
 580         yourls_do_action( 'update_clicks', $keyword, $update );\r
 581         return $update;\r
 582 }\r
 583 \r
 584 // Return array of stats. (string)$filter is 'bottom', 'last', 'rand' or 'top'. (int)$limit is the number of links to return\r
 585 function yourls_get_stats( $filter = 'top', $limit = 10 ) {\r
 586         global $ydb;\r
 587 \r
 588         switch( $filter ) {\r
 589                 case 'bottom':\r
 590                         $sort_by = 'clicks';\r
 591                         $sort_order = 'asc';\r
 592                         break;\r
 593                 case 'last':\r
 594                         $sort_by = 'timestamp';\r
 595                         $sort_order = 'desc';\r
 596                         break;\r
 597                 case 'rand':\r
 598                 case 'random':\r
 599                         $sort_by = 'RAND()';\r
 600                         $sort_order = '';\r
 601                         break;\r
 602                 case 'top':\r
 603                 default:\r
 604                         $sort_by = 'clicks';\r
 605                         $sort_order = 'desc';\r
 606                         break;\r
 607         }\r
 608         \r
 609         $limit = intval( $limit );\r
 610         if ( $limit == 0 )\r
 611                 $limit = 1;\r
 612         $table_url = YOURLS_DB_TABLE_URL;\r
 613         $results = $ydb->get_results("SELECT * FROM `$table_url` WHERE 1=1 ORDER BY `$sort_by` $sort_order LIMIT 0, $limit;");\r
 614         \r
 615         $return = array();\r
 616         $i = 1;\r
 617         \r
 618         foreach ($results as $res) {\r
 619                 $return['links']['link_'.$i++] = array(\r
 620                         'shorturl' => YOURLS_SITE .'/'. $res->keyword,\r
 621                         'url' => $res->url,\r
 622                         'timestamp' => $res->timestamp,\r
 623                         'ip' => $res->ip,\r
 624                         'clicks' => $res->clicks,\r
 625                 );\r
 626         }\r
 627 \r
 628         $return['stats'] = yourls_get_db_stats();\r
 629         \r
 630         $return['statusCode'] = 200;\r
 631 \r
 632         return yourls_apply_filter( 'get_stats', $return);\r
 633 }\r
 634 \r
 635 // Return array of stats. (string)$filter is 'bottom', 'last', 'rand' or 'top'. (int)$limit is the number of links to return\r
 636 function yourls_get_link_stats( $shorturl ) {\r
 637         global $ydb;\r
 638 \r
 639         $table_url = YOURLS_DB_TABLE_URL;\r
 640         $res = $ydb->get_row("SELECT * FROM `$table_url` WHERE keyword = '$shorturl';");\r
 641         $return = array();\r
 642 \r
 643         if( !$res ) {\r
 644                 // non existent link\r
 645                 $return = array(\r
 646                         'statusCode' => 404,\r
 647                         'message'    => 'Error: short URL not found',\r
 648                 );\r
 649         } else {\r
 650                 $return = array(\r
 651                         'statusCode' => 200,\r
 652                         'message'    => 'success',\r
 653                         'link'       => array(\r
 654                         'shorturl' => YOURLS_SITE .'/'. $res->keyword,\r
 655                         'url' => $res->url,\r
 656                         'timestamp' => $res->timestamp,\r
 657                         'ip' => $res->ip,\r
 658                         'clicks' => $res->clicks,\r
 659                         )\r
 660                 );\r
 661         }\r
 662 \r
 663         return yourls_apply_filter( 'get_link_stats', $return );\r
 664 }\r
 665 \r
 666 // Return array for API stat requests\r
 667 function yourls_api_stats( $filter = 'top', $limit = 10 ) {\r
 668         $return = yourls_get_stats( $filter, $limit );\r
 669         $return['simple']  = 'Need either XML or JSON format for stats';\r
 670         $return['message'] = 'success';\r
 671         return yourls_apply_filter( 'api_stats', $return );\r
 672 }\r
 673 \r
 674 // Return array for API stat requests\r
 675 function yourls_api_url_stats($shorturl) {\r
 676         $keyword = str_replace( YOURLS_SITE . '/' , '', $shorturl ); // accept either 'http://ozh.in/abc' or 'abc'\r
 677         $keyword = yourls_sanitize_string( $keyword );\r
 678 \r
 679         $return = yourls_get_link_stats( $keyword );\r
 680         $return['simple']  = 'Need either XML or JSON format for stats';\r
 681         return yourls_apply_filter( 'api_url_stats', $return );\r
 682 }\r
 683 \r
 684 // Expand short url to long url\r
 685 function yourls_api_expand( $shorturl ) {\r
 686         $keyword = str_replace( YOURLS_SITE . '/' , '', $shorturl ); // accept either 'http://ozh.in/abc' or 'abc'\r
 687         $keyword = yourls_sanitize_string( $keyword );\r
 688         \r
 689         $longurl = yourls_get_keyword_longurl( $keyword );\r
 690         \r
 691         if( $longurl ) {\r
 692                 $return = array(\r
 693                         'keyword'  => $keyword,\r
 694                         'shorturl' => YOURLS_SITE . "/$keyword",\r
 695                         'longurl'  => $longurl,\r
 696                         'simple'   => $longurl,\r
 697                         'message'  => 'success',\r
 698                         'statusCode' => 200,\r
 699                 );\r
 700         } else {\r
 701                 $return = array(\r
 702                         'keyword'  => $keyword,\r
 703                         'simple'   => 'not found',\r
 704                         'message'  => 'Error: short URL not found',\r
 705                         'errorCode' => 404,\r
 706                 );\r
 707         }\r
 708         \r
 709         return yourls_apply_filter( 'api_expand', $return );\r
 710 }\r
 711 \r
 712 \r
 713 // Get total number of URLs and sum of clicks. Input: optional "AND WHERE" clause. Returns array\r
 714 function yourls_get_db_stats( $where = '' ) {\r
 715         global $ydb;\r
 716         $table_url = YOURLS_DB_TABLE_URL;\r
 717 \r
 718         $totals = $ydb->get_row("SELECT COUNT(keyword) as count, SUM(clicks) as sum FROM `$table_url` WHERE 1=1 $where");\r
 719         $return = array( 'total_links' => $totals->count, 'total_clicks' => $totals->sum );\r
 720         \r
 721         return yourls_apply_filter( 'get_db_stats', $return );\r
 722 }\r
 723 \r
 724 // Return API result. Dies after this\r
 725 function yourls_api_output( $mode, $return ) {\r
 726         if( isset( $return['simple'] ) ) {\r
 727                 $simple = $return['simple'];\r
 728                 unset( $return['simple'] );\r
 729         }\r
 730         \r
 731         yourls_do_action( 'pre_api_output', $mode, $return );\r
 732         \r
 733         switch ( $mode ) {\r
 734                 case 'json':\r
 735                         header('Content-type: application/json');\r
 736                         echo json_encode($return);\r
 737                         break;\r
 738                 \r
 739                 case 'xml':\r
 740                         header('Content-type: application/xml');\r
 741                         echo yourls_xml_encode($return);\r
 742                         break;\r
 743                         \r
 744                 case 'simple':\r
 745                 default:\r
 746                         if( isset( $simple ) )\r
 747                                 echo $simple;\r
 748                         break;\r
 749         }\r
 750 \r
 751         yourls_do_action( 'api_output', $mode, $return );\r
 752         \r
 753         die();\r
 754 }\r
 755 \r
 756 // Get number of SQL queries performed\r
 757 function yourls_get_num_queries() {\r
 758         global $ydb;\r
 759 \r
 760         return yourls_apply_filter( 'get_num_queries', $ydb->num_queries );\r
 761 }\r
 762 \r
 763 // Returns a sanitized a user agent string. Given what I found on http://www.user-agents.org/ it should be OK.\r
 764 function yourls_get_user_agent() {\r
 765         if ( !isset( $_SERVER['HTTP_USER_AGENT'] ) )\r
 766                 return '-';\r
 767         \r
 768         $ua = strip_tags( html_entity_decode( $_SERVER['HTTP_USER_AGENT'] ));\r
 769         $ua = preg_replace('![^0-9a-zA-Z\':., /{}\(\)\[\]\+@&\!\?;_\-=~\*\#]!', '', $ua );\r
 770                 \r
 771         return yourls_apply_filter( 'get_user_agent', substr( $ua, 0, 254 ) );\r
 772 }\r
 773 \r
 774 // Redirect to another page\r
 775 function yourls_redirect( $location, $code = 301 ) {\r
 776         yourls_do_action( 'pre_redirect', $location, $code );\r
 777         // Redirect, either properly if possible, or via Javascript otherwise\r
 778         if( !headers_sent() ) {\r
 779                 yourls_status_header( $code );\r
 780                 header("Location: $location");\r
 781         } else {\r
 782                 yourls_redirect_javascript( $location );\r
 783         }\r
 784         die();\r
 785 }\r
 786 \r
 787 // Set HTTP status header\r
 788 function yourls_status_header( $code = 200 ) {\r
 789         if( headers_sent() )\r
 790                 return;\r
 791                 \r
 792         $protocol = $_SERVER["SERVER_PROTOCOL"];\r
 793         if ( 'HTTP/1.1' != $protocol && 'HTTP/1.0' != $protocol )\r
 794                 $protocol = 'HTTP/1.0';\r
 795 \r
 796         $code = intval( $code );\r
 797         $desc = yourls_get_HTTP_status($code);\r
 798 \r
 799         @header ("$protocol $code $desc"); // This causes problems on IIS and some FastCGI setups\r
 800         yourls_do_action( 'status_header', $code );\r
 801 }\r
 802 \r
 803 // Redirect to another page using Javascript. Set optional (bool)$dontwait to false to force manual redirection (make sure a message has been read by user)\r
 804 function yourls_redirect_javascript( $location, $dontwait = true ) {\r
 805         if( $dontwait ) {\r
 806         echo <<<REDIR\r
 807         <script type="text/javascript">\r
 808         window.location="$location";\r
 809         </script>\r
 810         <small>(if you are not redirected after 10 seconds, please <a href="$location">click here</a>)</small>\r
 811 REDIR;\r
 812         } else {\r
 813         echo <<<MANUAL\r
 814         <p>Please <a href="$location">click here</a></p>\r
 815 MANUAL;\r
 816         }\r
 817         yourls_do_action( 'redirect_javascript', $location );\r
 818 }\r
 819 \r
 820 // Return a HTTP status code\r
 821 function yourls_get_HTTP_status( $code ) {\r
 822         $code = intval( $code );\r
 823         $headers_desc = array(\r
 824                 100 => 'Continue',\r
 825                 101 => 'Switching Protocols',\r
 826                 102 => 'Processing',\r
 827 \r
 828                 200 => 'OK',\r
 829                 201 => 'Created',\r
 830                 202 => 'Accepted',\r
 831                 203 => 'Non-Authoritative Information',\r
 832                 204 => 'No Content',\r
 833                 205 => 'Reset Content',\r
 834                 206 => 'Partial Content',\r
 835                 207 => 'Multi-Status',\r
 836                 226 => 'IM Used',\r
 837 \r
 838                 300 => 'Multiple Choices',\r
 839                 301 => 'Moved Permanently',\r
 840                 302 => 'Found',\r
 841                 303 => 'See Other',\r
 842                 304 => 'Not Modified',\r
 843                 305 => 'Use Proxy',\r
 844                 306 => 'Reserved',\r
 845                 307 => 'Temporary Redirect',\r
 846 \r
 847                 400 => 'Bad Request',\r
 848                 401 => 'Unauthorized',\r
 849                 402 => 'Payment Required',\r
 850                 403 => 'Forbidden',\r
 851                 404 => 'Not Found',\r
 852                 405 => 'Method Not Allowed',\r
 853                 406 => 'Not Acceptable',\r
 854                 407 => 'Proxy Authentication Required',\r
 855                 408 => 'Request Timeout',\r
 856                 409 => 'Conflict',\r
 857                 410 => 'Gone',\r
 858                 411 => 'Length Required',\r
 859                 412 => 'Precondition Failed',\r
 860                 413 => 'Request Entity Too Large',\r
 861                 414 => 'Request-URI Too Long',\r
 862                 415 => 'Unsupported Media Type',\r
 863                 416 => 'Requested Range Not Satisfiable',\r
 864                 417 => 'Expectation Failed',\r
 865                 422 => 'Unprocessable Entity',\r
 866                 423 => 'Locked',\r
 867                 424 => 'Failed Dependency',\r
 868                 426 => 'Upgrade Required',\r
 869 \r
 870                 500 => 'Internal Server Error',\r
 871                 501 => 'Not Implemented',\r
 872                 502 => 'Bad Gateway',\r
 873                 503 => 'Service Unavailable',\r
 874                 504 => 'Gateway Timeout',\r
 875                 505 => 'HTTP Version Not Supported',\r
 876                 506 => 'Variant Also Negotiates',\r
 877                 507 => 'Insufficient Storage',\r
 878                 510 => 'Not Extended'\r
 879         );\r
 880 \r
 881         if ( isset( $headers_desc[$code] ) )\r
 882                 return $headers_desc[$code];\r
 883         else\r
 884                 return '';\r
 885 }\r
 886 \r
 887 \r
 888 // Log a redirect (for stats)\r
 889 function yourls_log_redirect( $keyword ) {\r
 890         if ( !yourls_do_log_redirect() )\r
 891                 return true;\r
 892 \r
 893         global $ydb;\r
 894         $table = YOURLS_DB_TABLE_LOG;\r
 895         \r
 896         $keyword = yourls_sanitize_string( $keyword );\r
 897         $referrer = ( isset( $_SERVER['HTTP_REFERER'] ) ? yourls_sanitize_url( $_SERVER['HTTP_REFERER'] ) : 'direct' );\r
 898         $ua = yourls_get_user_agent();\r
 899         $ip = yourls_get_IP();\r
 900         $location = yourls_geo_ip_to_countrycode( $ip );\r
 901         \r
 902         return $ydb->query( "INSERT INTO `$table` VALUES ('', NOW(), '$keyword', '$referrer', '$ua', '$ip', '$location')" );\r
 903 }\r
 904 \r
 905 // Check if we want to not log redirects (for stats)\r
 906 function yourls_do_log_redirect() {\r
 907         return ( !defined('YOURLS_NOSTATS') || YOURLS_NOSTATS != true );\r
 908 }\r
 909 \r
 910 // Converts an IP to a 2 letter country code, using GeoIP database if available in includes/geo/\r
 911 function yourls_geo_ip_to_countrycode( $ip = '', $default = '' ) {\r
 912         // allow a plugin to shortcircuit the Geo IP API\r
 913         $location = yourls_apply_filter( 'pre_geo_ip_to_countrycode', false, $ip, $default ); // at this point $ip can be '', check if your plugin hooks in here\r
 914         if ( false !== $location )\r
 915                 return $location;\r
 916 \r
 917         if ( !file_exists( YOURLS_INC.'/geo/GeoIP.dat') || !file_exists( YOURLS_INC.'/geo/geoip.inc') )\r
 918                 return $default;\r
 919 \r
 920         if ( $ip == '' )\r
 921                 $ip = yourls_get_IP();\r
 922         \r
 923         require_once( YOURLS_INC.'/geo/geoip.inc') ;\r
 924         $gi = geoip_open( YOURLS_INC.'/geo/GeoIP.dat', GEOIP_STANDARD);\r
 925         $location = geoip_country_code_by_addr($gi, $ip);\r
 926         geoip_close($gi);\r
 927 \r
 928         return yourls_apply_filter( 'geo_ip_to_countrycode', $location, $ip, $default );\r
 929 }\r
 930 \r
 931 // Converts a 2 letter country code to long name (ie AU -> Australia)\r
 932 function yourls_geo_countrycode_to_countryname( $code ) {\r
 933         // Load the Geo class if not already done\r
 934         if( !class_exists('GeoIP') ) {\r
 935                 $temp = yourls_geo_ip_to_countrycode('127.0.0.1');\r
 936         }\r
 937         \r
 938         if( class_exists('GeoIP') ) {\r
 939                 $geo = new GeoIP;\r
 940                 $id = $geo->GEOIP_COUNTRY_CODE_TO_NUMBER[$code];\r
 941                 $long = $geo->GEOIP_COUNTRY_NAMES[$id];\r
 942                 return $long;\r
 943         } else {\r
 944                 return false;\r
 945         }\r
 946 }\r
 947 \r
 948 // Return flag URL from 2 letter country code\r
 949 function yourls_geo_get_flag( $code ) {\r
 950         // Load the Geo class if not already done\r
 951         if( !class_exists('GeoIP') ) {\r
 952                 $temp = yourls_geo_ip_to_countrycode('127.0.0.1');\r
 953         }\r
 954         \r
 955         if( class_exists('GeoIP') ) {\r
 956                 return YOURLS_SITE.'/includes/geo/flags/flag_'.(strtolower($code)).'.gif';\r
 957         } else {\r
 958                 return false;\r
 959         }\r
 960 }\r
 961 \r
 962 \r
 963 // Check if an upgrade is needed\r
 964 function yourls_upgrade_is_needed() {\r
 965         // check YOURLS_DB_VERSION exist && match values stored in YOURLS_DB_TABLE_OPTIONS\r
 966         list( $currentver, $currentsql ) = yourls_get_current_version_from_sql();\r
 967         if( $currentsql < YOURLS_DB_VERSION )\r
 968                 return true;\r
 969                 \r
 970         return false;\r
 971 }\r
 972 \r
 973 // Get current version & db version as stored in the options DB. Prior to 1.4 there's no option table.\r
 974 function yourls_get_current_version_from_sql() {\r
 975         $currentver = yourls_get_option( 'version' );\r
 976         $currentsql = yourls_get_option( 'db_version' );\r
 977 \r
 978         // Values if version is 1.3\r
 979         if( !$currentver )\r
 980                 $currentver = '1.3';\r
 981         if( !$currentsql )\r
 982                 $currentsql = '100';\r
 983                 \r
 984         return array( $currentver, $currentsql);\r
 985 }\r
 986 \r
 987 // Read an option from DB (or from cache if available). Return value or $default if not found\r
 988 function yourls_get_option( $option_name, $default = false ) {\r
 989         global $ydb;\r
 990         if ( !isset( $ydb->option[$option_name] ) ) {\r
 991                 $table = YOURLS_DB_TABLE_OPTIONS;\r
 992                 $option_name = yourls_escape( $option_name );\r
 993                 $row = $ydb->get_row( "SELECT `option_value` FROM `$table` WHERE `option_name` = '$option_name' LIMIT 1" );\r
 994                 if ( is_object( $row) ) { // Has to be get_row instead of get_var because of funkiness with 0, false, null values\r
 995                         $value = $row->option_value;\r
 996                 } else { // option does not exist, so we must cache its non-existence\r
 997                         $value = $default;\r
 998                 }\r
 999                 $ydb->option[$option_name] = yourls_maybe_unserialize( $value );\r
1000         }\r
1001 \r
1002         return yourls_apply_filter( 'get_option_'.$option_name, $ydb->option[$option_name] );\r
1003 }\r
1004 \r
1005 // Read all options from DB at once\r
1006 function yourls_get_all_options() {\r
1007         global $ydb;\r
1008         $table = YOURLS_DB_TABLE_OPTIONS;\r
1009         \r
1010         $allopt = $ydb->get_results("SELECT `option_name`, `option_value` FROM `$table` WHERE 1=1");\r
1011         \r
1012         foreach( (array)$allopt as $option ) {\r
1013                 $ydb->option[$option->option_name] = yourls_maybe_unserialize( $option->option_value );\r
1014         }\r
1015 }\r
1016 \r
1017 // Update (add if doesn't exist) an option to DB\r
1018 function yourls_update_option( $option_name, $newvalue ) {\r
1019         global $ydb;\r
1020         $table = YOURLS_DB_TABLE_OPTIONS;\r
1021 \r
1022         $safe_option_name = yourls_escape( $option_name );\r
1023 \r
1024         $oldvalue = yourls_get_option( $safe_option_name );\r
1025 \r
1026         // If the new and old values are the same, no need to update.\r
1027         if ( $newvalue === $oldvalue )\r
1028                 return false;\r
1029 \r
1030         if ( false === $oldvalue ) {\r
1031                 yourls_add_option( $option_name, $newvalue );\r
1032                 return true;\r
1033         }\r
1034 \r
1035         $_newvalue = yourls_escape( yourls_maybe_serialize( $newvalue ) );\r
1036         \r
1037         yourls_do_action( 'update_option', $option_name, $oldvalue, $newvalue );\r
1038 \r
1039         $ydb->query( "UPDATE `$table` SET `option_value` = '$_newvalue' WHERE `option_name` = '$option_name'");\r
1040 \r
1041         if ( $ydb->rows_affected == 1 ) {\r
1042                 $ydb->option[$option_name] = $newvalue;\r
1043                 return true;\r
1044         }\r
1045         return false;\r
1046 }\r
1047 \r
1048 // Add an option to the DB\r
1049 function yourls_add_option( $name, $value = '' ) {\r
1050         global $ydb;\r
1051         $table = YOURLS_DB_TABLE_OPTIONS;\r
1052         $safe_name = yourls_escape( $name );\r
1053 \r
1054         // Make sure the option doesn't already exist\r
1055         if ( false !== yourls_get_option( $safe_name ) )\r
1056                 return;\r
1057 \r
1058         $_value = yourls_escape( yourls_maybe_serialize( $value ) );\r
1059 \r
1060         yourls_do_action( 'add_option', $safe_name, $_value );\r
1061 \r
1062         $ydb->query( "INSERT INTO `$table` (`option_name`, `option_value`) VALUES ('$name', '$_value')" );\r
1063         $ydb->option[$name] = $value;\r
1064         return;\r
1065 }\r
1066 \r
1067 \r
1068 // Delete an option from the DB\r
1069 function yourls_delete_option( $name ) {\r
1070         global $ydb;\r
1071         $table = YOURLS_DB_TABLE_OPTIONS;\r
1072         $name = yourls_escape( $name );\r
1073 \r
1074         // Get the ID, if no ID then return\r
1075         $option = $ydb->get_row( "SELECT option_id FROM `$table` WHERE `option_name` = '$name'" );\r
1076         if ( is_null($option) || !$option->option_id )\r
1077                 return false;\r
1078                 \r
1079         yourls_do_action( 'delete_option', $option_name );\r
1080                 \r
1081         $ydb->query( "DELETE FROM `$table` WHERE `option_name` = '$name'" );\r
1082         return true;\r
1083 }\r
1084 \r
1085 \r
1086 \r
1087 // Serialize data if needed. Stolen from WordPress\r
1088 function yourls_maybe_serialize( $data ) {\r
1089         if ( is_array( $data ) || is_object( $data ) )\r
1090                 return serialize( $data );\r
1091 \r
1092         if ( yourls_is_serialized( $data ) )\r
1093                 return serialize( $data );\r
1094 \r
1095         return $data;\r
1096 }\r
1097 \r
1098 // Check value to find if it was serialized. Stolen from WordPress\r
1099 function yourls_is_serialized( $data ) {\r
1100         // if it isn't a string, it isn't serialized\r
1101         if ( !is_string( $data ) )\r
1102                 return false;\r
1103         $data = trim( $data );\r
1104         if ( 'N;' == $data )\r
1105                 return true;\r
1106         if ( !preg_match( '/^([adObis]):/', $data, $badions ) )\r
1107                 return false;\r
1108         switch ( $badions[1] ) {\r
1109                 case 'a' :\r
1110                 case 'O' :\r
1111                 case 's' :\r
1112                         if ( preg_match( "/^{$badions[1]}:[0-9]+:.*[;}]\$/s", $data ) )\r
1113                                 return true;\r
1114                         break;\r
1115                 case 'b' :\r
1116                 case 'i' :\r
1117                 case 'd' :\r
1118                         if ( preg_match( "/^{$badions[1]}:[0-9.E-]+;\$/", $data ) )\r
1119                                 return true;\r
1120                         break;\r
1121         }\r
1122         return false;\r
1123 }\r
1124 \r
1125 // Unserialize value only if it was serialized. Stolen from WP\r
1126 function yourls_maybe_unserialize( $original ) {\r
1127         if ( yourls_is_serialized( $original ) ) // don't attempt to unserialize data that wasn't serialized going in\r
1128                 return @unserialize( $original );\r
1129         return $original;\r
1130 }\r
1131 \r
1132 // Determine if the current page is private\r
1133 function yourls_is_private() {\r
1134         $private = false;\r
1135 \r
1136         if (defined('YOURLS_PRIVATE') && YOURLS_PRIVATE == true) {\r
1137 \r
1138                 // Allow overruling of particular pages\r
1139                 $current = basename( $_SERVER["SCRIPT_NAME"] );\r
1140 \r
1141                 switch( $current ) {\r
1142                 \r
1143                 case 'yourls-api.php':\r
1144                         if( !defined('YOURLS_PRIVATE_API') || YOURLS_PRIVATE_API != false )\r
1145                                 $private = true;\r
1146                         break;\r
1147                                 \r
1148                 case 'yourls-infos.php':\r
1149                         if( !defined('YOURLS_PRIVATE_INFOS') || YOURLS_PRIVATE_INFOS !== false )\r
1150                                 $private = true;\r
1151                         break;\r
1152                 \r
1153                 default:\r
1154                         $private = true;\r
1155                         break;\r
1156                 }\r
1157         }\r
1158         \r
1159         return yourls_apply_filter( 'is_private', $private );\r
1160 }\r
1161 \r
1162 // Show login form if required\r
1163 function yourls_maybe_require_auth() {\r
1164         if( yourls_is_private() )\r
1165                 require_once( YOURLS_INC.'/auth.php' );\r
1166 }\r
1167 \r
1168 // Return word or words if more than one\r
1169 function yourls_plural( $word, $count=1 ) {\r
1170         return $word . ($count > 1 ? 's' : '');\r
1171 }\r
1172 \r
1173 // Return trimmed string\r
1174 function yourls_trim_long_string( $string, $length = 60, $append = '[...]' ) {\r
1175         $newstring = $string;\r
1176         if ( strlen( $newstring ) > $length ) {\r
1177                 $newstring = substr( $newstring, 0, $length - strlen( $append ) ) . $append;    \r
1178         }\r
1179         return yourls_apply_filter( 'trim_long_string', $newstring, $string, $length, $append );\r
1180 }\r
1181 \r
1182 // Allow several short URLs for the same long URL ?\r
1183 function yourls_allow_duplicate_longurls() {\r
1184         // special treatment if API to check for WordPress plugin requests\r
1185         if( yourls_is_API() ) {\r
1186                 if ( isset($_REQUEST['source']) && $_REQUEST['source'] == 'plugin' ) \r
1187                         return false;\r
1188         }\r
1189         return ( defined( 'YOURLS_UNIQUE_URLS' ) && YOURLS_UNIQUE_URLS == false );\r
1190 }\r
1191 \r
1192 // Return list of all shorturls associated to the same long URL. Returns NULL or array of keywords.\r
1193 function yourls_get_duplicate_keywords( $longurl ) {\r
1194         if( !yourls_allow_duplicate_longurls() )\r
1195                 return NULL;\r
1196         \r
1197         global $ydb;\r
1198         $longurl = yourls_escape( yourls_sanitize_url($longurl) );\r
1199         $table = YOURLS_DB_TABLE_URL;\r
1200         \r
1201         $return = $ydb->get_col( "SELECT `keyword` FROM `$table` WHERE `url` = '$longurl'" );\r
1202         return yourls_apply_filter( 'get_duplicate_keywords', $return );\r
1203 }\r
1204 \r
1205 // Check if an IP shortens URL too fast to prevent DB flood. Return true, or die.\r
1206 function yourls_check_IP_flood( $ip = '' ) {\r
1207 \r
1208         yourls_do_action( 'pre_check_ip_flood', $ip ); // at this point $ip can be '', check it if your plugin hooks in here\r
1209 \r
1210         if(\r
1211                 ( defined('YOURLS_FLOOD_DELAY_SECONDS') && YOURLS_FLOOD_DELAY_SECONDS === 0 ) ||\r
1212                 !defined('YOURLS_FLOOD_DELAY_SECONDS')\r
1213         )\r
1214                 return true;\r
1215 \r
1216         $ip = ( $ip ? yourls_sanitize_ip( $ip ) : yourls_get_IP() );\r
1217 \r
1218         // Don't throttle whitelist IPs\r
1219         if( defined('YOURLS_FLOOD_IP_WHITELIST' && YOURLS_FLOOD_IP_WHITELIST ) ) {\r
1220                 $whitelist_ips = explode( ',', YOURLS_FLOOD_IP_WHITELIST );\r
1221                 foreach( $whitelist_ips as $whitelist_ip ) {\r
1222                         $whitelist_ip = trim( $whitelist_ip );\r
1223                         if ( $whitelist_ip == $ip )\r
1224                                 return true;\r
1225                 }\r
1226         }\r
1227         \r
1228         // Don't throttle logged in users\r
1229         if( yourls_is_private() ) {\r
1230                  if( yourls_is_valid_user() === true )\r
1231                         return true;\r
1232         }\r
1233         \r
1234         yourls_do_action( 'check_ip_flood', $ip );\r
1235         \r
1236         global $ydb;\r
1237         $table = YOURLS_DB_TABLE_URL;\r
1238         \r
1239         $lasttime = $ydb->get_var( "SELECT `timestamp` FROM $table WHERE `ip` = '$ip' ORDER BY `timestamp` DESC LIMIT 1" );\r
1240         if( $lasttime ) {\r
1241                 $now = date( 'U' );\r
1242                 $then = date( 'U', strtotime( $lasttime ) );\r
1243                 if( ( $now - $then ) <= YOURLS_FLOOD_DELAY_SECONDS ) {\r
1244                         // Flood!\r
1245                         yourls_do_action( 'ip_flood', $ip, $now - $then );\r
1246                         yourls_die( 'Too many URLs added too fast. Slow down please.', 'Forbidden', 403 );\r
1247                 }\r
1248         }\r
1249         \r
1250         return true;\r
1251 }\r
1252 \r
1253 // Check if YOURLS is installed\r
1254 function yourls_is_installed() {\r
1255         static $is_installed = false;\r
1256         if ( $is_installed === false ) {\r
1257                 $check_14 = $check_13 = false;\r
1258                 global $ydb;\r
1259                 if( defined('YOURLS_DB_TABLE_NEXTDEC') )\r
1260                         $check_13 = $ydb->get_var('SELECT `next_id` FROM '.YOURLS_DB_TABLE_NEXTDEC);\r
1261                 $check_14 = yourls_get_option( 'version' );\r
1262                 $is_installed = $check_13 || $check_14;\r
1263         }\r
1264         return yourls_apply_filter( 'is_installed', $is_installed );\r
1265 }\r
1266 \r
1267 // Generate random string of (int)$lenght length and type $type (see function for details)\r
1268 function yourls_rnd_string ( $length = 5, $type = 1 ) {\r
1269         $str = '';\r
1270         $length = intval( $length );\r
1271 \r
1272         // define possible characters\r
1273         switch ( $type ) {\r
1274                 // no vowels to make no offending word, no 0 or 1 to avoid confusion betwee letters & digits. Perfect for passwords.\r
1275                 case '1':\r
1276                         $possible = "23456789bcdfghjkmnpqrstvwxyz";\r
1277                         break;\r
1278                 \r
1279                 // all letters, lowercase\r
1280                 case '2':\r
1281                         $possible = "abcdefghijklmnopqrstuvwxyz";\r
1282                         break;\r
1283                 \r
1284                 // all letters, lowercase + uppercase\r
1285                 case '3':\r
1286                         $possible = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";\r
1287                         break;\r
1288                 \r
1289                 // all digits & letters lowercase \r
1290                 case '4':\r
1291                         $possible = "0123456789abcdefghijklmnopqrstuvwxyz";\r
1292                         break;\r
1293                 \r
1294                 // all digits & letters lowercase + uppercase\r
1295                 case '5':\r
1296                         $possible = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ";\r
1297                         break;\r
1298                 \r
1299         }\r
1300 \r
1301         $i = 0;\r
1302         while ($i < $length) {\r
1303         $str .= substr($possible, mt_rand(0, strlen($possible)-1), 1);\r
1304                 $i++;\r
1305         }\r
1306         \r
1307         return yourls_apply_filter( 'rnd_string', $str);\r
1308 }\r
1309 \r
1310 // Return salted string\r
1311 function yourls_salt( $string ) {\r
1312         $salt = defined('YOURLS_COOKIEKEY') ? YOURLS_COOKIEKEY : md5(__FILE__) ;\r
1313         return yourls_apply_filter( 'yourls_salt', md5 ($string . $salt) );\r
1314 }\r
1315 \r
1316 // Return a time-dependent string for nonce creation\r
1317 function yourls_tick() {\r
1318         return ceil( time() / YOURLS_NONCE_LIFE );\r
1319 }\r
1320 \r
1321 // Create a time limited, action limited and user limited token\r
1322 function yourls_create_nonce( $action = '-1', $user = false ) {\r
1323         if( false == $user )\r
1324                 $user = defined('YOURLS_USER') ? YOURLS_USER : '-1';\r
1325         $tick = yourls_tick();\r
1326         return substr( yourls_salt($tick . $action . $user), 0, 10 );\r
1327 }\r
1328 \r
1329 // Check validity of a nonce (ie time span, user and action match)\r
1330 function yourls_verify_nonce( $nonce, $action = -1, $user = false ) {\r
1331         if( false == $user )\r
1332                 $user = defined('YOURLS_USER') ? YOURLS_USER : '-1';\r
1333         $valid = yourls_create_nonce( $action, $user );\r
1334         \r
1335         return $nonce == $valid ;\r
1336 }\r
1337 \r
1338 // Sanitize a version number (1.4.1-whatever -> 1.4.1)\r
1339 function yourls_sanitize_version( $ver ) {\r
1340         return preg_replace( '/[^0-9.]/', '', $ver );\r
1341 }\r
1342 \r
1343 // Converts keyword into short link\r
1344 function yourls_link( $keyword = '' ) {\r
1345         return YOURLS_SITE . '/' . yourls_sanitize_keyword( $keyword );\r
1346 }\r
1347 \r
1348 // Check if we're in API mode. Returns bool\r
1349 function yourls_is_API() {\r
1350         if ( defined('YOURLS_API') && YOURLS_API == true )\r
1351                 return true;\r
1352         return false;\r
1353 }\r
1354 \r
1355 // Check if we're in Ajax mode. Returns bool\r
1356 function yourls_is_Ajax() {\r
1357         if ( defined('YOURLS_AJAX') && YOURLS_AJAX == true )\r
1358                 return true;\r
1359         return false;\r
1360 }\r
1361 \r
1362 // Check if we're in GO mode (redirection on yourls-go.php). Returns bool\r
1363 function yourls_is_GO() {\r
1364         if ( defined('YOURLS_GO') && YOURLS_GO == true )\r
1365                 return true;\r
1366         return false;\r
1367 }\r
1368 \r
1369 // Check if we'll need interface display function (ie not API or redirection)\r
1370 function yourls_has_interface() {\r
1371         if( yourls_is_API() or yourls_is_GO() or yourls_is_Ajax() )\r
1372                 return false;\r
1373         return true;\r
1374 }\r
1375 \r
1376 // Check if we're in the admin area. Returns bool\r
1377 function yourls_is_admin() {\r
1378         if ( defined('YOURLS_ADMIN') && YOURLS_ADMIN == true )\r
1379                 return true;\r
1380         return false;\r
1381 }\r
1382 \r
1383 // Check if SSL is required. Returns bool.\r
1384 function yourls_needs_ssl() {\r
1385         if ( defined('YOURLS_ADMIN_SSL') && YOURLS_ADMIN_SSL == true )\r
1386                 return true;\r
1387         return false;\r
1388 }\r
1389 \r
1390 // Return admin link, with SSL preference if applicable.\r
1391 function yourls_admin_url( $page = '' ) {\r
1392         $admin = YOURLS_SITE . '/admin/' . $page;\r
1393         if( defined('YOURLS_ADMIN_SSL') && YOURLS_ADMIN_SSL == true )\r
1394                 $admin = str_replace('http://', 'https://', $admin);\r
1395         return yourls_apply_filter( 'admin_url', $admin, $page );\r
1396 }\r
1397 \r
1398 // Check if SSL is used, returns bool. Stolen from WP.\r
1399 function yourls_is_ssl() {\r
1400         $is_ssl = false;\r
1401         if ( isset($_SERVER['HTTPS']) ) {\r
1402                 if ( 'on' == strtolower($_SERVER['HTTPS']) )\r
1403                         $is_ssl = true;\r
1404                 if ( '1' == $_SERVER['HTTPS'] )\r
1405                         $is_ssl = true;\r
1406         } elseif ( isset($_SERVER['SERVER_PORT']) && ( '443' == $_SERVER['SERVER_PORT'] ) ) {\r
1407                 $is_ssl = true;\r
1408         }\r
1409         return yourls_apply_filter( 'is_ssl', $is_ssl );\r
1410 }\r
1411 \r
1412 \r
1413 // Get a remote page <title>, return a string (either title or url)\r
1414 function yourls_get_remote_title( $url ) {\r
1415         require_once( YOURLS_INC.'/functions-http.php' );\r
1416 \r
1417         $url = yourls_sanitize_url( $url );\r
1418 \r
1419         $title = false;\r
1420         \r
1421         $content = yourls_get_remote_content( $url );\r
1422 \r
1423         // look for <title>\r
1424         if( $content !== false ) {\r
1425                 if ( preg_match('/<title>(.*?)<\/title>/is', $content, $found ) ) {\r
1426                         $title = $found[1];\r
1427                         unset( $found );\r
1428                 }\r
1429         }\r
1430         \r
1431         // if title not found, guess if returned content was actually an error message\r
1432         if( $title == false && strpos( $content, 'Error' ) === 0 ) {\r
1433                 $title = $content;\r
1434         } else {\r
1435                 $title = $url;\r
1436         }\r
1437 \r
1438         return yourls_apply_filter( 'get_remote_title', $title );\r
1439 }\r
1440 \r
1441 // Sanitize a filename (no Win32 stuff)\r
1442 function yourls_sanitize_filename( $file ) {\r
1443         $file = str_replace( '\\', '/', $file ); // sanitize for Win32 installs\r
1444         $file = preg_replace( '|/+|' ,'/', $file ); // remove any duplicate slash\r
1445         return $file;\r
1446 }\r
1447 \r
1448 // Check for maintenance mode that will shortcut everything\r
1449 function yourls_check_maintenance_mode() {\r
1450         \r
1451         // TODO: all cases that always display the sites (is_admin but not is_ajax?)\r
1452         if( 1 )\r
1453                 return;\r
1454 \r
1455         // first case: /user/maintenance.php file\r
1456         if( file_exists( YOURLS_USERDIR.'/maintenance.php' ) ) {\r
1457                 include( YOURLS_USERDIR.'/maintenance.php' );\r
1458                 die();  \r
1459         }\r
1460         \r
1461         // second case: option in DB\r
1462         if( yourls_get_option( 'maintenance_mode' ) !== false ) {\r
1463                 require_once( YOURLS_INC.'/functions-html.php' );\r
1464                 $title = 'Service temporarily unavailable';\r
1465                 $message = 'Our service is currently undergoing scheduled maintenance.</p>\r
1466                 <p>Things should not last very long, thank you for your patience and please excuse the inconvenience';\r
1467                 yourls_die( $message, $title , 503 );\r
1468         }\r
1469         \r
1470 }\r
1471 \r
1472 // Toggle maintenance mode\r
1473 function yourls_maintenance_mode( $maintenance = true ) {\r
1474         yourls_update_option( 'maintenance_mode', (bool)$maintenance );\r
1475 }\r
1476 \r