(转）JAVA获得机器码的实现

http://yangshangchuan.iteye.com/blog/2012401

首先，定义了一个统一的接口，以支持不同操作系统不同实现的透明切换：

Java代码  收藏代码
/**
*生成机器码的接口，不同平台有不同实现
* @author 杨尚川
*/ 
public interface SequenceService { 
    /**
     * 获取机器码
     * @return  机器码
     */ 
    public String getSequence();     
} 


其次，定义了一个抽象类，实现了一些通用的功能，如生成MD5摘要、将很长的字符串以固定的位数分割开，以便于人类阅读（如将机器码 71F5DA7F495E7F706D47F3E63DC6349A 每4个一组，以-分割为71F5-DA7F-495E-7F70-6D47-F3E6-3DC6-349A）、利用sigar来生成机器码。这里需要在pom.xml中加入sigar依赖。

Xml代码  收藏代码
<dependency> 
    <groupId>org.fusesource</groupId> 
    <artifactId>sigar</artifactId> 
    <version>1.6.4</version> 
    <exclusions> 
        <exclusion> 
            <groupId>log4j</groupId> 
            <artifactId>log4j</artifactId> 
        </exclusion> 
    </exclusions> 
</dependency> 


Java代码  收藏代码
import org.apdplat.module.system.service.PropertyHolder; 
import org.apdplat.platform.log.APDPlatLogger; 
import org.apdplat.platform.util.ConvertUtils; 
import org.apdplat.platform.util.FileUtils; 
import java.io.File; 
import java.io.UnsupportedEncodingException; 
import java.security.MessageDigest; 
import java.security.NoSuchAlgorithmException; 
import java.util.HashSet; 
import java.util.Properties; 
import java.util.Set; 
import org.apdplat.platform.log.APDPlatLoggerFactory; 
import org.hyperic.sigar.Mem; 
import org.hyperic.sigar.NetFlags; 
import org.hyperic.sigar.NetInterfaceConfig; 
import org.hyperic.sigar.Sigar; 
 
/**
*机器码生成的通用服务
* @author 杨尚川
*/ 
public abstract class AbstractSequenceService   implements SequenceService{ 
    protected final APDPlatLogger LOG = APDPlatLoggerFactory.getAPDPlatLogger(getClass()); 
    /**
     * 对一段String生成MD5摘要信息
     * @param message 要摘要的String
     * @return 生成的MD5摘要信息
     */ 
    protected String getMD5(String message) { 
        message += "{apdplat}"; 
        try { 
            MessageDigest md = MessageDigest.getInstance("MD5"); 
            LOG.debug("MD5摘要长度：" + md.getDigestLength()); 
            byte[] b = md.digest(message.getBytes("utf-8")); 
            String md5 = ConvertUtils.byte2HexString(b)+message.length(); 
            return getSplitString(md5); 
        } catch (NoSuchAlgorithmException | UnsupportedEncodingException e) { 
            LOG.error("MD5摘要失败",e); 
        } 
        return null; 
    } 
    /**
     * 将很长的字符串以固定的位数分割开，以便于人类阅读
     * @param str
     * @return 
     */ 
    protected String getSplitString(String str){  
        return getSplitString(str, "-", 4); 
    } 
    /**
     * 将很长的字符串以固定的位数分割开，以便于人类阅读
     * 如将
     * 71F5DA7F495E7F706D47F3E63DC6349A
     * 以-，每4个一组，则分割为
     * 71F5-DA7F-495E-7F70-6D47-F3E6-3DC6-349A
     * @param str 字符串
     * @param split 分隔符
     * @param length 长度
     * @return 
     */ 
    protected String getSplitString(String str, String split, int length){         
        int len=str.length(); 
        StringBuilder temp=new StringBuilder(); 
        for(int i=0;i<len;i++){ 
            if(i%length==0 && i>0){ 
                temp.append(split); 
            } 
            temp.append(str.charAt(i)); 
        } 
        String[] attrs=temp.toString().split(split); 
        StringBuilder finalMachineCode=new StringBuilder(); 
        for(String attr : attrs){ 
            if(attr.length()==length){ 
                finalMachineCode.append(attr).append(split); 
            } 
        } 
        String result=finalMachineCode.toString().substring(0, finalMachineCode.toString().length()-1); 
        return result; 
    }     
    /**
     * 利用sigar来生成机器码，当然这个实现不是很好，无法获得CPU ID，希望有兴趣的朋友来改进这个实现
     * @param osName 操作系统类型
     * @return 机器码
     */ 
    protected String getSigarSequence(String osName) { 
        try { 
            File libFile = new File(FileUtils.getAbsolutePath("/WEB-INF/lib/"+PropertyHolder.getProperty("libsigar."+osName))); 
            LOG.debug("libsigar."+osName+" : "+libFile.getAbsolutePath()); 
             
            System.load(libFile.getAbsolutePath()); 
            Set<String> result = new HashSet<>(); 
            Sigar sigar = new Sigar(); 
            String[] ifaces = sigar.getNetInterfaceList(); 
            for (String iface : ifaces) { 
                NetInterfaceConfig cfg = sigar.getNetInterfaceConfig(iface);                if (NetFlags.LOOPBACK_ADDRESS.equals(cfg.getAddress()) || (cfg.getFlags() & NetFlags.IFF_LOOPBACK) != 0 
                        || NetFlags.NULL_HWADDR.equals(cfg.getHwaddr())) { 
                    continue; 
                } 
                String mac = cfg.getHwaddr(); 
                result.add(mac); 
                LOG.debug("mac: " + mac); 
            } 
            if(result.size()<1){ 
                return null; 
            } 
            Properties props = System.getProperties(); 
            String javaVersion = props.getProperty("java.version"); 
            result.add(javaVersion); 
            LOG.debug("Java的运行环境版本：    " + javaVersion); 
            String javaVMVersion = props.getProperty("java.vm.version"); 
            result.add(javaVMVersion); 
            LOG.debug("Java的虚拟机实现版本：    " + props.getProperty("java.vm.version")); 
            String osVersion = props.getProperty("os.version"); 
            result.add(osVersion); 
            LOG.debug("操作系统的版本：    " + props.getProperty("os.version")); 
 
            Mem mem = sigar.getMem(); 
            // 内存总量 
            String totalMem = mem.getTotal() / 1024L + "K av"; 
            LOG.debug("内存总量:    " + totalMem); 
            result.add(totalMem); 
 
            LOG.debug("result:    " + result); 
            String machineCode = getMD5(result.toString()); 
 
            return machineCode; 
        } catch (Throwable ex) { 
            LOG.error("生成 "+osName+" 平台下的机器码失败", ex); 
        } 
        return null; 
    } 
} 


最后，我们分别看看不同平台的实现。

Windows:

Java代码  收藏代码
/**
*在Windows平台上生成机器码
* @author 杨尚川
*/ 
public final class WindowsSequenceService extends AbstractSequenceService{     
    @Override 
    public String getSequence() {         
        String cpuID=getCPUSerial(); 
        String hdID=getHDSerial("C"); 
        if(cpuID==null || hdID==null){ 
            return null; 
        } 
        String machineCode = getMD5(cpuID+hdID); 
                 
        return machineCode; 
    } 
     
    /**
     *
     * @param drive 硬盘驱动器分区 如C,D
     * @return 该分区的卷标
     */ 
    private String getHDSerial(String drive) { 
        StringBuilder result = new StringBuilder(); 
        try { 
            File file = File.createTempFile("tmp", ".vbs"); 
            file.deleteOnExit(); 
            try (FileWriter fw = new java.io.FileWriter(file)) { 
                String vbs = "Set objFSO = CreateObject(\"Scripting.FileSystemObject\")\n" 
                        + "Set colDrives = objFSO.Drives\n" + "Set objDrive = colDrives.item(\"" + drive + "\")\n" 
                        + "Wscript.Echo objDrive.SerialNumber"; 
                fw.write(vbs); 
            } 
            Process p = Runtime.getRuntime().exec("cscript //NoLogo " + file.getPath()); 
            try (BufferedReader input = new BufferedReader(new InputStreamReader(p.getInputStream()))) { 
                String line; 
                while ((line = input.readLine()) != null) { 
                    result.append(line); 
                } 
            } 
            file.delete(); 
        } catch (Throwable e) { 
            LOG.error("生成HDSerial失败", e); 
        } 
        if (result.length() < 1) { 
            LOG.info("无磁盘ID被读取"); 
        } 
 
        return result.toString(); 
    } 
 
    /**
     * 获取CPU号,多CPU时,只取第一个
     * @return
     */ 
    private String getCPUSerial() { 
        StringBuilder result = new StringBuilder(); 
        try { 
            File file = File.createTempFile("tmp", ".vbs"); 
            file.deleteOnExit(); 
            try (FileWriter fw = new FileWriter(file)) { 
                String vbs = "On Error Resume Next \r\n\r\n" + "strComputer = \".\"  \r\n" 
                        + "Set objWMIService = GetObject(\"winmgmts:\" _ \r\n" 
                        + "    & \"{impersonationLevel=impersonate}!\\\\\" & strComputer & \"\\root\\cimv2\") \r\n" 
                        + "Set colItems = objWMIService.ExecQuery(\"Select * from Win32_Processor\")  \r\n " 
                        + "For Each objItem in colItems\r\n " + "    Wscript.Echo objItem.ProcessorId  \r\n " 
                        + "    exit for  ' do the first cpu only! \r\n" + "Next                    "; 
 
                fw.write(vbs); 
            } 
            Process p = Runtime.getRuntime().exec("cscript //NoLogo " + file.getPath()); 
            try (BufferedReader input = new BufferedReader(new InputStreamReader(p.getInputStream()))) { 
                String line; 
                while ((line = input.readLine()) != null) { 
                    result.append(line); 
                } 
            } 
            file.delete(); 
        } catch (Throwable e) { 
            LOG.error("生成CPUSerial失败", e); 
        } 
        if (result.length() < 1) { 
            LOG.info("无CPU_ID被读取"); 
        } 
        return result.toString(); 
    } 
     
    public static void main(String[] args) {         
        SequenceService s = new WindowsSequenceService(); 
        String seq = s.getSequence(); 
        System.out.println(seq); 
    } 
} 


Linux:

Java代码  收藏代码
/**
*在Linux平台上生成机器码
* @author 杨尚川
*/ 
public class LinuxSequenceService  extends AbstractSequenceService{ 
    @Override 
    public String getSequence() { 
        return getSigarSequence("linux"); 
    } 
 
    public static void main(String[] args) { 
        SequenceService s = new LinuxSequenceService(); 
        String seq = s.getSequence(); 
        System.out.println(seq); 
    } 
} 


Mac OS X：

Java代码  收藏代码
/**
*在Mac OS X平台上生成机器码
* @author 杨尚川
*/ 
public class MacSequenceService    extends AbstractSequenceService{ 
    @Override 
    public String getSequence() { 
        return getSigarSequence("mac"); 
    } 
    
    public static void main(String[] args) { 
        SequenceService s = new MacSequenceService(); 
        String seq = s.getSequence(); 
        System.out.println(seq); 
    } 
} 


Solaris：

Java代码  收藏代码
/**
*在Solaris平台上生成机器码
* @author 杨尚川
*/ 
public class SolarisSequenceService    extends AbstractSequenceService{ 
    @Override 
    public String getSequence() { 
        return getSigarSequence("solaris"); 
    } 
 
    public static void main(String[] args) { 
        SequenceService s = new SolarisSequenceService(); 
        String seq = s.getSequence(); 
        System.out.println(seq); 
    }     
} 

评论

1 楼 masuweng 2018-03-15