query("DELETE FROM `$table` WHERE `keyword` = '$keyword';"); yourls_do_action( 'delete_link', $keyword, $delete ); return $delete; } // SQL query to insert a new link in the DB. Returns boolean for success or failure of the inserting function yourls_insert_link_in_db( $url, $keyword, $title = '' ) { global $ydb; $url = addslashes( yourls_sanitize_url( $url ) ); $keyword = addslashes( yourls_sanitize_keyword( $keyword ) ); $title = addslashes( yourls_sanitize_title( $title ) ); $table = YOURLS_DB_TABLE_URL; $timestamp = date('Y-m-d H:i:s'); $ip = yourls_get_IP(); $insert = $ydb->query("INSERT INTO `$table` (`keyword`, `url`, `title`, `timestamp`, `ip`, `clicks`) VALUES('$keyword', '$url', '$title', '$timestamp', '$ip', 0);"); yourls_do_action( 'insert_link', (bool)$insert, $url, $keyword, $title, $timestamp, $ip ); return (bool)$insert; } // Check if a URL already exists in the DB. Return NULL (doesn't exist) or an object with URL informations. function yourls_url_exists( $url ) { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_url_exists', false, $url ); if ( false !== $pre ) return $pre; global $ydb; $table = YOURLS_DB_TABLE_URL; $strip_url = stripslashes($url); $url_exists = $ydb->get_row("SELECT * FROM `$table` WHERE `url` = '".$strip_url."';"); return yourls_apply_filter( 'url_exists', $url_exists, $url ); } // Add a new link in the DB, either with custom keyword, or find one function yourls_add_new_link( $url, $keyword = '', $title = '' ) { global $ydb; // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_add_new_link', false, $url, $keyword, $title ); if ( false !== $pre ) return $pre; if ( !$url || $url == 'http://' || $url == 'https://' ) { $return['status'] = 'fail'; $return['code'] = 'error:nourl'; $return['message'] = 'Missing URL input'; $return['errorCode'] = '400'; return yourls_apply_filter( 'add_new_link_fail_nourl', $return, $url, $keyword, $title ); } // Prevent DB flood $ip = yourls_get_IP(); yourls_check_IP_flood( $ip ); // Prevent internal redirection loops: cannot shorten a shortened URL $url = yourls_escape( yourls_sanitize_url($url) ); if( preg_match( '!^'.YOURLS_SITE.'/!', $url ) ) { if( yourls_is_shorturl( $url ) ) { $return['status'] = 'fail'; $return['code'] = 'error:noloop'; $return['message'] = 'URL is a short URL'; $return['errorCode'] = '400'; return yourls_apply_filter( 'add_new_link_fail_noloop', $return, $url, $keyword, $title ); } } yourls_do_action( 'pre_add_new_link', $url, $keyword, $title ); $strip_url = stripslashes($url); $return = array(); // duplicates allowed or new URL => store it if( yourls_allow_duplicate_longurls() || !yourls_url_exists( $url ) ) { if( isset( $title ) && !empty( $title ) ) { $title = yourls_sanitize_title( $title ); } else { $title = yourls_get_remote_title( $url ); } $title = yourls_apply_filter( 'add_new_title', $title, $url, $keyword ); // Custom keyword provided if ( $keyword ) { yourls_do_action( 'add_new_link_custom_keyword', $url, $keyword, $title ); $keyword = yourls_escape( yourls_sanitize_string($keyword) ); $keyword = yourls_apply_filter( 'custom_keyword', $keyword, $url, $title ); if ( !yourls_keyword_is_free($keyword) ) { // This shorturl either reserved or taken already $return['status'] = 'fail'; $return['code'] = 'error:keyword'; $return['message'] = 'Short URL '.$keyword.' already exists in database or is reserved'; } else { // all clear, store ! yourls_insert_link_in_db( $url, $keyword, $title ); $return['url'] = array('keyword' => $keyword, 'url' => $strip_url, 'title' => $title, 'date' => date('Y-m-d H:i:s'), 'ip' => $ip ); $return['status'] = 'success'; $return['message'] = yourls_trim_long_string( $strip_url ).' added to database'; $return['title'] = $title; $return['html'] = yourls_table_add_row( $keyword, $url, $title, $ip, 0, time() ); $return['shorturl'] = YOURLS_SITE .'/'. $keyword; } // Create random keyword } else { yourls_do_action( 'add_new_link_create_keyword', $url, $keyword, $title ); $timestamp = date('Y-m-d H:i:s'); $id = yourls_get_next_decimal(); $ok = false; do { $keyword = yourls_int2string( $id ); $keyword = yourls_apply_filter( 'random_keyword', $keyword, $url, $title ); $free = yourls_keyword_is_free($keyword); $add_url = @yourls_insert_link_in_db( $url, $keyword, $title ); $ok = ($free && $add_url); if ( $ok === false && $add_url === 1 ) { // we stored something, but shouldn't have (ie reserved id) $delete = yourls_delete_link_by_keyword( $keyword ); $return['extra_info'] .= '(deleted '.$keyword.')'; } else { // everything ok, populate needed vars $return['url'] = array('keyword' => $keyword, 'url' => $strip_url, 'title' => $title, 'date' => $timestamp, 'ip' => $ip ); $return['status'] = 'success'; $return['message'] = yourls_trim_long_string( $strip_url ).' added to database'; $return['title'] = $title; $return['html'] = yourls_table_add_row( $keyword, $url, $title, $ip, 0, time() ); $return['shorturl'] = YOURLS_SITE .'/'. $keyword; } $id++; } while (!$ok); @yourls_update_next_decimal($id); } // URL was already stored } else { yourls_do_action( 'add_new_link_already_stored', $url, $keyword, $title ); $return['status'] = 'fail'; $return['code'] = 'error:url'; $return['url'] = array( 'keyword' => $url_exists->keyword, 'url' => $strip_url, 'title' => $url_exists->title, 'date' => $url_exists->timestamp, 'ip' => $url_exists->ip, 'clicks' => $url_exists->clicks ); $return['message'] = yourls_trim_long_string( $strip_url ).' already exists in database'; $return['title'] = $url_exists->title; $return['shorturl'] = YOURLS_SITE .'/'. $url_exists->keyword; } yourls_do_action( 'post_add_new_link', $url, $keyword, $title ); $return['statusCode'] = 200; // regardless of result, this is still a valid request return yourls_apply_filter( 'add_new_link', $return, $url, $keyword, $title ); } // Edit a link function yourls_edit_link( $url, $keyword, $newkeyword='', $title='' ) { global $ydb; $table = YOURLS_DB_TABLE_URL; $url = yourls_escape(yourls_sanitize_url($url)); $keyword = yourls_escape(yourls_sanitize_string( $keyword )); $title = yourls_escape(yourls_sanitize_title( $title )); $newkeyword = yourls_escape(yourls_sanitize_string( $newkeyword )); $strip_url = stripslashes($url); $strip_title = stripslashes($title); $old_url = $ydb->get_var("SELECT `url` FROM `$table` WHERE `keyword` = '$keyword';"); // Check if new URL is not here already if ( $old_url != $url && !yourls_allow_duplicate_longurls() ) { $new_url_already_there = intval($ydb->get_var("SELECT COUNT(keyword) FROM `$table` WHERE `url` = '$strip_url';")); } else { $new_url_already_there = false; } // Check if the new keyword is not here already if ( $newkeyword != $keyword ) { $keyword_is_ok = yourls_keyword_is_free( $newkeyword ); } else { $keyword_is_ok = true; } yourls_do_action( 'pre_edit_link', $url, $keyword, $newkeyword, $new_url_already_there, $keyword_is_ok ); // All clear, update if ( ( !$new_url_already_there || yourls_allow_duplicate_longurls() ) && $keyword_is_ok ) { $update_url = $ydb->query("UPDATE `$table` SET `url` = '$url', `keyword` = '$newkeyword', `title` = '$title' WHERE `keyword` = '$keyword';"); if( $update_url ) { $return['url'] = array( 'keyword' => $newkeyword, 'shorturl' => YOURLS_SITE.'/'.$newkeyword, 'url' => $strip_url, 'display_url' => yourls_trim_long_string( $strip_url ), 'title' => $strip_title, 'display_title' => yourls_trim_long_string( $strip_title ) ); $return['status'] = 'success'; $return['message'] = 'Link updated in database'; } else { $return['status'] = 'fail'; $return['message'] = 'Error updating '. yourls_trim_long_string( $strip_url ).' (Short URL: '.$keyword.') to database'; } // Nope } else { $return['status'] = 'fail'; $return['message'] = 'URL or keyword already exists in database'; } return yourls_apply_filter( 'edit_link', $return, $url, $keyword, $newkeyword, $title, $new_url_already_there, $keyword_is_ok ); } // Update a title link (no checks for duplicates etc..) function yourls_edit_link_title( $keyword, $title ) { global $ydb; $keyword = yourls_escape( yourls_sanitize_keyword( $keyword ) ); $title = yourls_escape( yourls_sanitize_title( $title ) ); $table = YOURLS_DB_TABLE_URL; $update = $ydb->query("UPDATE `$table` SET `title` = '$title' WHERE `keyword` = '$keyword';"); return $update; } // Check if keyword id is free (ie not already taken, and not reserved). Return bool. function yourls_keyword_is_free( $keyword ) { $free = true; if ( yourls_keyword_is_reserved( $keyword ) or yourls_keyword_is_taken( $keyword ) ) $free = false; return yourls_apply_filter( 'keyword_is_free', $free, $keyword ); } // Check if a keyword is taken (ie there is already a short URL with this id). Return bool. function yourls_keyword_is_taken( $keyword ) { global $ydb; $keyword = yourls_sanitize_keyword( $keyword ); $taken = false; $table = YOURLS_DB_TABLE_URL; $already_exists = $ydb->get_var("SELECT COUNT(`keyword`) FROM `$table` WHERE `keyword` = '$keyword';"); if ( $already_exists ) $taken = true; return yourls_apply_filter( 'keyword_is_taken', $taken, $keyword ); } // Connect to DB function yourls_db_connect() { global $ydb; if (!defined('YOURLS_DB_USER') or !defined('YOURLS_DB_PASS') or !defined('YOURLS_DB_NAME') or !defined('YOURLS_DB_HOST') or !class_exists('ezSQL_mysql') ) yourls_die ('DB config missing, or could not find DB class', 'Fatal error', 503); // Are we standalone or in the WordPress environment? if ( class_exists('wpdb') ) { $ydb = new wpdb(YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST); } else { $ydb = new ezSQL_mysql(YOURLS_DB_USER, YOURLS_DB_PASS, YOURLS_DB_NAME, YOURLS_DB_HOST); } if ( $ydb->last_error ) yourls_die( $ydb->last_error, 'Fatal error', 503 ); if ( defined('YOURLS_DEBUG') && YOURLS_DEBUG === true ) $ydb->show_errors = true; return $ydb; } // Return XML output. function yourls_xml_encode($array) { require_once(YOURLS_INC.'/functions-xml.php'); $converter= new yourls_array2xml; return $converter->array2xml($array); } // 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 function yourls_get_keyword_infos( $keyword, $use_cache = true ) { global $ydb; $keyword = yourls_sanitize_string( $keyword ); yourls_do_action( 'pre_get_keyword', $keyword, $use_cache ); if( isset( $ydb->infos[$keyword] ) && $use_cache == true ) { return yourls_apply_filter( 'get_keyword_infos', $ydb->infos[$keyword], $keyword ); } yourls_do_action( 'get_keyword_not_cached', $keyword ); $table = YOURLS_DB_TABLE_URL; $infos = $ydb->get_row("SELECT * FROM `$table` WHERE `keyword` = '$keyword'"); if( $infos ) { $infos = (array)$infos; $ydb->infos[$keyword] = $infos; } else { $ydb->infos[$keyword] = false; } return yourls_apply_filter( 'get_keyword_infos', $ydb->infos[$keyword], $keyword ); } // Return (string) selected information associated with a keyword. Optional $notfound = string default message if nothing found function yourls_get_keyword_info( $keyword, $field, $notfound = false ) { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_get_keyword_info', false, $keyword, $field, $notfound ); if ( false !== $pre ) return $pre; $keyword = yourls_sanitize_string( $keyword ); $infos = yourls_get_keyword_infos( $keyword ); $return = $notfound; if ( isset($infos[$field]) && $infos[$field] !== false ) $return = $infos[$field]; return yourls_apply_filter( 'get_keyword_info', $return, $keyword, $field, $notfound ); } // Return title associated with keyword. Optional $notfound = string default message if nothing found function yourls_get_keyword_title( $keyword, $notfound = false ) { return yourls_get_keyword_info( $keyword, 'title', $notfound ); } // Return long URL associated with keyword. Optional $notfound = string default message if nothing found function yourls_get_keyword_longurl( $keyword, $notfound = false ) { return yourls_get_keyword_info( $keyword, 'url', $notfound ); } // Return number of clicks on a keyword. Optional $notfound = string default message if nothing found function yourls_get_keyword_clicks( $keyword, $notfound = false ) { return yourls_get_keyword_info( $keyword, 'clicks', $notfound ); } // Return IP that added a keyword. Optional $notfound = string default message if nothing found function yourls_get_keyword_IP( $keyword, $notfound = false ) { return yourls_get_keyword_info( $keyword, 'ip', $notfound ); } // Return timestamp associated with a keyword. Optional $notfound = string default message if nothing found function yourls_get_keyword_timestamp( $keyword, $notfound = false ) { return yourls_get_keyword_info( $keyword, 'timestamp', $notfound ); } // Update click count on a short URL. Return 0/1 for error/success. function yourls_update_clicks( $keyword, $clicks = false ) { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_update_clicks', false, $keyword, $clicks ); if ( false !== $pre ) return $pre; global $ydb; $keyword = yourls_sanitize_string( $keyword ); $table = YOURLS_DB_TABLE_URL; if ( $clicks !== false && is_int( $clicks ) && $clicks >= 0 ) $update = $ydb->query("UPDATE `$table` SET `clicks` = $clicks WHERE `keyword` = '$keyword'"); else $update = $ydb->query("UPDATE `$table` SET `clicks` = clicks + 1 WHERE `keyword` = '$keyword'"); yourls_do_action( 'update_clicks', $keyword, $update, $clicks ); return $update; } // Return array of stats. (string)$filter is 'bottom', 'last', 'rand' or 'top'. (int)$limit is the number of links to return function yourls_get_stats( $filter = 'top', $limit = 10, $start = 0 ) { global $ydb; switch( $filter ) { case 'bottom': $sort_by = 'clicks'; $sort_order = 'asc'; break; case 'last': $sort_by = 'timestamp'; $sort_order = 'desc'; break; case 'rand': case 'random': $sort_by = 'RAND()'; $sort_order = ''; break; case 'top': default: $sort_by = 'clicks'; $sort_order = 'desc'; break; } // Fetch links $limit = intval( $limit ); $start = intval( $start ); if ( $limit > 0 ) { $table_url = YOURLS_DB_TABLE_URL; $results = $ydb->get_results("SELECT * FROM `$table_url` WHERE 1=1 ORDER BY `$sort_by` $sort_order LIMIT $start, $limit;"); $return = array(); $i = 1; foreach ( (array)$results as $res) { $return['links']['link_'.$i++] = array( 'shorturl' => YOURLS_SITE .'/'. $res->keyword, 'url' => $res->url, 'title' => $res->title, 'timestamp'=> $res->timestamp, 'ip' => $res->ip, 'clicks' => $res->clicks, ); } } $return['stats'] = yourls_get_db_stats(); $return['statusCode'] = 200; return yourls_apply_filter( 'get_stats', $return, $filter, $limit, $start ); } // Return array of stats. (string)$filter is 'bottom', 'last', 'rand' or 'top'. (int)$limit is the number of links to return function yourls_get_link_stats( $shorturl ) { global $ydb; $table_url = YOURLS_DB_TABLE_URL; $res = $ydb->get_row("SELECT * FROM `$table_url` WHERE keyword = '$shorturl';"); $return = array(); if( !$res ) { // non existent link $return = array( 'statusCode' => 404, 'message' => 'Error: short URL not found', ); } else { $return = array( 'statusCode' => 200, 'message' => 'success', 'link' => array( 'shorturl' => YOURLS_SITE .'/'. $res->keyword, 'url' => $res->url, 'title' => $res->title, 'timestamp'=> $res->timestamp, 'ip' => $res->ip, 'clicks' => $res->clicks, ) ); } return yourls_apply_filter( 'get_link_stats', $return, $shorturl ); } // Return array for API stat requests function yourls_api_stats( $filter = 'top', $limit = 10, $start = 0 ) { $return = yourls_get_stats( $filter, $limit, $start ); $return['simple'] = 'Need either XML or JSON format for stats'; $return['message'] = 'success'; return yourls_apply_filter( 'api_stats', $return, $filter, $limit, $start ); } // Return array for counts of shorturls and clicks function yourls_api_db_stats() { $return = yourls_get_db_stats(); $return['simple'] = 'Need either XML or JSON format for stats'; $return['message'] = 'success'; return yourls_apply_filter( 'api_db_stats', $return ); } // Return array for API stat requests function yourls_api_url_stats($shorturl) { $keyword = str_replace( YOURLS_SITE . '/' , '', $shorturl ); // accept either 'http://ozh.in/abc' or 'abc' $keyword = yourls_sanitize_string( $keyword ); $return = yourls_get_link_stats( $keyword ); $return['simple'] = 'Need either XML or JSON format for stats'; return yourls_apply_filter( 'api_url_stats', $return, $shorturl ); } // Expand short url to long url function yourls_api_expand( $shorturl ) { $keyword = str_replace( YOURLS_SITE . '/' , '', $shorturl ); // accept either 'http://ozh.in/abc' or 'abc' $keyword = yourls_sanitize_string( $keyword ); $longurl = yourls_get_keyword_longurl( $keyword ); if( $longurl ) { $return = array( 'keyword' => $keyword, 'shorturl' => YOURLS_SITE . "/$keyword", 'longurl' => $longurl, 'simple' => $longurl, 'message' => 'success', 'statusCode' => 200, ); } else { $return = array( 'keyword' => $keyword, 'simple' => 'not found', 'message' => 'Error: short URL not found', 'errorCode' => 404, ); } return yourls_apply_filter( 'api_expand', $return, $shorturl ); } // Get total number of URLs and sum of clicks. Input: optional "AND WHERE" clause. Returns array function yourls_get_db_stats( $where = '' ) { global $ydb; $table_url = YOURLS_DB_TABLE_URL; $totals = $ydb->get_row("SELECT COUNT(keyword) as count, SUM(clicks) as sum FROM `$table_url` WHERE 1=1 $where"); $return = array( 'total_links' => $totals->count, 'total_clicks' => $totals->sum ); return yourls_apply_filter( 'get_db_stats', $return, $where ); } // Return API result. Dies after this function yourls_api_output( $mode, $return ) { if( isset( $return['simple'] ) ) { $simple = $return['simple']; unset( $return['simple'] ); } yourls_do_action( 'pre_api_output', $mode, $return ); switch ( $mode ) { case 'jsonp': header('Content-type: application/javascript'); echo $return['callback'] . '(' . json_encode($return) . ')'; break; case 'json': header('Content-type: application/json'); echo json_encode($return); break; case 'xml': header('Content-type: application/xml'); echo yourls_xml_encode($return); break; case 'simple': default: if( isset( $simple ) ) echo $simple; break; } yourls_do_action( 'api_output', $mode, $return ); die(); } // Get number of SQL queries performed function yourls_get_num_queries() { global $ydb; return yourls_apply_filter( 'get_num_queries', $ydb->num_queries ); } // Returns a sanitized a user agent string. Given what I found on http://www.user-agents.org/ it should be OK. function yourls_get_user_agent() { if ( !isset( $_SERVER['HTTP_USER_AGENT'] ) ) return '-'; $ua = strip_tags( html_entity_decode( $_SERVER['HTTP_USER_AGENT'] )); $ua = preg_replace('![^0-9a-zA-Z\':., /{}\(\)\[\]\+@&\!\?;_\-=~\*\#]!', '', $ua ); return yourls_apply_filter( 'get_user_agent', substr( $ua, 0, 254 ) ); } // Redirect to another page function yourls_redirect( $location, $code = 301 ) { yourls_do_action( 'pre_redirect', $location, $code ); $location = yourls_apply_filter( 'redirect', $location, $code ); // Redirect, either properly if possible, or via Javascript otherwise if( !headers_sent() ) { yourls_status_header( $code ); header( "Location: $location" ); } else { yourls_redirect_javascript( $location ); } die(); } // Set HTTP status header function yourls_status_header( $code = 200 ) { if( headers_sent() ) return; $protocol = $_SERVER["SERVER_PROTOCOL"]; if ( 'HTTP/1.1' != $protocol && 'HTTP/1.0' != $protocol ) $protocol = 'HTTP/1.0'; $code = intval( $code ); $desc = yourls_get_HTTP_status( $code ); @header ("$protocol $code $desc"); // This causes problems on IIS and some FastCGI setups yourls_do_action( 'status_header', $code ); } // 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) function yourls_redirect_javascript( $location, $dontwait = true ) { yourls_do_action( 'pre_redirect_javascript', $location, $dontwait ); $location = yourls_apply_filter( 'redirect_javascript', $location, $dontwait ); if( $dontwait ) { echo << window.location="$location"; (if you are not redirected after 10 seconds, please click here) REDIR; } else { echo <<Please click here

