Implement Truecrypt ISO loader.
This commit is contained in:
Vladimir Serbinenko
parent
e14d409b6f
commit
3086175489
5 changed files with 296 additions and 3 deletions
|
|
@ -1,3 +1,7 @@
|
|||
2013-12-17 Vladimir Serbinenko <phcoder@gmail.com>
|
||||
|
||||
Implement Truecrypt ISO loader.
|
||||
|
||||
2013-12-17 Vladimir Serbinenko <phcoder@gmail.com>
|
||||
|
||||
* grub-core/lib/arg.c: Ensure at least a single space between commands.
|
||||
|
|
|
|||
|
|
@ -1623,6 +1623,14 @@ module = {
|
|||
enable = i386_pc;
|
||||
};
|
||||
|
||||
|
||||
module = {
|
||||
name = truecrypt;
|
||||
i386_pc = loader/i386/pc/truecrypt.c;
|
||||
enable = i386_pc;
|
||||
};
|
||||
|
||||
|
||||
module = {
|
||||
name = freedos;
|
||||
i386_pc = loader/i386/pc/freedos.c;
|
||||
|
|
|
|||
|
|
@ -1093,7 +1093,7 @@ inflate_window (grub_gzio_t gzio)
|
|||
}
|
||||
}
|
||||
|
||||
gzio->saved_offset += WSIZE;
|
||||
gzio->saved_offset += gzio->wp;
|
||||
|
||||
/* XXX do CRC calculation here! */
|
||||
}
|
||||
|
|
@ -1219,7 +1219,14 @@ grub_gzio_read_real (grub_gzio_t gzio, grub_off_t offset,
|
|||
register char *srcaddr;
|
||||
|
||||
while (offset >= gzio->saved_offset)
|
||||
inflate_window (gzio);
|
||||
{
|
||||
inflate_window (gzio);
|
||||
if (gzio->wp == 0)
|
||||
goto out;
|
||||
}
|
||||
|
||||
if (gzio->wp == 0)
|
||||
goto out;
|
||||
|
||||
srcaddr = (char *) ((offset & (WSIZE - 1)) + gzio->slide);
|
||||
size = gzio->saved_offset - offset;
|
||||
|
|
@ -1234,6 +1241,7 @@ grub_gzio_read_real (grub_gzio_t gzio, grub_off_t offset,
|
|||
offset += size;
|
||||
}
|
||||
|
||||
out:
|
||||
if (grub_errno != GRUB_ERR_NONE)
|
||||
ret = -1;
|
||||
|
||||
|
|
@ -1243,7 +1251,15 @@ grub_gzio_read_real (grub_gzio_t gzio, grub_off_t offset,
|
|||
static grub_ssize_t
|
||||
grub_gzio_read (grub_file_t file, char *buf, grub_size_t len)
|
||||
{
|
||||
return grub_gzio_read_real (file->data, file->offset, buf, len);
|
||||
grub_ssize_t ret;
|
||||
ret = grub_gzio_read_real (file->data, file->offset, buf, len);
|
||||
|
||||
if (!grub_errno && ret != (grub_ssize_t) len)
|
||||
{
|
||||
grub_error (GRUB_ERR_BAD_COMPRESSED_DATA, "premature end of compressed");
|
||||
ret = -1;
|
||||
}
|
||||
return ret;
|
||||
}
|
||||
|
||||
/* Release everything, including the underlying file object. */
|
||||
|
|
@ -1286,10 +1302,38 @@ grub_zlib_decompress (char *inbuf, grub_size_t insize, grub_off_t off,
|
|||
ret = grub_gzio_read_real (gzio, off, outbuf, outsize);
|
||||
grub_free (gzio);
|
||||
|
||||
if (!grub_errno && ret != (grub_ssize_t) outsize)
|
||||
{
|
||||
grub_error (GRUB_ERR_BAD_COMPRESSED_DATA, "premature end of compressed");
|
||||
ret = -1;
|
||||
}
|
||||
|
||||
/* FIXME: Check Adler. */
|
||||
return ret;
|
||||
}
|
||||
|
||||
grub_ssize_t
|
||||
grub_deflate_decompress (char *inbuf, grub_size_t insize, grub_off_t off,
|
||||
char *outbuf, grub_size_t outsize)
|
||||
{
|
||||
grub_gzio_t gzio = 0;
|
||||
grub_ssize_t ret;
|
||||
|
||||
gzio = grub_zalloc (sizeof (*gzio));
|
||||
if (! gzio)
|
||||
return -1;
|
||||
gzio->mem_input = (grub_uint8_t *) inbuf;
|
||||
gzio->mem_input_size = insize;
|
||||
gzio->mem_input_off = 0;
|
||||
|
||||
initialize_tables (gzio);
|
||||
|
||||
ret = grub_gzio_read_real (gzio, off, outbuf, outsize);
|
||||
grub_free (gzio);
|
||||
|
||||
return ret;
|
||||
}
|
||||
|
||||
|
||||
|
||||
static struct grub_fs grub_gzio_fs =
|
||||
|
|
|
|||
233
grub-core/loader/i386/pc/truecrypt.c
Normal file
233
grub-core/loader/i386/pc/truecrypt.c
Normal file
|
|
@ -0,0 +1,233 @@
|
|||
/* chainloader.c - boot another boot loader */
|
||||
/*
|
||||
* GRUB -- GRand Unified Bootloader
|
||||
* Copyright (C) 2002,2004,2007,2009,2010 Free Software Foundation, Inc.
|
||||
*
|
||||
* GRUB is free software: you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation, either version 3 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* GRUB is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* You should have received a copy of the GNU General Public License
|
||||
* along with GRUB. If not, see <http://www.gnu.org/licenses/>.
|
||||
*/
|
||||
|
||||
#include <grub/loader.h>
|
||||
#include <grub/file.h>
|
||||
#include <grub/err.h>
|
||||
#include <grub/device.h>
|
||||
#include <grub/disk.h>
|
||||
#include <grub/misc.h>
|
||||
#include <grub/types.h>
|
||||
#include <grub/partition.h>
|
||||
#include <grub/dl.h>
|
||||
#include <grub/command.h>
|
||||
#include <grub/machine/biosnum.h>
|
||||
#include <grub/i18n.h>
|
||||
#include <grub/video.h>
|
||||
#include <grub/mm.h>
|
||||
#include <grub/cpu/relocator.h>
|
||||
#include <grub/deflate.h>
|
||||
|
||||
GRUB_MOD_LICENSE ("GPLv3+");
|
||||
|
||||
static grub_dl_t my_mod;
|
||||
static struct grub_relocator *rel;
|
||||
static grub_uint32_t edx = 0xffffffff;
|
||||
static grub_uint16_t sp;
|
||||
static grub_uint32_t destaddr;
|
||||
|
||||
#define GRUB_TRUECRYPT_SEGMENT 0x2000
|
||||
|
||||
static grub_err_t
|
||||
grub_truecrypt_boot (void)
|
||||
{
|
||||
grub_uint16_t segment = destaddr >> 4;
|
||||
struct grub_relocator16_state state = {
|
||||
.cs = segment,
|
||||
.ds = segment,
|
||||
.es = segment,
|
||||
.fs = segment,
|
||||
.gs = segment,
|
||||
.ss = segment,
|
||||
.ip = 0x100,
|
||||
.sp = sp,
|
||||
.edx = edx,
|
||||
.a20 = 1
|
||||
};
|
||||
grub_video_set_mode ("text", 0, 0);
|
||||
|
||||
return grub_relocator16_boot (rel, state);
|
||||
}
|
||||
|
||||
static grub_err_t
|
||||
grub_truecrypt_unload (void)
|
||||
{
|
||||
grub_relocator_unload (rel);
|
||||
rel = NULL;
|
||||
grub_dl_unref (my_mod);
|
||||
return GRUB_ERR_NONE;
|
||||
}
|
||||
|
||||
/* Information on protocol supplied by Attila Lendvai. */
|
||||
#define MAGIC "\0CD001\1EL TORITO SPECIFICATION"
|
||||
|
||||
static grub_err_t
|
||||
grub_cmd_truecrypt (grub_command_t cmd __attribute__ ((unused)),
|
||||
int argc, char *argv[])
|
||||
{
|
||||
grub_file_t file = 0;
|
||||
grub_err_t err;
|
||||
void *truecrypt;
|
||||
grub_size_t truecryptsize;
|
||||
const grub_size_t truecryptmemsize = 42 * 1024;
|
||||
grub_uint8_t dh;
|
||||
grub_uint32_t catalog, rba;
|
||||
grub_uint8_t buf[128];
|
||||
char *compressed = NULL;
|
||||
char *uncompressed = NULL;
|
||||
|
||||
if (argc == 0)
|
||||
return grub_error (GRUB_ERR_BAD_ARGUMENT, N_("filename expected"));
|
||||
|
||||
rel = NULL;
|
||||
|
||||
grub_dl_ref (my_mod);
|
||||
|
||||
file = grub_file_open (argv[0]);
|
||||
if (! file)
|
||||
goto fail;
|
||||
|
||||
if (grub_file_seek (file, 17 * 2048) == (grub_size_t) -1)
|
||||
goto fail;
|
||||
|
||||
if (grub_file_read (file, buf, sizeof (buf))
|
||||
!= sizeof (buf))
|
||||
goto fail;
|
||||
|
||||
if (grub_memcmp (buf, MAGIC, sizeof (MAGIC)) != 0)
|
||||
{
|
||||
grub_error (GRUB_ERR_BAD_OS, "invalid eltorito signature");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
catalog = grub_get_unaligned32 (buf + 0x47);
|
||||
|
||||
if (grub_file_seek (file, catalog * 2048) == (grub_size_t)-1)
|
||||
goto fail;
|
||||
|
||||
if (grub_file_read (file, buf, sizeof (buf))
|
||||
!= sizeof (buf))
|
||||
goto fail;
|
||||
|
||||
if (buf[0] != 1 || buf[1] != 0 || buf[0x1e] != 0x55
|
||||
|| buf[0x1f] != 0xaa || buf[0x20] != 0x88
|
||||
|| buf[0x26] != 1 || buf[0x27] != 0)
|
||||
{
|
||||
grub_error (GRUB_ERR_BAD_OS, "invalid eltorito catalog");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
rba = grub_get_unaligned32 (buf + 0x28);
|
||||
|
||||
if (grub_file_seek (file, rba * 2048 + 0x1b7) == (grub_size_t) -1)
|
||||
goto fail;
|
||||
|
||||
if (grub_file_read (file, &dh, 1)
|
||||
!= 1)
|
||||
goto fail;
|
||||
|
||||
if (grub_file_seek (file, rba * 2048 + 512 + 2048) == (grub_size_t) -1)
|
||||
goto fail;
|
||||
|
||||
compressed = grub_malloc (57 * 512);
|
||||
if (!compressed)
|
||||
goto fail;
|
||||
|
||||
if (grub_file_read (file, compressed, 57 * 512)
|
||||
!= 57 * 512)
|
||||
goto fail;
|
||||
|
||||
uncompressed = grub_malloc (truecryptmemsize);
|
||||
if (!uncompressed)
|
||||
goto fail;
|
||||
|
||||
/* It's actually gzip but our gzip decompressor isn't able to handle
|
||||
trailing garbage, hence let's use raw decompressor. */
|
||||
truecryptsize = grub_deflate_decompress (compressed + 10, 57 * 512 - 10,
|
||||
0, uncompressed, truecryptmemsize);
|
||||
if ((grub_ssize_t) truecryptsize < 0)
|
||||
goto fail;
|
||||
|
||||
if (truecryptmemsize <= truecryptsize + 0x100)
|
||||
{
|
||||
grub_error (GRUB_ERR_BAD_OS, "file is too big");
|
||||
goto fail;
|
||||
}
|
||||
|
||||
rel = grub_relocator_new ();
|
||||
if (!rel)
|
||||
goto fail;
|
||||
|
||||
edx = (dh << 8) | grub_get_root_biosnumber ();
|
||||
|
||||
destaddr = ALIGN_DOWN (grub_min (0x90000, grub_mmap_get_lower ())
|
||||
- truecryptmemsize, 64 * 1024);
|
||||
|
||||
{
|
||||
grub_relocator_chunk_t ch;
|
||||
err = grub_relocator_alloc_chunk_addr (rel, &ch, destaddr,
|
||||
truecryptmemsize);
|
||||
if (err)
|
||||
goto fail;
|
||||
truecrypt = get_virtual_current_address (ch);
|
||||
}
|
||||
|
||||
grub_memset (truecrypt, 0, 0x100);
|
||||
grub_memcpy ((char *) truecrypt + 0x100, uncompressed, truecryptsize);
|
||||
|
||||
grub_memset ((char *) truecrypt + truecryptsize + 0x100,
|
||||
0, truecryptmemsize - truecryptsize - 0x100);
|
||||
sp = truecryptmemsize - 4;
|
||||
|
||||
grub_loader_set (grub_truecrypt_boot, grub_truecrypt_unload, 1);
|
||||
|
||||
grub_free (uncompressed);
|
||||
grub_free (compressed);
|
||||
|
||||
return GRUB_ERR_NONE;
|
||||
|
||||
fail:
|
||||
|
||||
if (!grub_errno)
|
||||
return grub_error (GRUB_ERR_BAD_OS, "bad truecrypt ISO");
|
||||
|
||||
if (file)
|
||||
grub_file_close (file);
|
||||
|
||||
grub_truecrypt_unload ();
|
||||
|
||||
grub_free (uncompressed);
|
||||
grub_free (compressed);
|
||||
|
||||
return grub_errno;
|
||||
}
|
||||
|
||||
static grub_command_t cmd;
|
||||
|
||||
GRUB_MOD_INIT(truecrypt)
|
||||
{
|
||||
cmd = grub_register_command ("truecrypt", grub_cmd_truecrypt,
|
||||
0, N_("Load Truecrypt ISO."));
|
||||
my_mod = mod;
|
||||
}
|
||||
|
||||
GRUB_MOD_FINI(truecrypt)
|
||||
{
|
||||
grub_unregister_command (cmd);
|
||||
}
|
||||
|
|
@ -23,4 +23,8 @@ grub_ssize_t
|
|||
grub_zlib_decompress (char *inbuf, grub_size_t insize, grub_off_t off,
|
||||
char *outbuf, grub_size_t outsize);
|
||||
|
||||
grub_ssize_t
|
||||
grub_deflate_decompress (char *inbuf, grub_size_t insize, grub_off_t off,
|
||||
char *outbuf, grub_size_t outsize);
|
||||
|
||||
#endif
|
||||
|
|
|
|||
Loading…
Reference in a new issue