snmp_api.c 6.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264
  1. /********************************* (C) ROTEK ***********************************
  2. * @module template
  3. * @file template.c
  4. * @version 1.0.0
  5. * @date XX.XX.XXXX
  6. * $brief Template
  7. *******************************************************************************
  8. * @history Version Author Comment
  9. * XX.XX.XXXX 1.0.0 Telenkov D.A. First release.
  10. *******************************************************************************
  11. */
  12. #include "stm32f4xx.h"
  13. #include "snmp_api.h"
  14. #include "trap_api.h"
  15. #include "settings_api.h"
  16. #include "common_config.h"
  17. #include "rtc.h"
  18. #ifdef PRINTF_STDLIB
  19. #include <stdio.h>
  20. #endif
  21. #ifdef PRINTF_CUSTOM
  22. #include "tinystdio.h"
  23. #endif
  24. #include <stdbool.h>
  25. #include <string.h>
  26. #include "lwip/opt.h"
  27. #include "lwip/api.h"
  28. #include "lwip/sys.h"
  29. #include "lwip/udp.h"
  30. #include "snmp.h"
  31. #include "snmp_msg.h"
  32. #include "private_mib.h"
  33. #include "FreeRTOS.h"
  34. #include "task.h"
  35. #include "queue.h"
  36. /**
  37. * @brief Пул всех возможных трапов устройства
  38. */
  39. extern TRAP_t traps[];
  40. /**
  41. * @brief Общая структура настроек
  42. */
  43. extern SETTINGS_t sSettings;
  44. /**
  45. * @brief Очередь для отправки трапов
  46. */
  47. QueueHandle_t SNMP_TrapQueue;
  48. /**
  49. * @brief Инициализация параметров SNMP
  50. * @retval
  51. */
  52. void SNMP_Init(void)
  53. {
  54. //SETTINGS_SetSnmpDef();
  55. SNMP_AgentInit();
  56. }
  57. /**
  58. * @brief Системный тик SNMP. Таск должен вызываться с частотой 100Гц.
  59. * @retval
  60. */
  61. void SNMP_SysUpTimeTask(void *arg)
  62. {
  63. TickType_t xLastWakeTime;
  64. const TickType_t xFrequency = 10;
  65. xLastWakeTime = xTaskGetTickCount();
  66. while(1)
  67. {
  68. vTaskDelayUntil( &xLastWakeTime, xFrequency );
  69. snmp_inc_sysuptime();
  70. }
  71. }
  72. /**
  73. * @brief Тестовый таск для проверки отправки трапов
  74. * @retval
  75. */
  76. void snmp_trap_tread(void *arg)
  77. {
  78. uint8_t trapName;
  79. while(1)
  80. {
  81. if (xQueueReceive(SNMP_TrapQueue, &trapName, 0) == pdTRUE)
  82. {
  83. SNMP_SetManagerIP(sSettings.sSnmp.managerIP);
  84. SNMP_SendVarbindTrap(&traps[trapName]);
  85. SNMP_SetManagerIP(sSettings.sSnmp.managerIP2);
  86. SNMP_SendVarbindTrap(&traps[trapName]);
  87. SNMP_SetManagerIP(sSettings.sSnmp.managerIP3);
  88. SNMP_SendVarbindTrap(&traps[trapName]);
  89. }
  90. }
  91. }
  92. /**
  93. * @brief Инициализация SNMP агента
  94. * @retval
  95. */
  96. void SNMP_AgentInit(void)
  97. {
  98. SNMP_SetObjDescr();
  99. SNMP_SetReadCommunity(sSettings.sSnmp.readCommunity);
  100. SNMP_SetWriteCommunity(sSettings.sSnmp.writeCommunity);
  101. SNMP_SetSysContact(sSettings.sSnmp.sysContact);
  102. SNMP_SetSysName(sSettings.sSnmp.sysName);
  103. SNMP_SetSysLocation(sSettings.sSnmp.sysLocation);
  104. SNMP_SetManagerIP(sSettings.sSnmp.managerIP);
  105. SNMP_SetObjID();
  106. SNMP_SetTrapOnOff(1);
  107. SNMP_InitTrapsBase();
  108. snmp_init();
  109. udp_init();
  110. SNMP_TrapQueue = xQueueCreate(SNMP_TRAP_QUEUE_SIZE, sizeof(uint8_t));
  111. }
  112. /**
  113. * @brief Пользовательская функция для отправки трапа из массива traps[]
  114. * Трап помещается в очередь. Работа с очередью происходит по принципу
  115. * FIFO буфера.
  116. * Если в настройках трапа отправка отключена, то трап игнорируется.
  117. * @retval
  118. */
  119. void SNMP_SendUserTrap(uint8_t trapName)
  120. {
  121. uint16_t availableSpace;
  122. uint8_t dummyTrap;
  123. if (traps[trapName].trapEnable)
  124. {
  125. availableSpace = uxQueueSpacesAvailable(SNMP_TrapQueue);
  126. if (availableSpace == 0)
  127. xQueueReceive(SNMP_TrapQueue, &dummyTrap, 0);
  128. xQueueSend(SNMP_TrapQueue, &trapName, 0);
  129. }
  130. }
  131. /**
  132. * @brief Установить SNMP Descriptor
  133. * @retval
  134. */
  135. // TODO
  136. void SNMP_SetObjDescr(void)
  137. {
  138. static uint8_t len;
  139. strcpy(sSettings.sSnmp.sysDescr, sSettings.sSnmp.sysName);
  140. strcat(sSettings.sSnmp.sysDescr, " ");
  141. strcat(sSettings.sSnmp.sysDescr, VERSION);
  142. strcat(sSettings.sSnmp.sysDescr, " ");
  143. strcat(sSettings.sSnmp.sysDescr, sSettings.sSnmp.sysContact);
  144. strcat(sSettings.sSnmp.sysDescr, " ");
  145. strcat(sSettings.sSnmp.sysDescr, sSettings.sInfo.serialNumber);
  146. len = strlen(sSettings.sSnmp.sysDescr);
  147. snmp_set_sysdesr((u8_t*)sSettings.sSnmp.sysDescr, &len);
  148. }
  149. /**
  150. * @brief Установить SNMP Community для чтения
  151. */
  152. void SNMP_SetReadCommunity(char *comm)
  153. {
  154. strcpy(sSettings.sSnmp.readCommunity, comm);
  155. }
  156. /**
  157. * @brief Установить SNMP Community для записи
  158. */
  159. void SNMP_SetWriteCommunity(char *comm)
  160. {
  161. strcpy(sSettings.sSnmp.writeCommunity, comm);
  162. }
  163. /**
  164. * @brief Установить SNMP SysContact
  165. * @retval
  166. */
  167. void SNMP_SetSysContact(char *con)
  168. {
  169. static uint8_t len;
  170. len = strlen(con);
  171. snmp_set_syscontact((u8_t*)con, &len);
  172. }
  173. /**
  174. * @brief Установить SNMP SysName
  175. * @retval
  176. */
  177. void SNMP_SetSysName(char *name)
  178. {
  179. static uint8_t len;
  180. len = strlen(name);
  181. snmp_set_sysname((u8_t*)name, &len);
  182. }
  183. /**
  184. * @brief Установить SNMP SysLocation
  185. * @retval
  186. */
  187. void SNMP_SetSysLocation(char *loc)
  188. {
  189. static uint8_t len;
  190. len = strlen(loc);
  191. snmp_set_syslocation((u8_t*)loc, &len);
  192. }
  193. /**
  194. * @brief Установить SNMP SysManagerIP
  195. * @retval
  196. */
  197. void SNMP_SetManagerIP(char *ip)
  198. {
  199. static ip_addr_t trap_addr;
  200. ipaddr_aton(ip, &trap_addr);
  201. snmp_trap_dst_ip_set(0, &trap_addr);
  202. }
  203. /**
  204. * @brief Установить SNMP Object ID
  205. * @retval
  206. */
  207. void SNMP_SetObjID(void)
  208. {
  209. static struct snmp_obj_id my_object_id = {9, {1, 3, 6, 1, 4, 1, 41752, 911, 3}};
  210. snmp_set_sysobjid(&my_object_id);
  211. }
  212. /**
  213. * @brief Вкл/выкл трапы
  214. * @retval
  215. */
  216. void SNMP_SetTrapOnOff(uint8_t state)
  217. {
  218. snmp_trap_dst_enable(0, (u8_t)state);
  219. }
  220. /********************************* (C) ROTEK **********************************/