Как я могу создать .torrent в Java? - PullRequest
Новая
       11

Как я могу создать .torrent в Java?

7 голосов
/ 09 января 2010

Я хочу сгенерировать файл .torrent в Java, но мне не нужен большой API, который делает что-то вроде очистки трекеров, заполнения и т. Д. Это только для клиента, который генерирует метаданные. Какие легкие решения существуют? Я только генерирую .torrent одного файла .zip.

Спасибо!

Ответы [ 2 ]

22 голосов
/ 09 января 2010

Я собрал этот автономный фрагмент кода Java, чтобы подготовить файл .torrent с одним файлом.

Файл .torrent создается путем вызова createTorrent() с передачей имени файла .torrent, имени общего файла и URL-адреса трекера.

createTorrent() использует hashPieces() для хеширования файла штук с использованием Java MessageDigest класса. Затем createTorrent() подготавливает словарь метаинформации , содержащий метаданные торрента. Затем этот словарь сериализуется в правильном формате bencode с использованием методов encode*() и сохраняется в файле .torrent.

Подробнее см. BitTorrent spec . 

public class Torrent {
    private static void encodeObject(Object o, OutputStream out) throws IOException {
        if (o instanceof String)
            encodeString((String)o, out);
        else if (o instanceof Map)
            encodeMap((Map)o, out);
        else if (o instanceof byte[])
            encodeBytes((byte[])o, out);
        else if (o instanceof Number)
            encodeLong(((Number) o).longValue(), out);
        else
            throw new Error("Unencodable type");
    }
    private static void encodeLong(long value, OutputStream out) throws IOException {
        out.write('i');
        out.write(Long.toString(value).getBytes("US-ASCII"));
        out.write('e');
    }
    private static void encodeBytes(byte[] bytes, OutputStream out) throws IOException {
        out.write(Integer.toString(bytes.length).getBytes("US-ASCII"));
        out.write(':');
        out.write(bytes);
    }
    private static void encodeString(String str, OutputStream out) throws IOException {
        encodeBytes(str.getBytes("UTF-8"), out);
    }
    private static void encodeMap(Map<String,Object> map, OutputStream out) throws IOException{
        // Sort the map. A generic encoder should sort by key bytes
        SortedMap<String,Object> sortedMap = new TreeMap<String, Object>(map);
        out.write('d');
        for (Entry<String, Object> e : sortedMap.entrySet()) {
            encodeString(e.getKey(), out);
            encodeObject(e.getValue(), out);
        }
        out.write('e');
    }
    private static byte[] hashPieces(File file, int pieceLength) throws IOException {
        MessageDigest sha1;
        try {
            sha1 = MessageDigest.getInstance("SHA");
        } catch (NoSuchAlgorithmException e) {
            throw new Error("SHA1 not supported");
        }
        InputStream in = new FileInputStream(file);
        ByteArrayOutputStream pieces = new ByteArrayOutputStream();
        byte[] bytes = new byte[pieceLength];
        int pieceByteCount  = 0, readCount = in.read(bytes, 0, pieceLength);
        while (readCount != -1) {
            pieceByteCount += readCount;
            sha1.update(bytes, 0, readCount);
            if (pieceByteCount == pieceLength) {
                pieceByteCount = 0;
                pieces.write(sha1.digest());
            }
            readCount = in.read(bytes, 0, pieceLength-pieceByteCount);
        }
        in.close();
        if (pieceByteCount > 0)
            pieces.write(sha1.digest());
        return pieces.toByteArray();
    }
    public static void createTorrent(File file, File sharedFile, String announceURL) throws IOException {
        final int pieceLength = 512*1024;
        Map<String,Object> info = new HashMap<String,Object>();
        info.put("name", sharedFile.getName());
        info.put("length", sharedFile.length());
        info.put("piece length", pieceLength);
        info.put("pieces", hashPieces(sharedFile, pieceLength));
        Map<String,Object> metainfo = new HashMap<String,Object>();
        metainfo.put("announce", announceURL);
        metainfo.put("info", info);
        OutputStream out = new FileOutputStream(file);
        encodeMap(metainfo, out);
        out.close();
    }

    public static void main(String[] args) throws Exception {
        createTorrent(new File("C:/x.torrent"), new File("C:/file"), "http://example.com/announce");
    }
}

Редактирование кода: Сделайте это немного компактнее, исправьте видимость методов, используйте символьные литералы, где это уместно, используйте instanceof Number. И совсем недавно читал файл, используя блок ввода / вывода , потому что я пытаюсь использовать его для реального, а байтовый ввод / вывод просто медленный,

10 голосов
/ 09 января 2010

Я бы начал с Java Bittorrent API . Jar составляет около 70 КБ, но вы, вероятно, можете сократить его, удалив классы, не необходимые для создания торрентов. В SDK есть образец ExampleCreateTorrent.java , иллюстрирующий, как сделать именно то, что вам нужно.

Вы также можете посмотреть, как это реализовано в клиентах Java с открытым исходным кодом, таких как Azureus.

...