server/utils.cc at master · Eraru/server · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
#include "utils.h"

#include <fstream>
#ifdef DEBUG
	#include <iostream>
#endif

#include <unistd.h>
#include <sys/reboot.h>

#include "constants.h"

using namespace std;
using namespace os;

void os::check_or_create(const string& path, Logger* logger)
{
	if ( ! file_exists(path))
	{
		if (logger != NULL)
			logger->log("No '"+path+"' directory, creating...");
		#ifdef DEBUG
			else
				cout << "[OpenStratos] No '"+path+"' directory, creating..." << endl;
		#endif
		if (mkdir(path.c_str(), 0755) != 0)
		{
			if (logger != NULL)
				logger->log("Error creating '"+path+"' directory.");
			#ifdef DEBUG
				else
					cout << "[OpenStratos] Error creating '"+path+"' directory." << endl;
			#endif

			sync();
			reboot(RB_POWER_OFF);
		}
		else
		{
			if (logger != NULL)
				logger->log("'"+path+"' directory created.");
			#ifdef DEBUG
				else
					cout << "[OpenStratos] '"+path+"' directory created." << endl;
			#endif
		}
	}
}

State os::set_state(State new_state)
{
	ofstream state_file(STATE_FILE);
	state_file << state_to_string(new_state);
	state_file.close();

	return new_state;
}

State os::get_last_state()
{
	ifstream state_file(STATE_FILE);
	string state_str((istreambuf_iterator<char>(state_file)),
				 istreambuf_iterator<char>());
	state_file.close();

	if (state_str == "INITIALIZING") return INITIALIZING;
	if (state_str == "ACQUIRING_FIX") return ACQUIRING_FIX;
	if (state_str == "FIX_ACQUIRED") return FIX_ACQUIRED;
	if (state_str == "WAITING_LAUNCH") return WAITING_LAUNCH;
	if (state_str == "GOING_UP") return GOING_UP;
	if (state_str == "GOING_DOWN") return GOING_DOWN;
	if (state_str == "LANDED") return LANDED;
	if (state_str == "SHUT_DOWN") return SHUT_DOWN;

	return SAFE_MODE;
}

const string os::state_to_string(State state)
{
	switch (state)
	{
		case INITIALIZING:
			return "INITIALIZING";
		break;
		case ACQUIRING_FIX:
			return "ACQUIRING_FIX";
		break;
		case FIX_ACQUIRED:
			return "FIX_ACQUIRED";
		break;
		case WAITING_LAUNCH:
			return "WAITING_LAUNCH";
		break;
		case GOING_UP:
			return "GOING_UP";
		break;
		case GOING_DOWN:
			return "GOING_DOWN";
		break;
		case LANDED:
			return "LANDED";
		break;
		case SHUT_DOWN:
			return "SHUT_DOWN";
		break;
		case SAFE_MODE:
			return "SAFE_MODE";
	}
}