[161] | 1 | /* Virginia Tech Cognitive Radio Open Source Systems |
---|
| 2 | * Virginia Tech, 2009 |
---|
| 3 | * |
---|
| 4 | * LICENSE INFORMATION GOES HERE |
---|
| 5 | */ |
---|
| 6 | |
---|
| 7 | /* DESCRIPTION OF FILE. |
---|
| 8 | */ |
---|
| 9 | |
---|
| 10 | |
---|
| 11 | #include <cstdlib> |
---|
| 12 | #include <cstring> |
---|
| 13 | #include <stdint.h> |
---|
| 14 | |
---|
| 15 | #include "vtcross/common.h" |
---|
| 16 | #include "vtcross/components.h" |
---|
| 17 | #include "vtcross/containers.h" |
---|
| 18 | #include "vtcross/debug.h" |
---|
| 19 | #include "vtcross/error.h" |
---|
| 20 | #include "vtcross/socketcomm.h" |
---|
| 21 | |
---|
| 22 | |
---|
| 23 | PolicyEngine::PolicyEngine() |
---|
| 24 | { |
---|
| 25 | LOG("Creating Policy Engine.\n"); |
---|
[181] | 26 | SML = NULL; |
---|
[161] | 27 | LoadPolicies(); |
---|
| 28 | } |
---|
| 29 | |
---|
| 30 | |
---|
| 31 | PolicyEngine::~PolicyEngine() |
---|
| 32 | { |
---|
[181] | 33 | if(SML) |
---|
| 34 | delete SML; |
---|
[161] | 35 | } |
---|
| 36 | |
---|
| 37 | |
---|
[181] | 38 | PolicyEngine::PolicyEngine(const char* serverName, const char* portNumber) |
---|
| 39 | { |
---|
| 40 | LOG("Creating Policy Engine.\n"); |
---|
| 41 | |
---|
| 42 | SML = new SML_Info; |
---|
| 43 | SML->hostname = std::string(serverName); |
---|
| 44 | SML->port = std::string(portNumber); |
---|
| 45 | |
---|
| 46 | SML->socketFD = ClientSocket(serverName, portNumber); |
---|
| 47 | RegisterServices(SML->socketFD); |
---|
| 48 | |
---|
| 49 | LOG("Policy Engine connected to SML at %s.\n", serverName); |
---|
| 50 | |
---|
| 51 | LoadPolicies(); |
---|
| 52 | } |
---|
| 53 | |
---|
| 54 | |
---|
[161] | 55 | void |
---|
| 56 | PolicyEngine::GetRemoteComponentType(int32_t socketFD) |
---|
| 57 | { |
---|
| 58 | } |
---|
| 59 | |
---|
| 60 | |
---|
| 61 | void |
---|
| 62 | PolicyEngine::WaitForSignal(int32_t socketFD) |
---|
| 63 | { |
---|
| 64 | LOG("Policy Engine:: Waiting for Policy Check Request.\n"); |
---|
| 65 | |
---|
| 66 | int32_t decision_array[10]; |
---|
| 67 | struct Parameter pList[10]; |
---|
| 68 | struct CE_Info ce_info; |
---|
[164] | 69 | |
---|
| 70 | // TODO should this be a GetRequest call or something else? |
---|
| 71 | if(GetRequest(socketFD, pList, &ce_info)) { |
---|
[161] | 72 | LOG("Policy Engine:: Validating Transmission Parameters.\n"); |
---|
| 73 | ValidateParameters(pList, &ce_info, decision_array); |
---|
| 74 | |
---|
| 75 | LOG("Policy Engine:: Sending Policy decision to Server.\n"); |
---|
| 76 | SendPEDecision(socketFD, pList, &ce_info, decision_array); |
---|
| 77 | } |
---|
| 78 | } |
---|
| 79 | |
---|
| 80 | |
---|
| 81 | void |
---|
| 82 | PolicyEngine::Shutdown() |
---|
| 83 | { |
---|
[182] | 84 | if(SML) |
---|
| 85 | DeregisterServices(SML->socketFD); |
---|
| 86 | // TODO otherwise, deregister from shell |
---|
[161] | 87 | } |
---|
| 88 | |
---|
| 89 | |
---|
| 90 | void |
---|
| 91 | PolicyEngine::Reset() |
---|
| 92 | { |
---|
| 93 | } |
---|
| 94 | |
---|
| 95 | |
---|
| 96 | void |
---|
| 97 | PolicyEngine::RegisterComponent(int32_t socketFD) |
---|
| 98 | { |
---|
[181] | 99 | SendMessage(socketFD, "register_engine_policy"); |
---|
[161] | 100 | LOG("Policy Engine:: Registration message sent.\n"); |
---|
[181] | 101 | |
---|
[161] | 102 | } |
---|
| 103 | |
---|
| 104 | |
---|
| 105 | void |
---|
| 106 | PolicyEngine::DeregisterComponent(int32_t socketFD) |
---|
| 107 | { |
---|
[182] | 108 | SendMessage(socketFD, "deregister_engine_policy"); |
---|
| 109 | LOG("Policy Engine:: Registration message sent.\n"); |
---|
[161] | 110 | } |
---|
| 111 | |
---|
| 112 | |
---|
| 113 | void |
---|
| 114 | PolicyEngine::RegisterServices(int32_t socketFD) |
---|
| 115 | { |
---|
[181] | 116 | LOG("Policy Engine:: Registering services.\n"); |
---|
| 117 | SendMessage(socketFD, "register_service_pe_geo"); |
---|
| 118 | SendMessage(socketFD, "register_service_pe_time"); |
---|
| 119 | SendMessage(socketFD, "register_service_pe_spectrum"); |
---|
| 120 | SendMessage(socketFD, "register_service_pe_spacial"); |
---|
[161] | 121 | } |
---|
| 122 | |
---|
| 123 | |
---|
| 124 | void |
---|
| 125 | PolicyEngine::DeregisterServices(int32_t socketFD) |
---|
| 126 | { |
---|
[181] | 127 | LOG("Policy Engine:: Deregistering services.\n"); |
---|
| 128 | SendMessage(socketFD, "deregister_service_pe_geo"); |
---|
| 129 | SendMessage(socketFD, "deregister_service_pe_time"); |
---|
| 130 | SendMessage(socketFD, "deregister_service_pe_spectrum"); |
---|
| 131 | SendMessage(socketFD, "deregister_service_pe_spacial"); |
---|
[161] | 132 | } |
---|
| 133 | |
---|
| 134 | |
---|
| 135 | void |
---|
| 136 | PolicyEngine::LoadPolicies() |
---|
| 137 | { |
---|
| 138 | LOG("PolicyEngine:: Loading policies.\n"); |
---|
| 139 | } |
---|
| 140 | |
---|
| 141 | |
---|
| 142 | void |
---|
| 143 | PolicyEngine::ReloadPolicies() |
---|
| 144 | { |
---|
[182] | 145 | LOG("PolicyEngine:: Reloading policies.\n"); |
---|
[161] | 146 | } |
---|
| 147 | |
---|
| 148 | |
---|
| 149 | void |
---|
| 150 | PolicyEngine::SendPEDecision(int32_t socketFD, struct Parameter pList[], \ |
---|
| 151 | struct CE_Info *ce_info, int32_t decision_array[]) |
---|
| 152 | { |
---|
| 153 | char var[50]; |
---|
| 154 | |
---|
| 155 | for (size_t i = 0; i < ce_info->numParameters; i++) { |
---|
| 156 | sprintf(var, "%i", decision_array[i]); |
---|
| 157 | SendMessage(socketFD, var); |
---|
| 158 | } |
---|
| 159 | } |
---|
| 160 | |
---|
| 161 | |
---|
| 162 | void |
---|
| 163 | PolicyEngine::ValidateParameters(struct Parameter pList[], \ |
---|
| 164 | struct CE_Info *ce_info, int decision_array[]) |
---|
| 165 | { |
---|
| 166 | LOG("Policy Engine:: Policies Validated.\n"); |
---|
| 167 | for (size_t i = 0; i < ce_info->numParameters; i++) |
---|
| 168 | decision_array[i] = 1; |
---|
| 169 | } |
---|
| 170 | |
---|