summaryrefslogtreecommitdiff
path: root/libstore/encode.c
blob: df27250d1f22983081fbec5cd125a42551798f71 (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
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
/* Store wire encoding

   Copyright (C) 1996,97,99,2001,02 Free Software Foundation, Inc.
   Written by Miles Bader <miles@gnu.org>
   This file is part of the GNU Hurd.

   The GNU Hurd 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 2, or (at
   your option) any later version.

   The GNU Hurd 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 this program; if not, write to the Free Software
   Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA. */

#include <string.h>
#include <sys/mman.h>

#include "store.h"

/* Standard encoding used for most leaf store types.  */

error_t
store_std_leaf_allocate_encoding (const struct store *store,
				  struct store_enc *enc)
{
  enc->num_ports++;
  enc->num_ints += 6;
  enc->num_offsets += store->num_runs * 2;
  if (store->name)
    enc->data_len += strlen (store->name) + 1;
  enc->data_len += store->misc_len;
  return 0;
}

/* The RPC protocol uses 32-bit off_t's, but store_offset_t is now 64 bits.  */
static inline int too_big (store_offset_t ofs)
{
  off_t o = (off_t) ofs;
  return o < 0 || ((store_offset_t) o != ofs);
}

error_t
store_std_leaf_encode (const struct store *store, struct store_enc *enc)
{
  int i;
  size_t name_len = (store->name ? strlen (store->name) + 1 : 0);

  enc->ports[enc->cur_port++] = store->port;

  enc->ints[enc->cur_int++] = store->class->id;
  enc->ints[enc->cur_int++] = store->flags;
  enc->ints[enc->cur_int++] = store->block_size;
  enc->ints[enc->cur_int++] = store->num_runs;
  enc->ints[enc->cur_int++] = name_len;
  enc->ints[enc->cur_int++] = store->misc_len;

  for (i = 0; i < store->num_runs; i++)
    {
      if (sizeof (*enc->offsets) != sizeof (store->runs[i].start)
	  && (too_big (store->runs[i].start)
	      || too_big (store->runs[i].start + store->runs[i].length)))
	return EOVERFLOW;
      enc->offsets[enc->cur_offset++] = store->runs[i].start;
      enc->offsets[enc->cur_offset++] = store->runs[i].length;
    }

  if (store->name)
    {
      bcopy (store->name, enc->data + enc->cur_data, name_len);
      enc->cur_data += name_len;
    }
  if (store->misc_len)
    {
      bcopy (store->misc, enc->data + enc->cur_data, store->misc_len);
      enc->cur_data += store->misc_len;
    }

  return 0;
}

/* Encode STORE into ENC, which should have been prepared with
   store_enc_init, or return an error.  The contents of ENC may then be
   return as the value of file_get_storage_info; if for some reason this
   can't be done, store_enc_dealloc may be used to deallocate the mmemory
   used by the unsent vectors.  */
error_t
store_encode (const struct store *store, struct store_enc *enc)
{
  void *buf;
  error_t err;
  const struct store_class *class = store->class;
  /* We zero each vector length for the allocate_encoding method to work, so
     save the old values.  */
  mach_msg_type_number_t init_num_ports = enc->num_ports;
  mach_msg_type_number_t init_num_ints = enc->num_ints;
  mach_msg_type_number_t init_num_offsets = enc->num_offsets;
  mach_msg_type_number_t init_data_len = enc->data_len;

  if (!class->allocate_encoding || !class->encode)
    return EOPNOTSUPP;

  enc->num_ports = 0;
  enc->num_ints = 0;
  enc->num_offsets = 0;
  enc->data_len = 0;
  err = (*class->allocate_encoding) (store, enc);
  if (err)
    return err;

  errno = 0;
  if (enc->num_ports > init_num_ports)
    {
      buf = mmap (0, enc->num_ports * sizeof *enc->ports,
		  PROT_READ|PROT_WRITE, MAP_ANON, 0, 0);
      if (buf != (void *) -1)
	enc->ports = buf;
    }
  if (!errno && enc->num_ints > init_num_ints)
    {
      buf = mmap (0, enc->num_ints * sizeof *enc->ints,
		  PROT_READ|PROT_WRITE, MAP_ANON, 0, 0);
      if (buf != (void *) -1)
	enc->ints = buf;
    }
  if (!errno && enc->num_offsets > init_num_offsets)
    {
      buf = mmap (0, enc->num_offsets * sizeof *enc->offsets,
		  PROT_READ|PROT_WRITE, MAP_ANON, 0, 0);
      if (buf != (void *) -1)
	enc->offsets = buf;

    }
  if (!errno && enc->data_len > init_data_len)
    {
      buf = mmap (0, enc->data_len, PROT_READ|PROT_WRITE, MAP_ANON, 0, 0);
      if (buf != (void *) -1)
	enc->data = buf;
    }
  err = errno;
  if (! err)
    err = (*class->encode) (store, enc);

  enc->cur_port = enc->cur_int = enc->cur_offset = enc->cur_data = 0;

  if (err)
    store_enc_dealloc (enc);

  return err;
}

/* Encode STORE into the given return variables, suitably for returning from a
   file_get_storage_info rpc.  */
error_t
store_return (const struct store *store,
	      mach_port_t **ports, mach_msg_type_number_t *num_ports,
	      int **ints, mach_msg_type_number_t *num_ints,
	      off_t **offsets, mach_msg_type_number_t *num_offsets,
	      char **data, mach_msg_type_number_t *data_len)
{
  error_t err;
  struct store_enc enc;

  store_enc_init (&enc, *ports, *num_ports, *ints, *num_ints,
		  *offsets, *num_offsets, *data, *data_len);
  err = store_encode (store, &enc);
  if (err)
    store_enc_dealloc (&enc);
  else
    store_enc_return (&enc, ports, num_ports, ints, num_ints,
		      offsets, num_offsets, data, data_len);
  return err;
}