<?php
// Создание stored-RAR архивов
// Версия 0.3
// Автор: Алексей Рембиш a.k.a Ramon
// E-mail: alex@rembish.ru
// Copyright 2009

// Класс создания RAR-архивов со stored-сжатием
// Пример работы:
// $rar = new store_rar;
//  $rar->create("archive.rar"); # создаём архив
//  $rar->addFile("a.txt");      # пишем в него файл a.txt
//  $rar->addDirectory("b/c");   # создаём в архиве директорию "b" с поддиректорией "c"
//  $rar->addFile("d/e.txt");    # создаём директорию "d" и пишем в неё e.txt
// $rar->close();                # закрываем архив
class store_rar {
    // Указатель на архив
    private $id = null;
    // Внутренняя структура каталогов, чтобы не создавать лишние.
    private $tree = array();

    // Функция, создающая новый архив, после чего записывающая в него обязательные заголовки.
    public function create($filename) {
        $this->id = fopen($filename, "wb");
        if (!$this->id)
            return false;

        $this->tree = array();

        $this->writeHeader(0x72, 0x1a21);
        $this->writeHeader(0x73, 0x0000, array(array(0, 2), array(0, 4)));
        return true;
    }
    // Функция, закрывающая записанный архив
    public function close() {
        fclose($this->id);
    }

    // Функция, что добавляет в архив директорию. Нормально справляется с рекурсивными
    // директориями. Например, для $name = "a/b" создаст директорию "a", а в ней поддиректорию
    // "b". Не будет создавать дубликаты директорий.
    public function addDirectory($name) {
        $name = str_replace("/", "\\", $name);
        if ($name[0] == "\\") $name = substr($name, 1);
        if ($name[strlen($name) - 1] == "\\") $name = substr($name, 0, -1);

        $parts = explode("\\", $name);
        $c = &$this->tree;
        $cname = ""; $delim = "";
        for ($i = 0; $i < count($parts); $i++) {
            $cname .= $delim.$parts[$i];
            if (!isset($c[$parts[$i]])) {
                $c[$parts[$i]] = array();
                $this->writeHeader(0x74, $this->setBits(array(5, 6, 7, 15)), array(
                    array(0, 4),                    // Packed size = 0 for directories
                    array(0, 4),                    // Unpacked size = 0 for directories
                    array(0, 1),                    // Host OS = "MS DOS"
                    array(0, 4),                    // File CRC = 0 for directories
                    array($this->getDateTime(), 4),    // File time = Current date and time in MS DOS format
                    array(20, 1),                    // RAR version = 2.0
                    array(0x30, 1),                    // Method = Store
                    array(strlen($cname), 2),        // Name size
                    array(0x10, 4),                    // File attributes = Directory
                    $cname,                            // Filename = Directory name
                ));
            }
            $c = &$c[$parts[$i]];
            $delim = "\\";
        }

        return $name;
    }

    // Функция записывающая файл $name в архив. Если задан параметр $dir, то файл будет записан
    // в соответствующую директорию внутри архива (директория может не существовать до записи
    // файла). $name - путь к файлу на сервере. Файл будет записан с basename($name).
    public function addFile($name, $dir = null) {
        if (!file_exists($name))
            return false;

        $c = &$this->tree;
        if (!is_null($dir)) {
            $dir = $this->addDirectory($dir);

            $parts = explode("\\", $dir);
            for($i = 0; $i < count($parts); $i++)
                $c = &$c[$parts[$i]];
        }

        $fname = pathinfo($name, PATHINFO_BASENAME);
        if (in_array($fname, $c))
            return true;

        $data = file_get_contents($name);
        $size = strlen($data);
        if (!is_null($dir))
            $fname = $dir."\\".$name;

        $this->writeHeader(0x74, $this->setBits(array(15)), array(
            array($size, 4),                                // Packed size = File size
            array($size, 4),                                // Unpacked size = File size
            array(0, 1),                                    // Host OS = "MS DOS"
            array(crc32($data), 4),                            // File CRC
            array($this->getDateTime(filemtime($name)), 4),    // File time
            array(20, 1),                                    // RAR version = 2.0
            array(0x30, 1),                                    // Method = store
            array(strlen($fname), 2),                        // Name size
            array(0x20, 4),                                    // File attributes = Archived
            $fname,                                            // Filename
        ));

        fwrite($this->id, $data);
        $c[] = $fname;

        return true;
    }

    // Внутренняя функция, пишущая заголовок блока в соответствии с форматом RAR.
    // Работает только с тремя типами заголовков: блок-маркер, заголовок архива и
    // заголовок файла - $headType. $headFlags - флаги заголовка, $data - возможно
    // пустой массив дополнительных параметров заголовка, что идут после первых
    // обязательных 7 байт.
    private function writeHeader($headType, $headFlags, $data = array()) {
        if (!in_array($headType, array(0x72, 0x73, 0x74)))
            return false;

        $headSize = 2 + 1 + 2 + 2;
        foreach ($data as $key => $value)
            $headSize += is_array($value) ? $value[1] : strlen($value);

        $header = $this->writeBytesToString(array_merge(array($headType, array($headFlags, 2), array($headSize, 2)), $data));
        $header = ($headType == 0x72 ? "Ra" : $this->getCRC($header)).$header;

        fwrite($this->id, $header);
    }

    // Расчёт CRC для заголовка блока. CRC урезается до 2 байт из 4х.
    private function getCRC($string) {
        $crc = crc32($string);
        return chr($crc & 0xFF).chr(($crc >> 8) & 0xFF);
    }

    // Внутренняя функция записи данных в обратном порядке байтов.
    private function getBytes($data, $bytes = 0) {
        $output = "";
        if (!$bytes)
            $bytes = strlen($bytes);

        if (is_int($data) || is_float($data)) {
            $data = sprintf("%0".($bytes * 2)."x", $data);

            for ($i = 0; $i < strlen($data); $i += 2)
                $output = chr(hexdec(substr($data, $i, 2))).$output;
        } else
            $output = $data;

        return $output;
    }
    // Запись массива данных в байт-строку с учётом размерности переданных данных.
    private function writeBytesToString($data) {
        $output = "";
        for ($i = 0; $i < count($data); $i++) {
            if (is_array($data[$i]))
                $output .= $this->getBytes($data[$i][0], $data[$i][1]);
            else
                $output .= $this->getBytes($data[$i]);
        }

        return $output;
    }

    // Установка соответствующих битов в числе. $bits - массив номеров битов или отдельный номер бита.
    private function setBits($bits) {
        $out = 0;
        if (is_int($bits))
            $bits[] = $bits;

        for ($i = 0; $i < count($bits); $i++)
            $out |= 1 << $bits[$i];

        return $out;
    }

    // Получение даты в 4хбитном формате MSDOS. $time - timestamp, от которого получить дату.
    private function getDateTime($time = null) {
        if (!is_null($time))
            $time = time();

        $dt = getdate();
        $out = $dt["seconds"] | ($dt["minutes"] << 5) | ($dt["hours"] << 11) | ($dt["mday"] << 16) | ($dt["mon"] << 21) | (($dt["year"] - 1980) << 25);
        return $out;
    }
};
?>