Work
[yaz-moved-to-github.git] / odr / ber_int.c
1 /*
2  * Copyright (c) 1995-2000, Index Data
3  * See the file LICENSE for details.
4  * Sebastian Hammer, Adam Dickmeiss
5  *
6  * $Log: ber_int.c,v $
7  * Revision 1.19  2000-10-06 12:46:27  adam
8  * Including sys/types.h and netinet/in.h instead on arpa/inet.h on Unix.
9  *
10  * Revision 1.18  2000/02/29 13:44:55  adam
11  * Check for config.h (currently not generated).
12  *
13  * Revision 1.17  2000/02/28 11:20:06  adam
14  * Using autoconf. New definitions: YAZ_BEGIN_CDECL/YAZ_END_CDECL.
15  *
16  * Revision 1.16  2000/01/31 13:15:21  adam
17  * Removed uses of assert(3). Cleanup of ODR. CCL parser update so
18  * that some characters are not surrounded by spaces in resulting term.
19  * ILL-code updates.
20  *
21  * Revision 1.15  1999/11/30 13:47:11  adam
22  * Improved installation. Moved header files to include/yaz.
23  *
24  * Revision 1.14  1999/05/26 07:49:35  adam
25  * C++ compilation.
26  *
27  * Revision 1.13  1999/01/08 11:23:22  adam
28  * Added const modifier to some of the BER/ODR encoding routines.
29  *
30  * Revision 1.12  1996/07/06 19:58:33  quinn
31  * System headerfiles gathered in yconfig
32  *
33  * Revision 1.11  1995/09/29  17:12:16  quinn
34  * Smallish
35  *
36  * Revision 1.10  1995/09/29  17:01:50  quinn
37  * More Windows work
38  *
39  * Revision 1.9  1995/09/28  10:12:39  quinn
40  * Windows-support changes
41  *
42  * Revision 1.8  1995/09/27  15:02:55  quinn
43  * Modified function heads & prototypes.
44  *
45  * Revision 1.7  1995/05/16  08:50:44  quinn
46  * License, documentation, and memory fixes
47  *
48  * Revision 1.6  1995/04/18  08:15:14  quinn
49  * Added dynamic memory allocation on encoding (whew). Code is now somewhat
50  * neater. We'll make the same change for decoding one day.
51  *
52  * Revision 1.5  1995/03/27  15:01:44  quinn
53  * Added include of sys/types to further portability
54  *
55  * Revision 1.4  1995/03/08  12:12:07  quinn
56  * Added better error checking.
57  *
58  * Revision 1.3  1995/02/09  15:51:46  quinn
59  * Works better now.
60  *
61  * Revision 1.2  1995/02/07  17:52:58  quinn
62  * A damn mess, but now things work, I think.
63  *
64  * Revision 1.1  1995/02/02  16:21:52  quinn
65  * First kick.
66  *
67  */
68 #if HAVE_CONFIG_H
69 #include <config.h>
70 #endif
71
72 #include <string.h>
73
74 #ifdef WIN32
75 #include <winsock.h>
76 #else
77 #include <sys/types.h>
78 #include <netinet/in.h>
79 #endif
80
81 #include <yaz/odr.h>
82
83 static int ber_encinteger(ODR o, int val);
84 static int ber_decinteger(const unsigned char *buf, int *val);
85
86 int ber_integer(ODR o, int *val)
87 {
88     int res;
89
90     switch (o->direction)
91     {
92         case ODR_DECODE:
93             if ((res = ber_decinteger(o->bp, val)) <= 0)
94             {
95                 o->error = OPROTO;
96                 return 0;
97             }
98             o->bp += res;
99             return 1;
100         case ODR_ENCODE:
101             if ((res = ber_encinteger(o, *val)) < 0)
102                 return 0;
103             return 1;
104         case ODR_PRINT: return 1;
105         default: o->error = OOTHER;  return 0;
106     }
107 }
108
109 /*
110  * Returns: number of bytes written or -1 for error (out of bounds).
111  */
112 int ber_encinteger(ODR o, int val)
113 {
114     int lenpos;
115     int a, len;
116     union { int i; unsigned char c[sizeof(int)]; } tmp;
117
118     lenpos = odr_tell(o);
119     if (odr_putc(o, 0) < 0)  /* dummy */
120         return -1;
121     tmp.i = htonl(val);   /* ensure that that we're big-endian */
122     for (a = 0; a < (int) sizeof(int) - 1; a++)  /* skip superfluous octets */
123         if (!((tmp.c[a] == 0 && !(tmp.c[a+1] & 0X80)) ||
124             (tmp.c[a] == 0XFF && (tmp.c[a+1] & 0X80))))
125             break;
126     len = sizeof(int) - a;
127     if (odr_write(o, (unsigned char*) tmp.c + a, len) < 0)
128         return -1;
129     odr_seek(o, ODR_S_SET, lenpos);
130     if (ber_enclen(o, len, 1, 1) != 1)
131         return -1;
132     odr_seek(o, ODR_S_END, 0);
133 #ifdef ODR_DEBUG
134     fprintf(stderr, "[val=%d]", val);
135 #endif
136     return 0;
137 }
138
139 /*
140  * Returns: Number of bytes read or 0 if no match, -1 if error.
141  */
142 int ber_decinteger(const unsigned char *buf, int *val)
143 {
144     const unsigned char *b = buf;
145     unsigned char fill;
146     int res, len, remains;
147     union { int i; unsigned char c[sizeof(int)]; } tmp;
148
149     if ((res = ber_declen(b, &len)) < 0)
150         return -1;
151     if (len > (int) sizeof(int))    /* let's be reasonable, here */
152         return -1;
153     b+= res;
154
155     remains = sizeof(int) - len;
156     memcpy(tmp.c + remains, b, len);
157     if (*b & 0X80)
158         fill = 0XFF;
159     else
160         fill = 0X00;
161     memset(tmp.c, fill, remains);
162     *val = ntohl(tmp.i);
163
164     b += len;
165 #ifdef ODR_DEBUG
166     fprintf(stderr, "[val=%d]", *val);
167 #endif
168     return b - buf;
169 }