ncurses 6.0 - patch 20170506

[ncurses.git] / ncurses / tinfo / getenv_num.c

diff --git a/ncurses/tinfo/getenv_num.c b/ncurses/tinfo/getenv_num.c
index 62944048175ac90024ba7dba1b6d056d372b9e4b..d5e35cb46638d9236e01024beb7eba2b8c302391 100644 (file)
--- a/ncurses/tinfo/getenv_num.c
+++ b/ncurses/tinfo/getenv_num.c
@@ -1,5 +1,5 @@
 /****************************************************************************
- * Copyright (c) 1998 Free Software Foundation, Inc.                        *
+ * Copyright (c) 1998-2012,2013 Free Software Foundation, Inc.              *
  *                                                                          *
  * Permission is hereby granted, free of charge, to any person obtaining a  *
  * copy of this software and associated documentation files (the            *
@@ -36,21 +36,38 @@
 
 #include <curses.priv.h>
 
-MODULE_ID("$Id: getenv_num.c,v 1.1 1998/09/19 21:30:23 tom Exp $")
+MODULE_ID("$Id: getenv_num.c,v 1.6 2013/09/28 20:25:08 tom Exp $")
 
-int
+NCURSES_EXPORT(int)
 _nc_getenv_num(const char *name)
 {
-       char *dst = 0;
-       char *src = getenv(name);
-       long value;
-
-       if ((src == 0)
-        || (value = strtol(src, &dst, 0)) < 0
-        || (dst == src)
-        || (*dst != '\0')
-        || (int)value < value)
-               value = -1;
-
-       return (int) value;
+    char *dst = 0;
+    char *src = getenv(name);
+    long value;
+
+    if ((src == 0)
+       || (value = strtol(src, &dst, 0)) < 0
+       || (dst == src)
+       || (*dst != '\0')
+       || (int) value < value)
+       value = -1;
+
+    return (int) value;
+}
+
+NCURSES_EXPORT(void)
+_nc_setenv_num(const char *name, int value)
+{
+    if (name != 0 && value >= 0) {
+       char buffer[128];
+#if HAVE_SETENV
+       _nc_SPRINTF(buffer, _nc_SLIMIT(sizeof(buffer)) "%d", value);
+       setenv(name, buffer, 1);
+#elif HAVE_PUTENV
+       char *s;
+       _nc_SPRINTF(buffer, _nc_SLIMIT(sizeof(buffer)) "%s=%d", name, value);
+       if ((s = strdup(buffer)) != 0)
+           putenv(s);
+#endif
+    }
 }