Merge branch 'main' of feistymeow.org:feisty_meow
[feisty_meow.git] / basis / environment.cpp
1 //////////////
2 // Name   : environment
3 // Author : Chris Koeritz
4 //////////////
5 // Copyright (c) 1994-$now By Author.  This program is free software; you can
6 // redistribute it and/or modify it under the terms of the GNU General Public
7 // License as published by the Free Software Foundation:
8 //     http://www.gnu.org/licenses/gpl.html
9 // or under the terms of the GNU Library license:
10 //     http://www.gnu.org/licenses/lgpl.html
11 // at your preference.  Those licenses describe your legal rights to this
12 // software, and no other rights or warranties apply.
13 // Please send updates for this code to: fred@gruntose.com -- Thanks, fred.
14 //////////////
15
16 #include "environment.h"
17
18 #include <stdlib.h>
19 #include <sys/types.h>
20 #ifndef _MSC_VER
21   #include <unistd.h>
22   #include <sys/times.h>
23 #endif
24 #ifdef _MSC_VER
25   #define _WINSOCKAPI_  // make windows.h happy about winsock.
26   // winsock support...
27 //  #undef FD_SETSIZE
28 //  #define FD_SETSIZE 1000
29     // if you don't set this, you can only select on a default of 64 sockets.
30   #include <winsock2.h>
31   #include <windows.h>
32   #include <mmsystem.h>
33 #endif
34
35 namespace basis {
36
37 astring environment::TMP()
38 {
39   const static astring TMP_VARIABLE_NAME("TMP");
40   astring to_return = get(TMP_VARIABLE_NAME);
41   if (!to_return) {
42     // they did not see fit to set this in the environment.  let's make something up.
43 #ifdef __WIN32__
44     // windows default does not necessarily exist.
45     to_return = "c:/tmp";
46 #else
47     // most reasonable OSes have a /tmp directory.
48     to_return = "/tmp";
49 #endif
50     if (!!to_return) set("TMP", to_return);
51   }
52   return to_return;
53 }
54
55 astring environment::get(const astring &variable_name)
56 {
57 #ifdef _MSC_VER
58   char *value = getenv(variable_name.upper().observe());
59     // dos & os/2 require upper case for the name, so we just do it that way.
60 #else
61   char *value = getenv(variable_name.observe());
62     // reasonable OSes support mixed-case environment variables.
63 #endif
64   astring to_return;
65   if (value)
66     to_return = astring(value);
67   return to_return;
68 }
69
70 bool environment::set(const astring &variable_name, const astring &value)
71 {
72   int ret = 0;
73 #ifdef _MSC_VER
74   astring assignment = variable_name + "=" + value;
75   ret = _putenv(assignment.s());
76 #else
77   ret = setenv(variable_name.s(), value.s(), true);
78 #endif
79   return !ret;
80 }
81
82 basis::un_int environment::system_uptime()
83 {
84 #ifdef _MSC_VER
85   return timeGetTime();
86 #else
87   static clock_t __ctps = sysconf(_SC_CLK_TCK);  // clock ticks per second.
88   static const double __multiplier = 1000.0 / double(__ctps);
89     // the multiplier gives us our full range for the tick counter.
90
91   // read uptime info from the OS.
92   tms uptime;
93   basis::un_int real_ticks = times(&uptime);
94
95   // now turn this into the number of milliseconds.
96   double ticks_up = (double)real_ticks;
97   ticks_up = ticks_up * __multiplier;  // convert to time here.
98
99   // we use the previous version of this calculation, which expected a basis::u_int
100   // to double conversion to provide a modulo operation rather than just leaving
101   // the basis::un_int at its maximum value (2^32-1).  however, that expectation is not
102   // guaranteed on some platforms (e.g., ARM processor with floating point
103   // emulation) and thus it becomes a bug around 49 days and 17 hours into
104   // OS uptime because the value gets stuck at 2^32-1 and never rolls over.
105   return basis::un_int(ticks_up);
106 #endif
107 }
108
109 } //namespace.
110