hc
2025-02-14 bbb9540dc49f70f6b703d1c8d1b85fa5f602d86e
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
/*
 * JFFS2 -- Journalling Flash File System, Version 2.
 *
 * Copyright © 2007 Nokia Corporation. All rights reserved.
 * Copyright © 2004-2010 David Woodhouse <dwmw2@infradead.org>
 *
 * Created by Richard Purdie <rpurdie@openedhand.com>
 *
 * For licensing information, see the file 'LICENCE' in this directory.
 *
 */
 
#include <linux/kernel.h>
#include <linux/sched.h>
#include <linux/vmalloc.h>
#include <linux/init.h>
#include <linux/lzo.h>
#include "compr.h"
 
static void *lzo_mem;
static void *lzo_compress_buf;
static DEFINE_MUTEX(deflate_mutex);    /* for lzo_mem and lzo_compress_buf */
 
static void free_workspace(void)
{
   vfree(lzo_mem);
   vfree(lzo_compress_buf);
}
 
static int __init alloc_workspace(void)
{
   lzo_mem = vmalloc(LZO1X_MEM_COMPRESS);
   lzo_compress_buf = vmalloc(lzo1x_worst_compress(PAGE_SIZE));
 
   if (!lzo_mem || !lzo_compress_buf) {
       free_workspace();
       return -ENOMEM;
   }
 
   return 0;
}
 
static int jffs2_lzo_compress(unsigned char *data_in, unsigned char *cpage_out,
                 uint32_t *sourcelen, uint32_t *dstlen)
{
   size_t compress_size;
   int ret;
 
   mutex_lock(&deflate_mutex);
   ret = lzo1x_1_compress(data_in, *sourcelen, lzo_compress_buf, &compress_size, lzo_mem);
   if (ret != LZO_E_OK)
       goto fail;
 
   if (compress_size > *dstlen)
       goto fail;
 
   memcpy(cpage_out, lzo_compress_buf, compress_size);
   mutex_unlock(&deflate_mutex);
 
   *dstlen = compress_size;
   return 0;
 
 fail:
   mutex_unlock(&deflate_mutex);
   return -1;
}
 
static int jffs2_lzo_decompress(unsigned char *data_in, unsigned char *cpage_out,
                uint32_t srclen, uint32_t destlen)
{
   size_t dl = destlen;
   int ret;
 
   ret = lzo1x_decompress_safe(data_in, srclen, cpage_out, &dl);
 
   if (ret != LZO_E_OK || dl != destlen)
       return -1;
 
   return 0;
}
 
static struct jffs2_compressor jffs2_lzo_comp = {
   .priority = JFFS2_LZO_PRIORITY,
   .name = "lzo",
   .compr = JFFS2_COMPR_LZO,
   .compress = &jffs2_lzo_compress,
   .decompress = &jffs2_lzo_decompress,
   .disabled = 0,
};
 
int __init jffs2_lzo_init(void)
{
   int ret;
 
   ret = alloc_workspace();
   if (ret < 0)
       return ret;
 
   ret = jffs2_register_compressor(&jffs2_lzo_comp);
   if (ret)
       free_workspace();
 
   return ret;
}
 
void jffs2_lzo_exit(void)
{
   jffs2_unregister_compressor(&jffs2_lzo_comp);
   free_workspace();
}