MANUAL; } yourls_do_action( 'post_redirect_javascript', $location ); } // Return a HTTP status code function yourls_get_HTTP_status( $code ) { $code = intval( $code ); $headers_desc = array( 100 => 'Continue', 101 => 'Switching Protocols', 102 => 'Processing', 200 => 'OK', 201 => 'Created', 202 => 'Accepted', 203 => 'Non-Authoritative Information', 204 => 'No Content', 205 => 'Reset Content', 206 => 'Partial Content', 207 => 'Multi-Status', 226 => 'IM Used', 300 => 'Multiple Choices', 301 => 'Moved Permanently', 302 => 'Found', 303 => 'See Other', 304 => 'Not Modified', 305 => 'Use Proxy', 306 => 'Reserved', 307 => 'Temporary Redirect', 400 => 'Bad Request', 401 => 'Unauthorized', 402 => 'Payment Required', 403 => 'Forbidden', 404 => 'Not Found', 405 => 'Method Not Allowed', 406 => 'Not Acceptable', 407 => 'Proxy Authentication Required', 408 => 'Request Timeout', 409 => 'Conflict', 410 => 'Gone', 411 => 'Length Required', 412 => 'Precondition Failed', 413 => 'Request Entity Too Large', 414 => 'Request-URI Too Long', 415 => 'Unsupported Media Type', 416 => 'Requested Range Not Satisfiable', 417 => 'Expectation Failed', 422 => 'Unprocessable Entity', 423 => 'Locked', 424 => 'Failed Dependency', 426 => 'Upgrade Required', 500 => 'Internal Server Error', 501 => 'Not Implemented', 502 => 'Bad Gateway', 503 => 'Service Unavailable', 504 => 'Gateway Timeout', 505 => 'HTTP Version Not Supported', 506 => 'Variant Also Negotiates', 507 => 'Insufficient Storage', 510 => 'Not Extended' ); if ( isset( $headers_desc[$code] ) ) return $headers_desc[$code]; else return ''; } // Log a redirect (for stats) function yourls_log_redirect( $keyword ) { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_log_redirect', false, $keyword ); if ( false !== $pre ) return $pre; if ( !yourls_do_log_redirect() ) return true; global $ydb; $table = YOURLS_DB_TABLE_LOG; $keyword = yourls_sanitize_string( $keyword ); $referrer = ( isset( $_SERVER['HTTP_REFERER'] ) ? yourls_sanitize_url( $_SERVER['HTTP_REFERER'] ) : 'direct' ); $ua = yourls_get_user_agent(); $ip = yourls_get_IP(); $location = yourls_geo_ip_to_countrycode( $ip ); return $ydb->query( "INSERT INTO `$table` (click_time, shorturl, referrer, user_agent, ip_address, country_code) VALUES (NOW(), '$keyword', '$referrer', '$ua', '$ip', '$location')" ); } // Check if we want to not log redirects (for stats) function yourls_do_log_redirect() { return ( !defined('YOURLS_NOSTATS') || YOURLS_NOSTATS != true ); } // Converts an IP to a 2 letter country code, using GeoIP database if available in includes/geo/ function yourls_geo_ip_to_countrycode( $ip = '', $default = '' ) { // Allow plugins to short-circuit the Geo IP API $location = yourls_apply_filter( 'shunt_geo_ip_to_countrycode', false, $ip, $default ); // at this point $ip can be '', check if your plugin hooks in here if ( false !== $location ) return $location; if ( !file_exists( YOURLS_INC.'/geo/GeoIP.dat') || !file_exists( YOURLS_INC.'/geo/geoip.inc') ) return $default; if ( $ip == '' ) $ip = yourls_get_IP(); require_once( YOURLS_INC.'/geo/geoip.inc') ; $gi = geoip_open( YOURLS_INC.'/geo/GeoIP.dat', GEOIP_STANDARD); $location = geoip_country_code_by_addr($gi, $ip); geoip_close($gi); return yourls_apply_filter( 'geo_ip_to_countrycode', $location, $ip, $default ); } // Converts a 2 letter country code to long name (ie AU -> Australia) function yourls_geo_countrycode_to_countryname( $code ) { // Allow plugins to short-circuit the Geo IP API $country = yourls_apply_filter( 'shunt_geo_countrycode_to_countryname', false, $code ); if ( false !== $country ) return $country; // Load the Geo class if not already done if( !class_exists('GeoIP') ) { $temp = yourls_geo_ip_to_countrycode('127.0.0.1'); } if( class_exists('GeoIP') ) { $geo = new GeoIP; $id = $geo->GEOIP_COUNTRY_CODE_TO_NUMBER[$code]; $long = $geo->GEOIP_COUNTRY_NAMES[$id]; return $long; } else { return false; } } // Return flag URL from 2 letter country code function yourls_geo_get_flag( $code ) { if( file_exists( YOURLS_INC.'/geo/flags/flag_'.strtolower($code).'.gif' ) ) { $img = yourls_match_current_protocol( YOURLS_SITE.'/includes/geo/flags/flag_'.(strtolower($code)).'.gif' ); } else { $img = false; } return yourls_apply_filter( 'geo_get_flag', $img, $code ); } // Check if an upgrade is needed function yourls_upgrade_is_needed() { // check YOURLS_DB_VERSION exist && match values stored in YOURLS_DB_TABLE_OPTIONS list( $currentver, $currentsql ) = yourls_get_current_version_from_sql(); if( $currentsql < YOURLS_DB_VERSION ) return true; return false; } // Get current version & db version as stored in the options DB. Prior to 1.4 there's no option table. function yourls_get_current_version_from_sql() { $currentver = yourls_get_option( 'version' ); $currentsql = yourls_get_option( 'db_version' ); // Values if version is 1.3 if( !$currentver ) $currentver = '1.3'; if( !$currentsql ) $currentsql = '100'; return array( $currentver, $currentsql); } // Read an option from DB (or from cache if available). Return value or $default if not found function yourls_get_option( $option_name, $default = false ) { global $ydb; // Allow plugins to short-circuit options $pre = yourls_apply_filter( 'shunt_option_'.$option_name, false ); if ( false !== $pre ) return $pre; // If option not cached already, get its value from the DB if ( !isset( $ydb->option[$option_name] ) ) { $table = YOURLS_DB_TABLE_OPTIONS; $option_name = yourls_escape( $option_name ); $row = $ydb->get_row( "SELECT `option_value` FROM `$table` WHERE `option_name` = '$option_name' LIMIT 1" ); if ( is_object( $row) ) { // Has to be get_row instead of get_var because of funkiness with 0, false, null values $value = $row->option_value; } else { // option does not exist, so we must cache its non-existence $value = $default; } $ydb->option[$option_name] = yourls_maybe_unserialize( $value ); } return yourls_apply_filter( 'get_option_'.$option_name, $ydb->option[$option_name] ); } // Read all options from DB at once function yourls_get_all_options() { global $ydb; // Allow plugins to short-circuit all options. (Note: regular plugins are loaded after all options) $pre = yourls_apply_filter( 'shunt_all_options', false ); if ( false !== $pre ) return $pre; $table = YOURLS_DB_TABLE_OPTIONS; $allopt = $ydb->get_results("SELECT `option_name`, `option_value` FROM `$table` WHERE 1=1"); foreach( (array)$allopt as $option ) { $ydb->option[$option->option_name] = yourls_maybe_unserialize( $option->option_value ); } $ydb->option = yourls_apply_filter( 'get_all_options', $ydb->option ); } // Update (add if doesn't exist) an option to DB function yourls_update_option( $option_name, $newvalue ) { global $ydb; $table = YOURLS_DB_TABLE_OPTIONS; $safe_option_name = yourls_escape( $option_name ); $oldvalue = yourls_get_option( $safe_option_name ); // If the new and old values are the same, no need to update. if ( $newvalue === $oldvalue ) return false; if ( false === $oldvalue ) { yourls_add_option( $option_name, $newvalue ); return true; } $_newvalue = yourls_escape( yourls_maybe_serialize( $newvalue ) ); yourls_do_action( 'update_option', $option_name, $oldvalue, $newvalue ); $ydb->query( "UPDATE `$table` SET `option_value` = '$_newvalue' WHERE `option_name` = '$option_name'"); if ( $ydb->rows_affected == 1 ) { $ydb->option[$option_name] = $newvalue; return true; } return false; } // Add an option to the DB function yourls_add_option( $name, $value = '' ) { global $ydb; $table = YOURLS_DB_TABLE_OPTIONS; $safe_name = yourls_escape( $name ); // Make sure the option doesn't already exist if ( false !== yourls_get_option( $safe_name ) ) return; $_value = yourls_escape( yourls_maybe_serialize( $value ) ); yourls_do_action( 'add_option', $safe_name, $_value ); $ydb->query( "INSERT INTO `$table` (`option_name`, `option_value`) VALUES ('$name', '$_value')" ); $ydb->option[$name] = $value; return; } // Delete an option from the DB function yourls_delete_option( $name ) { global $ydb; $table = YOURLS_DB_TABLE_OPTIONS; $name = yourls_escape( $name ); // Get the ID, if no ID then return $option = $ydb->get_row( "SELECT option_id FROM `$table` WHERE `option_name` = '$name'" ); if ( is_null($option) || !$option->option_id ) return false; yourls_do_action( 'delete_option', $option_name ); $ydb->query( "DELETE FROM `$table` WHERE `option_name` = '$name'" ); return true; } // Serialize data if needed. Stolen from WordPress function yourls_maybe_serialize( $data ) { if ( is_array( $data ) || is_object( $data ) ) return serialize( $data ); if ( yourls_is_serialized( $data ) ) return serialize( $data ); return $data; } // Check value to find if it was serialized. Stolen from WordPress function yourls_is_serialized( $data ) { // if it isn't a string, it isn't serialized if ( !is_string( $data ) ) return false; $data = trim( $data ); if ( 'N;' == $data ) return true; if ( !preg_match( '/^([adObis]):/', $data, $badions ) ) return false; switch ( $badions[1] ) { case 'a' : case 'O' : case 's' : if ( preg_match( "/^{$badions[1]}:[0-9]+:.*[;}]\$/s", $data ) ) return true; break; case 'b' : case 'i' : case 'd' : if ( preg_match( "/^{$badions[1]}:[0-9.E-]+;\$/", $data ) ) return true; break; } return false; } // Unserialize value only if it was serialized. Stolen from WP function yourls_maybe_unserialize( $original ) { if ( yourls_is_serialized( $original ) ) // don't attempt to unserialize data that wasn't serialized going in return @unserialize( $original ); return $original; } // Determine if the current page is private function yourls_is_private() { $private = false; if ( defined('YOURLS_PRIVATE') && YOURLS_PRIVATE == true ) { // Allow overruling for particular pages: // API if( yourls_is_API() ) { if( !defined('YOURLS_PRIVATE_API') || YOURLS_PRIVATE_API != false ) $private = true; // Infos } elseif( yourls_is_infos() ) { if( !defined('YOURLS_PRIVATE_INFOS') || YOURLS_PRIVATE_INFOS !== false ) $private = true; // Others } else { $private = true; } } return yourls_apply_filter( 'is_private', $private ); } // Show login form if required function yourls_maybe_require_auth() { if( yourls_is_private() ) require_once( YOURLS_INC.'/auth.php' ); } // Allow several short URLs for the same long URL ? function yourls_allow_duplicate_longurls() { // special treatment if API to check for WordPress plugin requests if( yourls_is_API() ) { if ( isset($_REQUEST['source']) && $_REQUEST['source'] == 'plugin' ) return false; } return ( defined( 'YOURLS_UNIQUE_URLS' ) && YOURLS_UNIQUE_URLS == false ); } // Return list of all shorturls associated to the same long URL. Returns NULL or array of keywords. function yourls_get_duplicate_keywords( $longurl ) { if( !yourls_allow_duplicate_longurls() ) return NULL; global $ydb; $longurl = yourls_escape( yourls_sanitize_url($longurl) ); $table = YOURLS_DB_TABLE_URL; $return = $ydb->get_col( "SELECT `keyword` FROM `$table` WHERE `url` = '$longurl'" ); return yourls_apply_filter( 'get_duplicate_keywords', $return, $longurl ); } // Check if an IP shortens URL too fast to prevent DB flood. Return true, or die. function yourls_check_IP_flood( $ip = '' ) { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_check_IP_flood', false, $ip ); if ( false !== $pre ) return $pre; yourls_do_action( 'pre_check_ip_flood', $ip ); // at this point $ip can be '', check it if your plugin hooks in here if( ( defined('YOURLS_FLOOD_DELAY_SECONDS') && YOURLS_FLOOD_DELAY_SECONDS === 0 ) || !defined('YOURLS_FLOOD_DELAY_SECONDS') ) return true; $ip = ( $ip ? yourls_sanitize_ip( $ip ) : yourls_get_IP() ); // Don't throttle whitelist IPs if( defined( 'YOURLS_FLOOD_IP_WHITELIST' ) && YOURLS_FLOOD_IP_WHITELIST ) { $whitelist_ips = explode( ',', YOURLS_FLOOD_IP_WHITELIST ); foreach( (array)$whitelist_ips as $whitelist_ip ) { $whitelist_ip = trim( $whitelist_ip ); if ( $whitelist_ip == $ip ) return true; } } // Don't throttle logged in users if( yourls_is_private() ) { if( yourls_is_valid_user() === true ) return true; } yourls_do_action( 'check_ip_flood', $ip ); global $ydb; $table = YOURLS_DB_TABLE_URL; $lasttime = $ydb->get_var( "SELECT `timestamp` FROM $table WHERE `ip` = '$ip' ORDER BY `timestamp` DESC LIMIT 1" ); if( $lasttime ) { $now = date( 'U' ); $then = date( 'U', strtotime( $lasttime ) ); if( ( $now - $then ) <= YOURLS_FLOOD_DELAY_SECONDS ) { // Flood! yourls_do_action( 'ip_flood', $ip, $now - $then ); yourls_die( 'Too many URLs added too fast. Slow down please.', 'Forbidden', 403 ); } } return true; } // Check if YOURLS is installed function yourls_is_installed() { static $is_installed = false; if ( $is_installed === false ) { $check_14 = $check_13 = false; global $ydb; if( defined('YOURLS_DB_TABLE_NEXTDEC') ) $check_13 = $ydb->get_var('SELECT `next_id` FROM '.YOURLS_DB_TABLE_NEXTDEC); $check_14 = yourls_get_option( 'version' ); $is_installed = $check_13 || $check_14; } return yourls_apply_filter( 'is_installed', $is_installed ); } // Generate random string of (int)$length length and type $type (see function for details) function yourls_rnd_string ( $length = 5, $type = 0, $charlist = '' ) { $str = ''; $length = intval( $length ); // define possible characters switch ( $type ) { // custom char list, or comply to charset as defined in config case '0': $possible = $charlist ? $charlist : yourls_get_shorturl_charset() ; break; // no vowels to make no offending word, no 0/1/o/l to avoid confusion between letters & digits. Perfect for passwords. case '1': $possible = "23456789bcdfghjkmnpqrstvwxyz"; break; // Same, with lower + upper case '2': $possible = "23456789bcdfghjkmnpqrstvwxyzBCDFGHJKMNPQRSTVWXYZ"; break; // all letters, lowercase case '3': $possible = "abcdefghijklmnopqrstuvwxyz"; break; // all letters, lowercase + uppercase case '4': $possible = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"; break; // all digits & letters lowercase case '5': $possible = "0123456789abcdefghijklmnopqrstuvwxyz"; break; // all digits & letters lowercase + uppercase case '6': $possible = "0123456789abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"; break; } $i = 0; while ($i < $length) { $str .= substr($possible, mt_rand(0, strlen($possible)-1), 1); $i++; } return yourls_apply_filter( 'rnd_string', $str, $length, $type, $charlist ); } // Return salted string function yourls_salt( $string ) { $salt = defined('YOURLS_COOKIEKEY') ? YOURLS_COOKIEKEY : md5(__FILE__) ; return yourls_apply_filter( 'yourls_salt', md5 ($string . $salt), $string ); } // Add a query var to a URL and return URL. Completely stolen from WP. // Works with one of these parameter patterns: // array( 'var' => 'value' ) // array( 'var' => 'value' ), $url // 'var', 'value' // 'var', 'value', $url // If $url ommited, uses $_SERVER['REQUEST_URI'] function yourls_add_query_arg() { $ret = ''; if ( is_array( func_get_arg(0) ) ) { if ( @func_num_args() < 2 || false === @func_get_arg( 1 ) ) $uri = $_SERVER['REQUEST_URI']; else $uri = @func_get_arg( 1 ); } else { if ( @func_num_args() < 3 || false === @func_get_arg( 2 ) ) $uri = $_SERVER['REQUEST_URI']; else $uri = @func_get_arg( 2 ); } $uri = str_replace( '&', '&', $uri ); if ( $frag = strstr( $uri, '#' ) ) $uri = substr( $uri, 0, -strlen( $frag ) ); else $frag = ''; if ( preg_match( '|^https?://|i', $uri, $matches ) ) { $protocol = $matches[0]; $uri = substr( $uri, strlen( $protocol ) ); } else { $protocol = ''; } if ( strpos( $uri, '?' ) !== false ) { $parts = explode( '?', $uri, 2 ); if ( 1 == count( $parts ) ) { $base = '?'; $query = $parts[0]; } else { $base = $parts[0] . '?'; $query = $parts[1]; } } elseif ( !empty( $protocol ) || strpos( $uri, '=' ) === false ) { $base = $uri . '?'; $query = ''; } else { $base = ''; $query = $uri; } parse_str( $query, $qs ); $qs = yourls_urlencode_deep( $qs ); // this re-URL-encodes things that were already in the query string if ( is_array( func_get_arg( 0 ) ) ) { $kayvees = func_get_arg( 0 ); $qs = array_merge( $qs, $kayvees ); } else { $qs[func_get_arg( 0 )] = func_get_arg( 1 ); } foreach ( (array) $qs as $k => $v ) { if ( $v === false ) unset( $qs[$k] ); } $ret = http_build_query( $qs ); $ret = trim( $ret, '?' ); $ret = preg_replace( '#=(&|$)#', '$1', $ret ); $ret = $protocol . $base . $ret . $frag; $ret = rtrim( $ret, '?' ); return $ret; } // Navigates through an array and encodes the values to be used in a URL. Stolen from WP, used in yourls_add_query_arg() function yourls_urlencode_deep($value) { $value = is_array($value) ? array_map('yourls_urlencode_deep', $value) : urlencode($value); return $value; } // Remove arg from query. Opposite of yourls_add_query_arg. Stolen from WP. function yourls_remove_query_arg( $key, $query = false ) { if ( is_array( $key ) ) { // removing multiple keys foreach ( $key as $k ) $query = yourls_add_query_arg( $k, false, $query ); return $query; } return yourls_add_query_arg( $key, false, $query ); } // Return a time-dependent string for nonce creation function yourls_tick() { return ceil( time() / YOURLS_NONCE_LIFE ); } // Create a time limited, action limited and user limited token function yourls_create_nonce( $action, $user = false ) { if( false == $user ) $user = defined('YOURLS_USER') ? YOURLS_USER : '-1'; $tick = yourls_tick(); return substr( yourls_salt($tick . $action . $user), 0, 10 ); } // Create a nonce field for inclusion into a form function yourls_nonce_field( $action, $name = 'nonce', $user = false, $echo = true ) { $field = ''; if( $echo ) echo $field."\n"; return $field; } // Add a nonce to a URL. If URL omitted, adds nonce to current URL function yourls_nonce_url( $action, $url = false, $name = 'nonce', $user = false ) { $nonce = yourls_create_nonce( $action, $user ); return yourls_add_query_arg( $name, $nonce, $url ); } // Check validity of a nonce (ie time span, user and action match). // Returns true if valid, dies otherwise (yourls_die() or die($return) if defined) // if $nonce is false or unspecified, it will use $_REQUEST['nonce'] function yourls_verify_nonce( $action, $nonce = false, $user = false, $return = '' ) { // get user if( false == $user ) $user = defined('YOURLS_USER') ? YOURLS_USER : '-1'; // get current nonce value if( false == $nonce && isset( $_REQUEST['nonce'] ) ) $nonce = $_REQUEST['nonce']; // what nonce should be $valid = yourls_create_nonce( $action, $user ); if( $nonce == $valid ) { return true; } else { if( $return ) die( $return ); yourls_die( 'Unauthorized action or expired link', 'Error', 403 ); } } // Converts keyword into short link (prepend with YOURLS base URL) function yourls_link( $keyword = '' ) { $link = YOURLS_SITE . '/' . yourls_sanitize_keyword( $keyword ); return yourls_apply_filter( 'yourls_link', $link, $keyword ); } // Converts keyword into stat link (prepend with YOURLS base URL, append +) function yourls_statlink( $keyword = '' ) { $link = YOURLS_SITE . '/' . yourls_sanitize_keyword( $keyword ) . '+'; return yourls_apply_filter( 'yourls_statlink', $link, $keyword ); } // Check if we're in API mode. Returns bool function yourls_is_API() { if ( defined('YOURLS_API') && YOURLS_API == true ) return true; return false; } // Check if we're in Ajax mode. Returns bool function yourls_is_Ajax() { if ( defined('YOURLS_AJAX') && YOURLS_AJAX == true ) return true; return false; } // Check if we're in GO mode (yourls-go.php). Returns bool function yourls_is_GO() { if ( defined('YOURLS_GO') && YOURLS_GO == true ) return true; return false; } // Check if we're displaying stats infos (yourls-infos.php). Returns bool function yourls_is_infos() { if ( defined('YOURLS_INFOS') && YOURLS_INFOS == true ) return true; return false; } // Check if we'll need interface display function (ie not API or redirection) function yourls_has_interface() { if( yourls_is_API() or yourls_is_GO() ) return false; return true; } // Check if we're in the admin area. Returns bool function yourls_is_admin() { if ( defined('YOURLS_ADMIN') && YOURLS_ADMIN == true ) return true; return false; } // Check if the server seems to be running on Windows. Not exactly sure how reliable this is. function yourls_is_windows() { return defined( 'DIRECTORY_SEPARATOR' ) && DIRECTORY_SEPARATOR == '\\'; } // Check if SSL is required. Returns bool. function yourls_needs_ssl() { if ( defined('YOURLS_ADMIN_SSL') && YOURLS_ADMIN_SSL == true ) return true; return false; } // Return admin link, with SSL preference if applicable. function yourls_admin_url( $page = '' ) { $admin = YOURLS_SITE . '/admin/' . $page; if( yourls_is_ssl() or yourls_needs_ssl() ) $admin = str_replace('http://', 'https://', $admin); return yourls_apply_filter( 'admin_url', $admin, $page ); } // Return YOURLS_SITE, with SSL preference function yourls_site_url( $echo = true ) { $site = YOURLS_SITE; // Do not enforce (checking yourls_need_ssl() ) but check current usage so it won't force SSL on non-admin pages if( yourls_is_ssl() ) $site = str_replace( 'http://', 'https://', $site ); $site = yourls_apply_filter( 'site_url', $site ); if( $echo ) echo $site; return $site; } // Check if SSL is used, returns bool. Stolen from WP. function yourls_is_ssl() { $is_ssl = false; if ( isset($_SERVER['HTTPS']) ) { if ( 'on' == strtolower($_SERVER['HTTPS']) ) $is_ssl = true; if ( '1' == $_SERVER['HTTPS'] ) $is_ssl = true; } elseif ( isset($_SERVER['SERVER_PORT']) && ( '443' == $_SERVER['SERVER_PORT'] ) ) { $is_ssl = true; } return yourls_apply_filter( 'is_ssl', $is_ssl ); } // Get a remote page , return a string (either title or url) function yourls_get_remote_title( $url ) { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_get_remote_title', false, $url ); if ( false !== $pre ) return $pre; require_once( YOURLS_INC.'/functions-http.php' ); $url = yourls_sanitize_url( $url ); $title = $charset = false; $content = yourls_get_remote_content( $url ); // If false, return url as title. // Todo: improve this with temporary title when shorturl_meta available? if( false === $content ) return $url; if( $content !== false ) { // look for <title> if ( preg_match('/<title>(.*?)<\/title>/is', $content, $found ) ) { $title = $found[1]; unset( $found ); } // look for charset // <meta http-equiv="Content-Type" content="text/html; charset=UTF-8" /> if ( preg_match('/<meta[^>]*?charset=([^>]*?)\/?>/is', $content, $found ) ) { $charset = trim($found[1], '"\' '); unset( $found ); } } // if title not found, guess if returned content was actually an error message if( $title == false && strpos( $content, 'Error' ) === 0 ) { $title = $content; } if( $title == false ) $title = $url; /* if( !yourls_seems_utf8( $title ) ) $title = utf8_encode( $title ); */ // Charset conversion. We use @ to remove warnings (mb_ functions are easily bitching about illegal chars) if( function_exists( 'mb_convert_encoding' ) ) { if( $charset ) { $title = @mb_convert_encoding( $title, 'UTF-8', $charset ); } else { $title = @mb_convert_encoding( $title, 'UTF-8' ); } } // Remove HTML entities $title = html_entity_decode( $title, ENT_QUOTES, 'UTF-8' ); // Strip out evil things $title = yourls_sanitize_title( $title ); return yourls_apply_filter( 'get_remote_title', $title, $url ); } // Check for maintenance mode that will shortcut everything function yourls_check_maintenance_mode() { // TODO: all cases that always display the sites (is_admin but not is_ajax?) if( 1 ) return; // first case: /user/maintenance.php file if( file_exists( YOURLS_USERDIR.'/maintenance.php' ) ) { include( YOURLS_USERDIR.'/maintenance.php' ); die(); } // second case: option in DB if( yourls_get_option( 'maintenance_mode' ) !== false ) { require_once( YOURLS_INC.'/functions-html.php' ); $title = 'Service temporarily unavailable'; $message = 'Our service is currently undergoing scheduled maintenance.</p> <p>Things should not last very long, thank you for your patience and please excuse the inconvenience'; yourls_die( $message, $title , 503 ); } } // Toggle maintenance mode function yourls_maintenance_mode( $maintenance = true ) { yourls_update_option( 'maintenance_mode', (bool)$maintenance ); } // Quick UA check for mobile devices. Return boolean. function yourls_is_mobile_device() { // Strings searched $mobiles = array( 'android', 'blackberry', 'blazer', 'compal', 'elaine', 'fennec', 'hiptop', 'iemobile', 'iphone', 'ipod', 'ipad', 'iris', 'kindle', 'opera mobi', 'opera mini', 'palm', 'phone', 'pocket', 'psp', 'symbian', 'treo', 'wap', 'windows ce', 'windows phone' ); // Current user-agent $current = strtolower( $_SERVER['HTTP_USER_AGENT'] ); // Check and return $is_mobile = ( str_replace( $mobiles, '', $current ) != $current ); return yourls_apply_filter( 'is_mobile_device', $is_mobile ); } // Get request in YOURLS base (eg in 'http://site.com/yourls/abcd' get 'abdc') function yourls_get_request() { // Allow plugins to short-circuit the whole function $pre = yourls_apply_filter( 'shunt_get_request', false ); if ( false !== $pre ) return $pre; yourls_do_action( 'pre_get_request' ); // Ignore protocol, www. prefix and extract keyword $base = str_replace( array( 'https://', 'http://', 'https://www.', 'http://www.' ), '', YOURLS_SITE ); $request = str_replace( $base.'/', '', $_SERVER['HTTP_HOST'] . $_SERVER['REQUEST_URI'] ); // Strip query string from keyword $request = current( explode( '?', $request ) ); return yourls_apply_filter( 'get_request', $request ); } // Change protocol to match current scheme used (http or https) function yourls_match_current_protocol( $url, $normal = 'http', $ssl = 'https' ) { if( yourls_is_ssl() ) $url = str_replace( $normal, $ssl, $url ); return yourls_apply_filter( 'match_current_protocol', $url ); } // Fix $_SERVER['REQUEST_URI'] variable for various setups. Stolen from WP. function yourls_fix_request_uri() { $default_server_values = array( 'SERVER_SOFTWARE' => '', 'REQUEST_URI' => '', ); $_SERVER = array_merge( $default_server_values, $_SERVER ); // Fix for IIS when running with PHP ISAPI if ( empty( $_SERVER['REQUEST_URI'] ) || ( php_sapi_name() != 'cgi-fcgi' && preg_match( '/^Microsoft-IIS\//', $_SERVER['SERVER_SOFTWARE'] ) ) ) { // IIS Mod-Rewrite if ( isset( $_SERVER['HTTP_X_ORIGINAL_URL'] ) ) { $_SERVER['REQUEST_URI'] = $_SERVER['HTTP_X_ORIGINAL_URL']; } // IIS Isapi_Rewrite else if ( isset( $_SERVER['HTTP_X_REWRITE_URL'] ) ) { $_SERVER['REQUEST_URI'] = $_SERVER['HTTP_X_REWRITE_URL']; } else { // Use ORIG_PATH_INFO if there is no PATH_INFO if ( !isset( $_SERVER['PATH_INFO'] ) && isset( $_SERVER['ORIG_PATH_INFO'] ) ) $_SERVER['PATH_INFO'] = $_SERVER['ORIG_PATH_INFO']; // Some IIS + PHP configurations puts the script-name in the path-info (No need to append it twice) if ( isset( $_SERVER['PATH_INFO'] ) ) { if ( $_SERVER['PATH_INFO'] == $_SERVER['SCRIPT_NAME'] ) $_SERVER['REQUEST_URI'] = $_SERVER['PATH_INFO']; else $_SERVER['REQUEST_URI'] = $_SERVER['SCRIPT_NAME'] . $_SERVER['PATH_INFO']; } // Append the query string if it exists and isn't null if ( ! empty( $_SERVER['QUERY_STRING'] ) ) { $_SERVER['REQUEST_URI'] .= '?' . $_SERVER['QUERY_STRING']; } } } } // Shutdown function, runs just before PHP shuts down execution. Stolen from WP function yourls_shutdown() { yourls_do_action( 'shutdown' ); }