snmp_api.c 6.8 KB

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