package com.walker.security;
|
|
//import org.slf4j.Logger;
|
//import org.slf4j.LoggerFactory;
|
|
import java.nio.ByteBuffer;
|
import java.util.List;
|
import java.util.concurrent.TimeUnit;
|
|
public class SystemLogMan {
|
|
// private final transient Logger logger = LoggerFactory.getLogger(getClass());
|
|
private static final SpiderLoader loader = new DefaultSpiderLoader();
|
|
private static final Man man = new Man();
|
|
private static final SystemLogMan slm = new SystemLogMan();
|
|
private static final String LOCALHOST_1 = "127.0.0.1";
|
private static final String LOCALHOST_2 = "localhost";
|
|
|
private SystemLogMan(){}
|
|
public static final SystemLogMan getInstance(){
|
return slm;
|
}
|
|
public final void checkMan(String host){
|
// 如果是开发环境，本地测试，不再检查注册
|
if(host != null && (host.indexOf(LOCALHOST_1) >= 0 || host.indexOf(LOCALHOST_2) >= 0)){
|
return;
|
}
|
doManCheck();
|
}
|
|
/**
|
* 真正判断授权的方法
|
*/
|
public final void checkMan(){
|
doCheckMan2();
|
}
|
|
private final void doCheckMan2(){
|
man.checkMan();
|
}
|
|
private final void doManCheck(){
|
if(man.manStatus){
|
// do nothing...
|
}
|
Object[] result = (Object[])loader.load();
|
if(result == null || result.length != 2){
|
man.manStatus = false;
|
man.getEating();
|
}
|
// 从用户配置文件中返回协议结果
|
int status = (Integer)result[0];
|
if(status == 1){
|
man.setMan((byte[])result[1]);
|
if(!man.manStatus){
|
man.getEating();
|
}
|
} else {
|
System.out.println("-------- load sysLogMan failed!");
|
man.getEating();
|
}
|
}
|
|
private static class Man{
|
|
private boolean manStatus = false;
|
|
private LocalSpider spider = new DefaultLocalSpider();
|
|
/**
|
* 设置协议内容。</p>
|
* Mac地址\r\n公匙
|
* @param content
|
*/
|
public void setMan(byte[] content){
|
doCheckMan(content);
|
}
|
|
private void doCheckMan(byte[] content){
|
if(content == null || content.length <= 0){
|
return;
|
}
|
int size = content.length;
|
// System.out.println("读取的文件长度 = " + size);
|
|
// 消息内容
|
ByteBuffer messageBuffer = ByteBuffer.allocate(1024);
|
|
ByteBuffer buffer = ByteBuffer.wrap(content);
|
byte temp;
|
for(int i=0; i<size; i++){
|
temp = buffer.get();
|
if(((char)temp) != '\n'){
|
messageBuffer.put(temp);
|
} else {
|
break;
|
}
|
}
|
messageBuffer.flip();
|
byte[] message = new byte[messageBuffer.remaining()];
|
messageBuffer.get(message);
|
display("消息体", message);
|
|
// 公匙内容
|
byte[] publicKeys = new byte[buffer.remaining()];
|
// System.out.println("剩余公匙长度，publicKeys: " + publicKeys.length);
|
buffer.get(publicKeys);
|
display("pk", publicKeys);
|
|
try {
|
byte[] addressBytes = RSAUtils.decryptByPublicKey(message, new String(publicKeys));
|
String source = spider.getMyIdentifier();
|
String now = new String(addressBytes);
|
// System.out.println("source = " + source + ", now = " + now);
|
if(source != null && now != null && source.equals(now)){
|
// System.out.println("=========== ok!");
|
manStatus = true;
|
}
|
} catch (Exception e) {
|
e.printStackTrace();
|
return;
|
}
|
}
|
|
private void getEating(){
|
try {
|
TimeUnit.SECONDS.sleep(3);
|
System.out.println("------> my walker ea logging...");
|
} catch (InterruptedException e) {
|
e.printStackTrace();
|
}
|
}
|
|
private void display(String msg, byte[] bytes){
|
// System.out.println(msg + "长度: " + bytes.length);
|
// System.out.println(msg + "内容: " + new String(bytes));
|
}
|
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
// 第二次修改的方法，只有下面的方法是真正的方法。
|
//~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
|
private int type = 0;
|
private String address;
|
private long st = 0;
|
private long et = 0;
|
|
public void setMan(String[] array){
|
doSetMan(array);
|
}
|
|
private void doSetMan(String[] array){
|
if(array == null || array.length != 4){
|
throw new IllegalArgumentException();
|
}
|
type = Integer.parseInt(array[0].toString());
|
address = array[1];
|
st = Long.parseLong(array[2]);
|
et = Long.parseLong(array[3]);
|
manStatus = true;
|
}
|
|
public void checkMan(){
|
doCheckLisence();
|
}
|
|
private void doCheckLisence(){
|
if(manStatus){
|
// 已经加载过协议就直接读取，判断
|
} else {
|
// 还没有加载
|
Object[] result = (Object[])loader.load();
|
|
// for (Object o : result){
|
// System.out.println(o.toString());
|
// }
|
|
if(result == null || result.length != 2)
|
throw new SecurityRuntimeException();
|
if(!result[0].toString().equals("1")){
|
throw new SecurityRuntimeException();
|
}
|
this.setMan((String[])result[1]);
|
}
|
if(type == 0){
|
// 仅通过ip授权
|
doValidateAddress();
|
|
} else if(type == 1){
|
// 仅通过时间段授权
|
doValidateDate();
|
|
} else if(type == 2){
|
// 两种同时使用
|
doValidateAddress();
|
doValidateDate();
|
|
} else {
|
throw new SecurityRuntimeException("unknown authen type: " + type, null);
|
}
|
}
|
|
private boolean doValidateAddress(){
|
List<String> addresses = spider.getMyIdentifiers();
|
// System.out.println("........... addresses: " + addresses);
|
if(addresses != null && addresses.contains(address)){
|
// ok
|
return true;
|
} else {
|
throw new SecurityRuntimeException("address error:", null);
|
}
|
}
|
|
private boolean doValidateDate(){
|
long _cuTime = System.currentTimeMillis();
|
if(_cuTime > st && _cuTime < et){
|
return true;
|
} else {
|
throw new SecurityRuntimeException("date expired:", null);
|
}
|
}
|
}
|
|
// public static void main(String[] args){
|
// SystemLogMan.getInstance().checkMan();
|
// }
|
}
|
