snmp_api.c 7.3 KB

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