d799ee53d7
git-svn-id: https://svn.code.sf.net/p/nsis/code/NSIS/trunk@6439 212acab6-be3b-0410-9dea-997c60f758d6
119 lines
2.4 KiB
C++
119 lines
2.4 KiB
C++
/*
|
|
* writer.cpp
|
|
*
|
|
* This file is a part of NSIS.
|
|
*
|
|
* Copyright (C) 1999-2013 Nullsoft and Contributors
|
|
*
|
|
* Licensed under the zlib/libpng license (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
*
|
|
* Licence details can be found in the file COPYING.
|
|
*
|
|
* This software is provided 'as-is', without any express or implied
|
|
* warranty.
|
|
*
|
|
* Unicode support by Jim Park -- 08/13/2007
|
|
*/
|
|
|
|
#include "exehead/config.h"
|
|
#include "writer.h"
|
|
#include "growbuf.h"
|
|
#include "util.h"
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <stdexcept>
|
|
#include "tchar.h"
|
|
|
|
void writer_sink::write_byte(const unsigned char b)
|
|
{
|
|
write_data(&b, 1);
|
|
}
|
|
|
|
void writer_sink::write_short(const short s)
|
|
{
|
|
short fs = FIX_ENDIAN_INT16(s);
|
|
write_data(&fs, sizeof(short));
|
|
}
|
|
|
|
void writer_sink::write_int(const int i)
|
|
{
|
|
int fi = FIX_ENDIAN_INT32(i);
|
|
write_data(&fi, sizeof(int));
|
|
}
|
|
void writer_sink::write_int64(const INT64 i)
|
|
{
|
|
INT64 fi = FIX_ENDIAN_INT64(i);
|
|
write_data(&fi, sizeof(INT64));
|
|
}
|
|
|
|
void writer_sink::write_int_array(const int i[], const size_t len)
|
|
{
|
|
for (size_t l = 0; l < len; l++)
|
|
{
|
|
write_int(i[l]);
|
|
}
|
|
}
|
|
|
|
// size in this case is the length of the string to write.
|
|
void writer_sink::write_string(const TCHAR *s, size_t size)
|
|
{
|
|
#ifdef _UNICODE
|
|
if (m_build_unicode)
|
|
{
|
|
bool strEnd = false;
|
|
TCHAR ch;
|
|
for (; size ; size--)
|
|
{
|
|
if (!strEnd)
|
|
{
|
|
ch = *s++;
|
|
if (ch == _T('\0'))
|
|
strEnd = true;
|
|
}
|
|
write_short(ch);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
char *wb = new char[size];
|
|
memset(wb, 0, size);
|
|
WideCharToMultiByte(CP_ACP, 0, s, -1, wb, (int)size, NULL, NULL);
|
|
write_data(wb, size);
|
|
delete [] wb;
|
|
}
|
|
#else
|
|
char *wb = new char[size];
|
|
memset(wb, 0, size);
|
|
strncpy(wb, s, size);
|
|
write_data(wb, size);
|
|
delete [] wb;
|
|
#endif
|
|
}
|
|
|
|
void writer_sink::write_growbuf(const IGrowBuf *b)
|
|
{
|
|
write_data(b->get(), b->getlen());
|
|
}
|
|
|
|
void growbuf_writer_sink::write_data(const void *data, const size_t size)
|
|
{
|
|
m_buf->add(data, BUGBUG64TRUNCATE(int, size));
|
|
}
|
|
|
|
void file_writer_sink::write_data(const void *data, const size_t size)
|
|
{
|
|
if (fwrite(data, 1, size, m_fp) != size)
|
|
{
|
|
throw std::runtime_error("error writing");
|
|
}
|
|
}
|
|
|
|
#ifdef NSIS_CONFIG_CRC_SUPPORT
|
|
#include "crc32.h"
|
|
|
|
void crc_writer_sink::write_data(const void *data, const size_t size)
|
|
{
|
|
*m_crc = CRC32(*m_crc, (const unsigned char *) data, BUGBUG64TRUNCATE(unsigned int, size));
|
|
}
|
|
#endif
|