summaryrefslogtreecommitdiffabout
path: root/pwmanager/pwmanager/compressbzip2.cpp
blob: c3a9edd783d97721dd0b3d584a4538088df619b7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
/***************************************************************************
 *                                                                         *
 *   copyright (C) 2003, 2004 by Michael Buesch                            *
 *   email: mbuesch@freenet.de                                             *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License version 2        *
 *   as published by the Free Software Foundation.                         *
 *                                                                         *
 ***************************************************************************/

/***************************************************************************
 * copyright (C) 2004 by Ulf Schenk
 * This file is originaly based on version 1.0.1 of pwmanager
 * and was modified to run on embedded devices that run microkde
 *
 * $Id$
 **************************************************************************/  

#include "compressbzip2.h"

#include <stdlib.h>

#define BZ_NO_STDIO
#include <bzlib.h>

#define BZ_BLOCK_SIZE				9
#define BZ_WORK_FACTOR				30
#define EXPAND_COMPRESS_DESTBUF_BYTES		1024
#define EXPAND_DECOMPRESS_DESTBUF_BYTES		(1024 * 10)


bool CompressBzip2::compress(string *d)
{
	int ret;
	char *dest, *source;
	unsigned int sourceLen, destLen;

	sourceLen = d->length();
	destLen = calcCompressDestLen(sourceLen);
	source = static_cast<char *>(malloc(sourceLen));
	if (!source)
		return false;
	memcpy(source, d->c_str(), sourceLen);
	dest = static_cast<char *>(malloc(destLen));
	if (!dest) {
		free(source);
		return false;
	}
	while (1) {
		ret = BZ2_bzBuffToBuffCompress(dest,
					       &destLen,
					       source,
					       sourceLen,
					       BZ_BLOCK_SIZE,
					       0,
					       BZ_WORK_FACTOR);
		switch (ret) {
		case BZ_OK:
			goto out;
		case BZ_OUTBUFF_FULL:
			/* we don't use realloc(), because it may
			 * (in this case) unneccessarily copy the old block
			 * to the new allocated block.
			 */
			free(dest);
			destLen += EXPAND_COMPRESS_DESTBUF_BYTES;
			dest = static_cast<char *>(malloc(destLen));
			if (!dest) {
				free(source);
				return false;
			}
			break;
		default:
			free(source),
			free(dest);
			return false;
		}
	}
out:
	free(source);
	d->assign(dest, destLen);
	free(dest);
	return true;
}

bool CompressBzip2::decompress(string *d)
{
	int ret;
	char *dest, *source;
	unsigned int sourceLen, destLen;

	sourceLen = d->length();
	destLen = calcDecompressDestLen(sourceLen);
	source = static_cast<char *>(malloc(sourceLen));
	if (!source)
		return false;
	memcpy(source, d->c_str(), sourceLen);
	dest = static_cast<char *>(malloc(destLen));
	if (!dest) {
		free(source);
		return false;
	}
	while (1) {
		ret = BZ2_bzBuffToBuffDecompress(dest,
						 &destLen,
						 source,
						 sourceLen,
						 0,
						 0);
		switch (ret) {
		case BZ_OK:
			goto out;
		case BZ_OUTBUFF_FULL:
			/* we don't use realloc(), because it may
			 * (in this case) unneccessarily copy the old block
			 * to the new allocated block.
			 */
			free(dest);
			destLen += EXPAND_DECOMPRESS_DESTBUF_BYTES;
			dest = static_cast<char *>(malloc(destLen));
			if (!dest) {
				free(source);
				return false;
			}
			break;
		default:
			free(source),
			free(dest);
			return false;
		}
	}
out:
	free(source);
	d->assign(dest, destLen);
	free(dest);
	return true;
}