X-Git-Url: http://git.indexdata.com/?a=blobdiff_plain;f=odr%2Fber_int.c;h=21aa01f2ad32c7ba9de2fa2e9becf1ae3c02b380;hb=6aa60532542a01bfe78be0c91cbf57ce7cf851ce;hp=d259da33595d58df90fdb65c6d39d8c0f68e91f1;hpb=a7d28a25277a208edb9c44d124f179cc2be6db70;p=yaz-moved-to-github.git

diff --git a/odr/ber_int.c b/odr/ber_int.c
index d259da3..21aa01f 100644
--- a/odr/ber_int.c
+++ b/odr/ber_int.c
@@ -1,20 +1,27 @@
 /*
- * Copyright (C) 1994, Index Data I/S 
- * All rights reserved.
+ * Copyright (c) 1995-2003, Index Data
+ * See the file LICENSE for details.
  * Sebastian Hammer, Adam Dickmeiss
  *
- * $Log: ber_int.c,v $
- * Revision 1.1  1995-02-02 16:21:52  quinn
- * First kick.
- *
+ * $Id: ber_int.c,v 1.22 2003-03-11 11:03:31 adam Exp $
  */
+#if HAVE_CONFIG_H
+#include <config.h>
+#endif
 
-#include <odr.h>
-#include <netinet/in.h>
 #include <string.h>
 
-static int ber_encinteger(unsigned char *buf, int val, int maxlen);
-static int ber_decinteger(unsigned char *buf, int *val);
+#ifdef WIN32
+#include <winsock.h>
+#else
+#include <sys/types.h>
+#include <netinet/in.h>
+#endif
+
+#include "odr-priv.h"
+
+static int ber_encinteger(ODR o, int val);
+static int ber_decinteger(const unsigned char *buf, int *val, int max);
 
 int ber_integer(ODR o, int *val)
 {
@@ -22,76 +29,77 @@ int ber_integer(ODR o, int *val)
 
     switch (o->direction)
     {
-    	case ODR_DECODE:
-	    if ((res = ber_decinteger(o->bp, val)) <= 0)
-	    	return 0;
-	    o->bp += res;
-	    o->left -= res;
-	    return 1;
-    	case ODR_ENCODE:
-	    if ((res = ber_encinteger(o->bp, *val, o->left)) <= 0)
-	    	return 0;
-	    o->bp += res;
-	    o->left -= res;
-	    return 1;
-    	case ODR_PRINT: return 1;
-    	default:  return 0;
+        case ODR_DECODE:
+            if ((res = ber_decinteger(o->bp, val, odr_max(o))) <= 0)
+            {
+                odr_seterror(o, OPROTO, 50);
+                return 0;
+            }
+            o->bp += res;
+            return 1;
+        case ODR_ENCODE:
+            if ((res = ber_encinteger(o, *val)) < 0)
+                return 0;
+            return 1;
+        case ODR_PRINT: return 1;
+        default: odr_seterror(o, OOTHER, 51);  return 0;
     }
 }
 
 /*
  * Returns: number of bytes written or -1 for error (out of bounds).
  */
-int ber_encinteger(unsigned char *buf, int val, int maxlen)
+int ber_encinteger(ODR o, int val)
 {
-    unsigned char *b = buf, *lenpos;
     int a, len;
     union { int i; unsigned char c[sizeof(int)]; } tmp;
 
-    lenpos = b;
-    maxlen--;
-    b++;
-
     tmp.i = htonl(val);   /* ensure that that we're big-endian */
-    for (a = 0; a < sizeof(int) - 1; a++)  /* skip superfluous octets */
-    	if (!((tmp.c[a] == 0 && !(tmp.c[a+1] & 0X80)) ||
-	    (tmp.c[a] == 0XFF && (tmp.c[a+1] & 0X80))))
-	    break;
-    if ((len = sizeof(int) - a) > maxlen)
-    	return -1;
-    memcpy(b, tmp.c + a, len);
-    b += len;
-    if (ber_enclen(lenpos, len, 1, 1) != 1)
-    	return -1;
-    fprintf(stderr, "[val=%d]\n", val);
-    return b - buf;
+    for (a = 0; a < (int) sizeof(int) - 1; a++)  /* skip superfluous octets */
+        if (!((tmp.c[a] == 0 && !(tmp.c[a+1] & 0X80)) ||
+            (tmp.c[a] == 0XFF && (tmp.c[a+1] & 0X80))))
+            break;
+    len = sizeof(int) - a;
+    if (ber_enclen(o, len, 1, 1) != 1)
+        return -1;
+    if (odr_write(o, (unsigned char*) tmp.c + a, len) < 0)
+        return -1;
+#ifdef ODR_DEBUG
+    fprintf(stderr, "[val=%d]", val);
+#endif
+    return 0;
 }
 
 /*
  * Returns: Number of bytes read or 0 if no match, -1 if error.
  */
-int ber_decinteger(unsigned char *buf, int *val)
+int ber_decinteger(const unsigned char *buf, int *val, int max)
 {
-    unsigned char *b = buf, fill;
+    const unsigned char *b = buf;
+    unsigned char fill;
     int res, len, remains;
     union { int i; unsigned char c[sizeof(int)]; } tmp;
 
-    if ((res = ber_declen(b, &len)) < 0)
-    	return -1;
-    if (len > sizeof(int))    /* let's be reasonable, here */
-    	return -1;
+    if ((res = ber_declen(b, &len, max)) < 0)
+        return -1;
+    if (len+res > max || len < 0) /* out of bounds or indefinite encoding */
+        return -1;  
+    if (len > (int) sizeof(int))  /* let's be reasonable, here */
+        return -1;
     b+= res;
 
     remains = sizeof(int) - len;
     memcpy(tmp.c + remains, b, len);
     if (*b & 0X80)
-    	fill = 0XFF;
+        fill = 0XFF;
     else
-    	fill = 0X00;
+        fill = 0X00;
     memset(tmp.c, fill, remains);
     *val = ntohl(tmp.i);
 
     b += len;
-    fprintf(stderr, "[val=%d]\n", *val);
+#ifdef ODR_DEBUG
+    fprintf(stderr, "[val=%d]", *val);
+#endif
     return b - buf;
 }