测试controller
cyt
parent
67bc687d0a
commit
af9e040f01
|
|
@ -0,0 +1,112 @@
|
||||||
|
package com.unisinsight.project.controller;
|
||||||
|
|
||||||
|
import lombok.extern.slf4j.Slf4j;
|
||||||
|
import org.springframework.web.bind.annotation.*;
|
||||||
|
|
||||||
|
import java.io.*;
|
||||||
|
|
||||||
|
@RestController("test")
|
||||||
|
@Slf4j
|
||||||
|
public class TestController {
|
||||||
|
|
||||||
|
|
||||||
|
private static final String BT_SCRIPT_PATH = "/var/lib/vdi/nodejs/bttorrent.sh";
|
||||||
|
private static final String RUN_USER = "java_usr"; // 替换为你的实际用户
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 制作种子并开始做种
|
||||||
|
*
|
||||||
|
* @param sourceFile 源文件路径（如 /data/file.iso）
|
||||||
|
* @param torrentFile 生成的种子文件路径（如 /data/file.torrent）
|
||||||
|
* @return 是否成功
|
||||||
|
*/
|
||||||
|
public static boolean createAndSeed(String sourceFile, String torrentFile) {
|
||||||
|
// 验证文件是否存在
|
||||||
|
if (!new File(sourceFile).exists()) {
|
||||||
|
System.err.println("源文件不存在: " + sourceFile);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
// 确保目标目录存在
|
||||||
|
new File(torrentFile).getParentFile().mkdirs();
|
||||||
|
return executeCommand("start", sourceFile, torrentFile);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 停止指定文件的做种
|
||||||
|
*
|
||||||
|
* @param sourceFile 源文件路径（如 /data/file.iso）
|
||||||
|
* @return 是否成功
|
||||||
|
*/
|
||||||
|
public static boolean stopSeeding(String sourceFile) {
|
||||||
|
return executeCommand("stop_path", sourceFile, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 执行 bttorrent.sh 脚本
|
||||||
|
*/
|
||||||
|
private static boolean executeCommand(String command, String arg1, String arg2) {
|
||||||
|
try {
|
||||||
|
// 构造命令
|
||||||
|
ProcessBuilder pb = new ProcessBuilder(
|
||||||
|
"sudo", "-u", RUN_USER,
|
||||||
|
BT_SCRIPT_PATH,
|
||||||
|
command,
|
||||||
|
arg1,
|
||||||
|
arg2 != null ? arg2 : "" // 处理可选参数
|
||||||
|
).redirectErrorStream(true);
|
||||||
|
// 启动进程
|
||||||
|
Process process = pb.start();
|
||||||
|
// 打印输出（调试用）
|
||||||
|
logProcessOutput(process);
|
||||||
|
// 等待执行完成
|
||||||
|
return process.waitFor() == 0;
|
||||||
|
} catch (IOException | InterruptedException e) {
|
||||||
|
e.printStackTrace();
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* 打印进程输出
|
||||||
|
*/
|
||||||
|
private static void logProcessOutput(Process process) throws IOException {
|
||||||
|
try (BufferedReader reader = new BufferedReader(
|
||||||
|
new InputStreamReader(process.getInputStream()))) {
|
||||||
|
String line;
|
||||||
|
while ((line = reader.readLine()) != null) {
|
||||||
|
System.out.println("[BT] " + line);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// 测试用例
|
||||||
|
public static void main(String[] args) {
|
||||||
|
String sourceFile = "/var/lib/vdi/test/example.iso";
|
||||||
|
String torrentFile = "/var/lib/vdi/test/example.torrent";
|
||||||
|
// 测试做种
|
||||||
|
System.out.println("开始做种...");
|
||||||
|
boolean seedResult = createAndSeed(sourceFile, torrentFile);
|
||||||
|
System.out.println("做种结果: " + (seedResult ? "成功" : "失败"));
|
||||||
|
// 测试停止
|
||||||
|
System.out.println("停止做种...");
|
||||||
|
boolean stopResult = stopSeeding(sourceFile);
|
||||||
|
System.out.println("停止结果: " + (stopResult ? "成功" : "失败"));
|
||||||
|
}
|
||||||
|
|
||||||
|
@GetMapping("/start")
|
||||||
|
public String start(@RequestParam("sourceFile")String sourceFile,
|
||||||
|
@RequestParam("torrentFile")String torrentFile) {
|
||||||
|
System.out.println("开始做种...");
|
||||||
|
boolean seedResult = createAndSeed(sourceFile, torrentFile);
|
||||||
|
System.out.println("做种结果: " + (seedResult ? "成功" : "失败"));
|
||||||
|
return "success";
|
||||||
|
}
|
||||||
|
|
||||||
|
@GetMapping("/stop")
|
||||||
|
public String stop(@RequestParam("sourceFile")String sourceFile) {
|
||||||
|
// 测试停止
|
||||||
|
System.out.println("停止做种...");
|
||||||
|
boolean stopResult = stopSeeding(sourceFile);
|
||||||
|
System.out.println("停止结果: " + (stopResult ? "成功" : "失败"));
|
||||||
|
return "success";
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -1,36 +0,0 @@
|
||||||
package com.unisinsight.project.tb;
|
|
||||||
|
|
||||||
import com.turn.ttorrent.client.Client;
|
|
||||||
import com.turn.ttorrent.client.SharedTorrent;
|
|
||||||
|
|
||||||
import java.io.File;
|
|
||||||
import java.net.InetAddress;
|
|
||||||
|
|
||||||
public class JavaSeeder {
|
|
||||||
public static void main(String[] args) throws Exception {
|
|
||||||
// 1. 种子文件和原始文件
|
|
||||||
File torrentFile = new File("4.txt.torrent"); // 输出的种子文件
|
|
||||||
File file = new File("D:\\temp\\log\\3.txt"); // 要共享的文件
|
|
||||||
//
|
|
||||||
String trackerUrl = "udp://tracker.openbittorrent.com:80/announce"; // Tracker URL
|
|
||||||
TorrentCreator111.createTorrent(file, trackerUrl, torrentFile);
|
|
||||||
System.out.println("Torrent file created successfully: " + torrentFile.getAbsolutePath());
|
|
||||||
|
|
||||||
// 1. 加载种子和文件
|
|
||||||
SharedTorrent torrent = SharedTorrent.fromFile(
|
|
||||||
torrentFile,
|
|
||||||
file.getParentFile()
|
|
||||||
);
|
|
||||||
|
|
||||||
// 2. 创建客户端并启用 DHT
|
|
||||||
Client client = new Client(
|
|
||||||
InetAddress.getLocalHost(), // 绑定本机 IP
|
|
||||||
torrent
|
|
||||||
);
|
|
||||||
|
|
||||||
// 3. 开始做种
|
|
||||||
client.share();
|
|
||||||
System.out.println("DHT 做种已启动...");
|
|
||||||
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
@ -1,138 +0,0 @@
|
||||||
package com.unisinsight.project.tb;
|
|
||||||
|
|
||||||
import java.io.*;
|
|
||||||
import java.nio.file.Files;
|
|
||||||
import java.security.MessageDigest;
|
|
||||||
import java.security.NoSuchAlgorithmException;
|
|
||||||
import java.util.ArrayList;
|
|
||||||
import java.util.HashMap;
|
|
||||||
import java.util.List;
|
|
||||||
import java.util.Map;
|
|
||||||
|
|
||||||
public class TorrentCreator111 {
|
|
||||||
private static final int PIECE_LENGTH = 256 * 1024; // 256KB 分块大小
|
|
||||||
|
|
||||||
public static void createTorrent(File file, String announceUrl, File outputTorrent)
|
|
||||||
throws IOException, NoSuchAlgorithmException {
|
|
||||||
// 计算文件哈希
|
|
||||||
List<byte[]> pieceHashes = calculatePieceHashes(file);
|
|
||||||
|
|
||||||
// 构建info字典
|
|
||||||
Map<String, Object> info = new HashMap<>();
|
|
||||||
info.put("name", file.getName());
|
|
||||||
info.put("piece length", PIECE_LENGTH);
|
|
||||||
info.put("pieces", concatenateHashes(pieceHashes));
|
|
||||||
info.put("length", file.length());
|
|
||||||
|
|
||||||
// 构建整个torrent字典
|
|
||||||
Map<String, Object> torrent = new HashMap<>();
|
|
||||||
torrent.put("announce", announceUrl);
|
|
||||||
torrent.put("info", info);
|
|
||||||
|
|
||||||
// 编码并写入文件
|
|
||||||
byte[] torrentData = Bencode.encode(torrent);
|
|
||||||
try (FileOutputStream fos = new FileOutputStream(outputTorrent)) {
|
|
||||||
fos.write(torrentData);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static List<byte[]> calculatePieceHashes(File file)
|
|
||||||
throws IOException, NoSuchAlgorithmException {
|
|
||||||
List<byte[]> pieceHashes = new ArrayList<>();
|
|
||||||
MessageDigest sha1 = MessageDigest.getInstance("SHA-1");
|
|
||||||
|
|
||||||
try (InputStream is = Files.newInputStream(file.toPath())) {
|
|
||||||
byte[] buffer = new byte[PIECE_LENGTH];
|
|
||||||
int bytesRead;
|
|
||||||
|
|
||||||
while ((bytesRead = is.read(buffer)) != -1) {
|
|
||||||
sha1.reset();
|
|
||||||
sha1.update(buffer, 0, bytesRead);
|
|
||||||
pieceHashes.add(sha1.digest());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return pieceHashes;
|
|
||||||
}
|
|
||||||
|
|
||||||
private static byte[] concatenateHashes(List<byte[]> pieceHashes) {
|
|
||||||
byte[] result = new byte[pieceHashes.size() * 20]; // SHA-1哈希是20字节
|
|
||||||
int offset = 0;
|
|
||||||
|
|
||||||
for (byte[] hash : pieceHashes) {
|
|
||||||
System.arraycopy(hash, 0, result, offset, hash.length);
|
|
||||||
offset += hash.length;
|
|
||||||
}
|
|
||||||
|
|
||||||
return result;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Bencode编码工具类
|
|
||||||
private static class Bencode {
|
|
||||||
public static byte[] encode(Object obj) throws IOException {
|
|
||||||
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
||||||
encode(obj, baos);
|
|
||||||
return baos.toByteArray();
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void encode(Object obj, OutputStream out) throws IOException {
|
|
||||||
if (obj instanceof String) {
|
|
||||||
encodeString((String) obj, out);
|
|
||||||
} else if (obj instanceof Integer || obj instanceof Long) {
|
|
||||||
encodeInteger(((Number) obj).longValue(), out);
|
|
||||||
} else if (obj instanceof Map) {
|
|
||||||
encodeDictionary((Map<?, ?>) obj, out);
|
|
||||||
} else if (obj instanceof List) {
|
|
||||||
encodeList((List<?>) obj, out);
|
|
||||||
} else if (obj instanceof byte[]) {
|
|
||||||
encodeBytes((byte[]) obj, out);
|
|
||||||
} else {
|
|
||||||
throw new IllegalArgumentException("Unsupported type: " + obj.getClass());
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void encodeString(String s, OutputStream out) throws IOException {
|
|
||||||
out.write((s.length() + ":").getBytes());
|
|
||||||
out.write(s.getBytes());
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void encodeInteger(long i, OutputStream out) throws IOException {
|
|
||||||
out.write(("i" + i + "e").getBytes());
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void encodeDictionary(Map<?, ?> dict, OutputStream out) throws IOException {
|
|
||||||
out.write('d');
|
|
||||||
for (Map.Entry<?, ?> entry : dict.entrySet()) {
|
|
||||||
encode(entry.getKey().toString(), out);
|
|
||||||
encode(entry.getValue(), out);
|
|
||||||
}
|
|
||||||
out.write('e');
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void encodeList(List<?> list, OutputStream out) throws IOException {
|
|
||||||
out.write('l');
|
|
||||||
for (Object item : list) {
|
|
||||||
encode(item, out);
|
|
||||||
}
|
|
||||||
out.write('e');
|
|
||||||
}
|
|
||||||
|
|
||||||
private static void encodeBytes(byte[] bytes, OutputStream out) throws IOException {
|
|
||||||
out.write((bytes.length + ":").getBytes());
|
|
||||||
out.write(bytes);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
public static void main(String[] args) {
|
|
||||||
try {
|
|
||||||
File fileToShare = new File("D:\\temp\\log\\2.txt"); // 要共享的文件
|
|
||||||
String trackerUrl = "http://tracker.openbittorrent.com:80/announce"; // Tracker URL
|
|
||||||
File torrentFile = new File("2.txt.torrent"); // 输出的种子文件
|
|
||||||
|
|
||||||
createTorrent(fileToShare, trackerUrl, torrentFile);
|
|
||||||
System.out.println("Torrent file created successfully: " + torrentFile.getAbsolutePath());
|
|
||||||
} catch (Exception e) {
|
|
||||||
e.printStackTrace();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue