Modifiying the Blacklist IP plugin to create the whitelisted domain plugin

This commit is contained in:
Panthro
2012-09-09 20:06:57 -03:00
parent 621c359ad4
commit a9cdb85b5c
2 changed files with 57 additions and 56 deletions

View File

@ -1,69 +1,73 @@
<?php
/*
Plugin Name: BlackListIP
Plugin URI: https://github.com/LudoBoggio/YourlsBlackListIPs
Description: Plugin which block blacklisted IPs
Version: 1.1
Author: Ludo
Author URI: http://ludo.boggio.fr
Plugin Name: WhiteListDomains
Plugin URI: https://github.com/Panthro/YourlsWhitelistDomains
Description: Plugin which allow only whitelisted domains, forked from: https://github.com/LudoBoggio/YourlsBlacklistIPs
Version: 1.0
Author: Panthro
Author URI: http://passeionaweb.com.br
*/
// No direct call
if( !defined( 'YOURLS_ABSPATH' ) ) die();
// Hook the custom function into the 'pre_check_ip_flood' event
yourls_add_action( 'pre_check_ip_flood', 'ludo_blacklist_ip_root' );
// Hook the custom function into the 'pre_check_domain_flood' event
yourls_add_filter( 'shunt_add_new_link', 'panthro_whitelist_domain_root' );
// Hook the admin page into the 'plugins_loaded' event
yourls_add_action( 'plugins_loaded', 'ludo_blacklist_ip_add_page' );
yourls_add_action( 'plugins_loaded', 'panthro_whitelist_domain_add_page' );
// Get blacklisted IPs from YOURLS options feature and compare with current IP address
function ludo_blacklist_ip_root ( $args ) {
$ip = $args[0];
$liste_ip = yourls_get_option ('ludo_blacklist_ip_liste');
if ( !$liste_ip ) {
$liste_ip_display = unserialize ( $liste_ip );
if (in_array($ip, $liste_ip_display)) {
// yourls_die ( "Your IP has been blacklisted.", "Black list",403);
echo "<center>Your IP has been blacklisted.</center>";
die();
// Get whitelisted domains from YOURLS options feature and compare with current domain address
function panthro_whitelist_domain_root ( $bol, $url ) {
$parse = parse_url($url);
$domain = str_ireplace('www.', '', parse_url($url, PHP_URL_HOST));
$return = false;
$domain_list = yourls_get_option ('panthro_whitelist_domain_list');
if ( $domain_list ) {
$domain_list_display = unserialize ( $domain_list );
//if (!in_array($domain, $domain_list_display)) {
if (strpos($domain_list_display,$domain) === false) {
$return['status'] = 'fail';
$return['code'] = 'error:domain-not-allowed';
$return['message'] = 'This domain is not allowed';
$return['errorCode'] = '400';
}
}
return $return;
}
// Add admin page
function ludo_blacklist_ip_add_page () {
yourls_register_plugin_page( 'ludo_blacklist_ip', 'Blacklist IPs', 'ludo_blacklist_ip_do_page' );
function panthro_whitelist_domain_add_page () {
yourls_register_plugin_page( 'panthro_whitelist_domain', 'Whitelist domains', 'panthro_whitelist_domain_do_page' );
}
// Display admin page
function ludo_blacklist_ip_do_page () {
if( isset( $_POST['action'] ) && $_POST['action'] == 'blacklist_ip' ) {
ludo_blacklist_ip_process ();
function panthro_whitelist_domain_do_page () {
if( isset( $_POST['action'] ) && $_POST['action'] == 'whitelist_domain' ) {
panthro_whitelist_domain_process ();
} else {
ludo_blacklist_ip_form ();
panthro_whitelist_domain_form ();
}
}
// Display form to administrate blacklisted IPs list
function ludo_blacklist_ip_form () {
$nonce = yourls_create_nonce( 'blacklist_ip' ) ;
$liste_ip = yourls_get_option ('ludo_blacklist_ip_liste','Enter IP addresses here, one per line');
if ($liste_ip != 'Enter IP addresses here, one per line' )
$liste_ip_display = implode ( "\n" , unserialize ( $liste_ip ) );
// Display form to administrate whitelisted domains list
function panthro_whitelist_domain_form () {
$nonce = yourls_create_nonce( 'whitelist_domain' ) ;
$domain_list = yourls_get_option ('panthro_whitelist_domain_list','Enter domain addresses here, one per line');
if ($domain_list != 'Enter domain addresses here, one per line' ){
$domain_list_display = unserialize ( $domain_list );
}else{
$domain_list_display = $domain_list;
}
echo <<<HTML
<h2> BlackList IPs</h2>
<h2> WhiteList domains</h2>
<form method="post">
<input type="hidden" name="action" value="blacklist_ip" />
<input type="hidden" name="action" value="whitelist_domain" />
<input type="hidden" name="nonce" value="$nonce" />
<p>Blacklist following IPs
<textarea cols="30" rows="5" name="blacklist_form">
$liste_ip_display
</textarea>
<p>Whitelist following domains
<textarea cols="30" rows="5" name="whitelist_form">$domain_list_display</textarea>
</p>
<p><input type="submit" value="Save" /></p>
@ -71,14 +75,13 @@ function ludo_blacklist_ip_form () {
HTML;
}
// Update blacklisted IPs list
function ludo_blacklist_ip_process () {
// Update whitelisted domains list
function panthro_whitelist_domain_process () {
// Check nonce
yourls_verify_nonce( 'blacklist_ip' ) ;
yourls_verify_nonce( 'whitelist_domain' ) ;
// Update list
$sent_list = serialize ( explode ( '\n' , $_POST['blacklist_form'] ) );
yourls_update_option ( $sent_list , 'ludo_blacklist_ip_liste' );
echo "Black list updated" ;
$sent_list = serialize($_POST['whitelist_form']);
yourls_update_option ( 'panthro_whitelist_domain_list',$sent_list );
echo "White list updated" ;
}
?>
?>