aboutsummaryrefslogtreecommitdiffstats
path: root/src/ea_map.c
blob: e8d103e70a106352cfb51a6cf8e41a56772f9e25 (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
#include "project.h"
#include <db.h>

static DB *ea_map_db = NULL;

static int ea_map_init (void)
{
  int ret;

  ret = db_create (&ea_map_db, NULL, 0);

  if (ret) {
    ea_map_db = NULL;
    return -1;
  }

  ret = ea_map_db->open (ea_map_db, NULL, NULL, NULL, DB_BTREE, DB_CREATE, 0x600);

  if (ret) {
    ea_map_db->err (ea_map_db, ret, "ea_map_db");
    ea_map_db->close (ea_map_db, 0);
    ea_map_db = NULL;
    return -1;
  }

  return 0;
}

int ea_map_lookup (blk64_t src_blk, blk64_t *dst_blk, blk64_t *len)
{
  DBT key = {0}, data = {0};
  blk64_t *buf;

  if (!ea_map_db)
    return -1;

  key.size = sizeof (src_blk);
  key.data = &src_blk;

  if (ea_map_db->get (ea_map_db, NULL, &key, &data, 0))
    return -1;

  if (data.size != (2 * sizeof (blk64_t)))
    return -1;

  buf = (blk64_t *) data.data;

  *dst_blk = buf[0];
  *len = buf[1];

  return 0;
}


int ea_map_set (blk64_t src_blk, blk64_t dst_blk, blk64_t len)
{
  DBT key = {0}, data = {0};
  blk64_t buf[2] = {dst_blk, len};
  int ret;

  if (!ea_map_db)
    if (ea_map_init())
      return -1;


  key.size = sizeof (src_blk);
  key.data = &src_blk;
  data.size = 2 * sizeof (dst_blk);
  data.data = buf;

  ret = ea_map_db->put (ea_map_db, NULL, &key, &data, 0);

  if (ret)
    ea_map_db->err (ea_map_db, ret, "DB->put");

  return ret;
}