Fixed possible buf in proto.c

[yaz-moved-to-github.git] / odr / ber_int.c

diff --git a/odr/ber_int.c b/odr/ber_int.c
index a5a8805..4f036c4 100644 (file)
--- a/odr/ber_int.c
+++ b/odr/ber_int.c
@@ -4,7 +4,13 @@
  * Sebastian Hammer, Adam Dickmeiss
  *
  * $Log: ber_int.c,v $
- * Revision 1.3  1995-02-09 15:51:46  quinn
+ * Revision 1.5  1995-03-27 15:01:44  quinn
+ * Added include of sys/types to further portability
+ *
+ * Revision 1.4  1995/03/08  12:12:07  quinn
+ * Added better error checking.
+ *
+ * Revision 1.3  1995/02/09  15:51:46  quinn
  * Works better now.
  *
  * Revision 1.2  1995/02/07  17:52:58  quinn
@@ -16,7 +22,8 @@
  */
 
 #include <odr.h>
-#include <netinet/in.h>
+#include <sys/types.h>
+#include <netinet/in.h>  /* for htons... */
 #include <string.h>
 
 static int ber_encinteger(unsigned char *buf, int val, int maxlen);
@@ -30,18 +37,24 @@ int ber_integer(ODR o, int *val)
     {
        case ODR_DECODE:
            if ((res = ber_decinteger(o->bp, val)) <= 0)
+           {
+               o->error = OPROTO;
                return 0;
+           }
            o->bp += res;
            o->left -= res;
            return 1;
        case ODR_ENCODE:
            if ((res = ber_encinteger(o->bp, *val, o->left)) <= 0)
+           {
+               o->error = OSPACE;
                return 0;
+           }
            o->bp += res;
            o->left -= res;
            return 1;
        case ODR_PRINT: return 1;
-       default:  return 0;
+       default: o->error = OOTHER;  return 0;
     }
 }