`
yanguz123
  • 浏览: 567680 次
  • 性别: Icon_minigender_1
  • 来自: 杭州
社区版块
存档分类
最新评论

类扫描

    博客分类:
  • Code
 
阅读更多

 

 

类路径扫描

package com.yuan.common.annotation;

import java.io.File;
import java.io.FileFilter;
import java.io.IOException;
import java.net.JarURLConnection;
import java.net.URL;
import java.net.URLDecoder;
import java.util.Enumeration;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;
import java.util.jar.JarEntry;
import java.util.jar.JarFile;
import java.util.regex.Pattern;

/**
 * 类路径扫描
 * 
 */
public class ClassPathScanner {

	public static void main(String[] args) {
		try {
			Set<Class<?>> classes = new ClassPathScanner().getPackageAllClasses("com.yuan", true);
			for (Class<?> clazz : classes) {
				System.out.println(clazz.getName());
			}
		} catch (ClassNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	// 是否包括内部类
	private boolean excludeInner = true;
	private boolean checkInOrEx = true;
	// 根据类名过滤
	private List<String> classFilters = null;

	public ClassPathScanner() {
	}

	public ClassPathScanner(Boolean excludeInner, Boolean checkInOrEx, List<String> classFilters) {
		this.excludeInner = excludeInner;
		this.checkInOrEx = checkInOrEx;
		this.classFilters = classFilters;
	}

	public boolean isExcludeInner() {
		return excludeInner;
	}

	public void setExcludeInner(boolean excludeInner) {
		this.excludeInner = excludeInner;
	}

	public boolean isCheckInOrEx() {
		return checkInOrEx;
	}

	public void setCheckInOrEx(boolean checkInOrEx) {
		this.checkInOrEx = checkInOrEx;
	}

	public List<String> getClassFilters() {
		return classFilters;
	}

	public void setClassFilters(List<String> classFilters) {
		this.classFilters = classFilters;
	}

	/**
	 * 获取包中的所有类
	 * 
	 * @param basePackage
	 * @param recursive
	 * @return
	 * @throws IOException
	 * @throws ClassNotFoundException
	 */
	public Set<Class<?>> getPackageAllClasses(String basePackage, boolean recursive) throws IOException, ClassNotFoundException {
		// 使用LinkedHashSet来存放扫描到的类
		Set<Class<?>> classes = new LinkedHashSet<Class<?>>();
		String packageName = basePackage;
		// 如果最后一个字符是“.”,则去掉
		if (packageName.endsWith(".")) {
			packageName = packageName.substring(0, packageName.lastIndexOf('.'));
		}
		// 将包名中的“.”换成系统文件夹的“/”
		String package2Path = packageName.replace('.', '/');
		// 使用当前线程来加载文件夹
		Enumeration<URL> dirs = Thread.currentThread().getContextClassLoader().getResources(package2Path);
		while (dirs.hasMoreElements()) {
			URL url = dirs.nextElement();
			// 获取URL协议
			String protocol = url.getProtocol();
			if ("file".equals(protocol)) {
				String filePath = URLDecoder.decode(url.getFile(), "UTF-8");
				// 扫描文件夹中的包和类
				doScanPackageClassesByFile(classes, packageName, filePath, recursive);
			} else if ("jar".equals(protocol)) {
				// 扫描jar包中的包和类
				doScanPackageClassesByJar(packageName, url, recursive, classes);
			}
		}
		return classes;

	}

	/**
	 * 在jar包中扫描包和类
	 * 
	 * @param basePackage
	 *            包名
	 * @param url
	 *            类路径
	 * @param recursive
	 *            是否递归
	 * @param classes
	 *            传引用,返回的结果
	 * @throws IOException
	 * @throws ClassNotFoundException
	 */
	private void doScanPackageClassesByJar(String basePackage, URL url, final boolean recursive, Set<Class<?>> classes) throws IOException, ClassNotFoundException {
		// 包名
		String packageName = basePackage;
		// 获取文件路径
		String package2Path = packageName.replace('.', '/');
		// 转为jar包
		JarFile jar = ((JarURLConnection) url.openConnection()).getJarFile();
		// 遍历jar包中的元素
		Enumeration<JarEntry> entries = jar.entries();
		while (entries.hasMoreElements()) {
			JarEntry entry = entries.nextElement();
			String name = entry.getName();
			// 如果路径不一致,或者是目录,则继续
			if (!name.startsWith(package2Path) || entry.isDirectory()) {
				continue;
			}
			// 判断是否递归搜索子包
			if (!recursive && name.lastIndexOf('/') != package2Path.length()) {
				continue;
			}
			// 判断是否过滤 inner class
			if (this.excludeInner && name.indexOf('$') != -1) {
				continue;
			}
			String classSimpleName = name.substring(name.lastIndexOf('/') + 1);
			// 判定是否符合过滤条件
			if (this.filterClassName(classSimpleName)) {
				String className = name.replace('/', '.');
				className = className.substring(0, className.length() - 6);
				// 用当前线程的类加载器加载类
				classes.add(Thread.currentThread().getContextClassLoader().loadClass(className));
			}
		}
	}

	/**
	 * 在文件夹中扫描包和类
	 * 
	 * @param classes
	 * @param packageName
	 * @param packagePath
	 * @param recursive
	 * @throws ClassNotFoundException
	 */
	private void doScanPackageClassesByFile(Set<Class<?>> classes, String packageName, String packagePath, boolean recursive) throws ClassNotFoundException {
		// 转为文件
		File dir = new File(packagePath);
		if (!dir.exists() || !dir.isDirectory()) {
			return;
		}
		final boolean fileRecursive = recursive;
		// 列出文件,进行过滤
		File[] dirfiles = dir.listFiles(new FileFilter() {
			// 自定义文件过滤规则
			public boolean accept(File file) {
				if (file.isDirectory()) {
					return fileRecursive;
				}
				String filename = file.getName();
				if (excludeInner && filename.indexOf('$') != -1) {
					return false;
				}
				return filterClassName(filename);
			}
		});
		for (File file : dirfiles) {
			if (file.isDirectory()) {
				// 如果是目录,则递归
				doScanPackageClassesByFile(classes, packageName + "." + file.getName(), file.getAbsolutePath(), recursive);
			} else {
				// 用当前类加载器加载
				String className = file.getName().substring(0, file.getName().length() - 6);
				classes.add(Thread.currentThread().getContextClassLoader().loadClass(packageName + '.' + className));
			}
		}
	}

	/**
	 * 过滤类文件
	 * 
	 * @param className
	 * @return
	 */
	private boolean filterClassName(String className) {
		// 文件后缀为class
		if (!className.endsWith(".class")) {
			return false;
		}
		// 没有类过滤规则
		if (null == this.classFilters || this.classFilters.isEmpty()) {
			return true;
		}
		String tmpName = className.substring(0, className.length() - 6);
		boolean flag = false;
		for (String str : classFilters) {
			String tmpreg = "^" + str.replace("*", ".*") + "$";
			Pattern p = Pattern.compile(tmpreg);
			if (p.matcher(tmpName).find()) {
				flag = true;
				break;
			}
		}
		return (checkInOrEx && flag) || (!checkInOrEx && !flag);
	}

}

 

 

分享到:
评论

相关推荐

    扫描接口实现类

    它们不仅包含了上述功能,还考虑了多线程安全、异步加载、排除指定类等问题,使得类扫描更加高效和可靠。 总之,"扫描接口实现类"是Java开发中的一个重要技巧,它涉及到类加载、反射、类路径遍历等多个核心概念。...

    spring启动componentscan类扫描加载过程

    这个过程涉及到Spring IoC(Inversion of Control)容器的初始化,包括类路径扫描、Bean定义的创建以及实例化。接下来,我们将深入探讨`@ComponentScan`的工作原理及其背后的源码分析。 首先,`@ComponentScan`注解...

    classgraph,一个Uber快速、超轻量级Java类路径扫描器、模块扫描仪和注释处理器。.zip

    这个工具由Luke Daley开发,旨在提供灵活、高效且功能丰富的类扫描解决方案。 一、类路径扫描器 ClassGraph能够深入分析应用程序的类路径,包括JAR文件、目录、模块以及类加载器中的所有类。它支持扫描特定的包、类...

    扫描全能王

    扫描全能王提供高品质、最专业、最便捷的扫描、传真服务,将您的智能手机变为随身携带的扫描仪、传真机

    ip扫描器深度

    然而,描述中的"最适合抓鸡"可能是对这类扫描器功能的一种夸张表述,实际上,专业的IP扫描器更多的是用于合法的安全测试和网络管理,例如识别开放端口、检测服务类型以及评估网络的安全状况。 【标签】"扫描器",这...

    ClassGraph-超快速超轻量级并行化的Java类路径扫描程序

    它不仅适用于常规的类扫描任务,还支持模块扫描和注解处理,使得在Java开发中进行元数据探索变得简单易行。 1. **类路径扫描**: ClassGraph的核心功能是扫描Java类路径,包括JAR文件、目录和模块。它能够遍历所有...

    同网段 IP扫描器

    这类扫描器通常能够快速发现网络中未授权的设备接入、开放的端口,以及可能存在的安全隐患。 IP 扫描器的工作原理基于 TCP/IP 协议栈,通过发送特定的网络请求(如 ARP 请求或 ICMP 数据包)来探测网络中活动的 IP ...

    端口扫描器集合

    S扫描器可能是一个特定的扫描工具,具体功能没有明确描述,但通常这类扫描器会包含更多高级功能,如隐秘扫描(防止被目标主机察觉),多线程扫描以提高速度,以及对特定服务的深度检测,比如HTTP、FTP等。...

    全新端口扫描器集合

    在使用这类扫描器时,需要注意的是,未经许可的扫描可能会违反法律,尤其是在目标未授权的情况下。因此,确保合规性和尊重他人网络隐私至关重要。 在使用"全新端口扫描器集合"之前,首先需要解压缩文件。提供的文件...

    1433端口通用扫描器

    通常,这类扫描器会提供设置扫描范围、选择扫描方式(如TCP连接、SYN扫描等)以及查看扫描结果等功能。 标签“S扫描器”可能意味着这是一个强调速度(Speed)或者全面性(Scope)的扫描工具,或者是开发者的简称。...

    超强WEB目录扫描器

    这类扫描器能够帮助安全专家、网站管理员以及研究人员定位潜在的安全漏洞,确保网站的安全性。"超强WEB目录扫描器"就是这样一个工具,它具有强大的扫描能力和深度探测功能。 在网络安全中,理解Web目录扫描的重要性...

    端口扫描工具源代码

    源代码可能包括这两类扫描的实现。 7. **多线程/异步处理**:为了提高扫描效率,端口扫描工具可能会使用多线程或异步I/O,同时扫描多个端口,这需要对C++的并发编程有深入理解。 8. **错误处理与异常安全**:一个...

    扫描内网ip地址 局域网mac地址扫描

    在执行这类扫描时,有几点需要注意: 1. 权限:进行网络扫描可能需要管理员权限,因为这涉及到读取网络配置和监听网络流量。 2. 隐私与安全:扫描他人IP和MAC地址可能侵犯他人的隐私,因此在操作前确保你有合法的...

    网吧专用扫描器

    在网吧运营中,为了确保网络安全、优化资源分配以及提高客户服务体验,这类扫描器扮演着至关重要的角色。以下是关于【网吧专用扫描器】的相关知识点: 1. **网络安全**:网吧专用扫描器可以定期扫描网吧内所有电脑...

    1433扫描器工具(绝版)

    1433端口是Microsoft SQL Server默认使用的端口,因此这类扫描器通常被用来检测SQL Server的存在和配置情况。 描述中提到的“1433全自动”意味着这个工具能够自动进行全范围的扫描,无需用户手动干预。它被描述为...

    流光扫描器

    流光扫描器能够执行TCP全连接扫描、半开扫描、UDP扫描等多种扫描方式,全面覆盖了端口扫描的各种需求。此外,它还支持SYN扫描、ACK扫描等不同扫描技术,帮助用户了解网络环境中的实时状态。 漏洞扫描是流光扫描器的...

    vc编写tcp syn扫描器源代码

    TCP SYN扫描,也称为半开放扫描,是这类扫描的一种常见方法。这种扫描方式利用了TCP三次握手协议的一部分,即SYN阶段,来检测目标主机是否在线以及特定端口是否开放。 标题"vc编写tcp syn扫描器源代码"表明这是一个...

    北极熊扫描器3.5【国产安全工具】

    本软件禁止用于黑客用途,不允许对除个人网站外进行扫描检测,不允许利用漏洞进行入侵操作,若您使用本程序进行违法操作,造成的任何后果与作者无关,作者不承担任何法律责任、纠纷,软件作为一款安全检测软件,...

    class-scanner-stream:类扫描器作为流

    类扫描器流 这是一个轻量级的类扫描器,它使我们可以在流管道中应用过滤器并收集结果。 扫描指定包下的所有类 List&lt; Class&gt; scanResults = ClassScanner . scan(classLoader, " com.beerboy.scanner " ) .collect...

Global site tag (gtag.js) - Google Analytics