361 lines
8.7 KiB
C
361 lines
8.7 KiB
C
/* USER CODE BEGIN Header */
|
|
/**
|
|
******************************************************************************
|
|
* @file : main.c
|
|
* @brief : Main program body
|
|
******************************************************************************
|
|
* @attention
|
|
*
|
|
* Copyright (c) 2026 STMicroelectronics.
|
|
* All rights reserved.
|
|
*
|
|
* This software is licensed under terms that can be found in the LICENSE file
|
|
* in the root directory of this software component.
|
|
* If no LICENSE file was not provided, it is provided AS-IS.
|
|
*
|
|
******************************************************************************
|
|
*/
|
|
/* USER CODE END Header */
|
|
/* Includes ------------------------------------------------------------------*/
|
|
#include "main.h"
|
|
#include "usart.h"
|
|
#include "usb_device.h"
|
|
#include "gpio.h"
|
|
|
|
/* Private includes ----------------------------------------------------------*/
|
|
/* USER CODE BEGIN Includes */
|
|
#include "usbd_cdc_if.h"
|
|
#include "stdio.h"
|
|
#include "string.h"
|
|
/* USER CODE END Includes */
|
|
|
|
/* Private typedef -----------------------------------------------------------*/
|
|
/* USER CODE BEGIN PTD */
|
|
|
|
/* USER CODE END PTD */
|
|
|
|
/* Private define ------------------------------------------------------------*/
|
|
/* USER CODE BEGIN PD */
|
|
|
|
/* USER CODE END PD */
|
|
|
|
/* Private macro -------------------------------------------------------------*/
|
|
/* USER CODE BEGIN PM */
|
|
|
|
/* USER CODE END PM */
|
|
|
|
/* Private variables ---------------------------------------------------------*/
|
|
|
|
/* USER CODE BEGIN PV */
|
|
uint8_t uart_rx_byte;
|
|
uint8_t uart_rx_buf[16];
|
|
uint8_t uart_rx_idx = 0;
|
|
|
|
uint8_t usb_rx_buf[32];
|
|
uint8_t usb_rx_idx = 0;
|
|
|
|
volatile uint8_t ok_received = 0;
|
|
volatile uint8_t blive_received = 0;
|
|
volatile uint8_t start_received = 0;
|
|
volatile uint8_t use_usb = 0;
|
|
/* USER CODE END PV */
|
|
|
|
/* Private function prototypes -----------------------------------------------*/
|
|
void SystemClock_Config(void);
|
|
/* USER CODE BEGIN PFP */
|
|
|
|
/* USER CODE END PFP */
|
|
|
|
/* Private user code ---------------------------------------------------------*/
|
|
/* USER CODE BEGIN 0 */
|
|
|
|
static void send_str(const char *str)
|
|
{
|
|
if (use_usb)
|
|
{
|
|
uint16_t len = (uint16_t)strlen(str);
|
|
uint8_t retry = 0;
|
|
while (retry < 10)
|
|
{
|
|
if (CDC_Transmit_FS((uint8_t *)str, len) == USBD_OK)
|
|
return;
|
|
HAL_Delay(1);
|
|
retry++;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
printf("%s", str);
|
|
}
|
|
}
|
|
|
|
static void usb_poll_rx(void)
|
|
{
|
|
if (usb_app_rx_len == 0)
|
|
return;
|
|
|
|
__disable_irq();
|
|
uint16_t len = usb_app_rx_len;
|
|
uint8_t buf[64];
|
|
memcpy(buf, (const void *)usb_app_rx_buf, len);
|
|
usb_app_rx_len = 0;
|
|
__enable_irq();
|
|
|
|
for (uint16_t i = 0; i < len; i++)
|
|
{
|
|
uint8_t byte = buf[i];
|
|
if (byte == '\r' || byte == '\n')
|
|
{
|
|
usb_rx_buf[usb_rx_idx] = '\0';
|
|
if (strcmp((char *)usb_rx_buf, "ok") == 0)
|
|
{
|
|
ok_received = 1;
|
|
}
|
|
else if (strcmp((char *)usb_rx_buf, "blive") == 0)
|
|
{
|
|
blive_received = 1;
|
|
}
|
|
else if (strcmp((char *)usb_rx_buf, "start") == 0)
|
|
{
|
|
if (!start_received)
|
|
{
|
|
start_received = 1;
|
|
use_usb = 1;
|
|
}
|
|
}
|
|
usb_rx_idx = 0;
|
|
}
|
|
else
|
|
{
|
|
if (usb_rx_idx < sizeof(usb_rx_buf) - 1)
|
|
{
|
|
usb_rx_buf[usb_rx_idx++] = byte;
|
|
}
|
|
else
|
|
{
|
|
usb_rx_idx = 0;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/* USER CODE END 0 */
|
|
|
|
/**
|
|
* @brief The application entry point.
|
|
* @retval int
|
|
*/
|
|
int main(void)
|
|
{
|
|
|
|
/* USER CODE BEGIN 1 */
|
|
|
|
/* USER CODE END 1 */
|
|
|
|
/* MCU Configuration--------------------------------------------------------*/
|
|
|
|
/* Reset of all peripherals, Initializes the Flash interface and the Systick. */
|
|
HAL_Init();
|
|
|
|
/* USER CODE BEGIN Init */
|
|
|
|
/* USER CODE END Init */
|
|
|
|
/* Configure the system clock */
|
|
SystemClock_Config();
|
|
|
|
/* USER CODE BEGIN SysInit */
|
|
|
|
/* USER CODE END SysInit */
|
|
|
|
/* Initialize all configured peripherals */
|
|
MX_GPIO_Init();
|
|
MX_USB_DEVICE_Init();
|
|
MX_USART1_UART_Init();
|
|
/* USER CODE BEGIN 2 */
|
|
HAL_UART_Receive_IT(&huart1, &uart_rx_byte, 1);
|
|
|
|
/* Wait for "start" via either UART or USB CDC */
|
|
while (!start_received)
|
|
{
|
|
usb_poll_rx();
|
|
}
|
|
/* USER CODE END 2 */
|
|
|
|
/* Infinite loop */
|
|
/* USER CODE BEGIN WHILE */
|
|
uint8_t lost_count = 0;
|
|
while (1)
|
|
{
|
|
uint8_t state = HAL_GPIO_ReadPin(Main12V_In_GPIO_Port, Main12V_In_Pin);
|
|
if (state) {
|
|
lost_count++;
|
|
if (lost_count >= 10) {
|
|
break;
|
|
}
|
|
} else {
|
|
lost_count = 0;
|
|
blive_received = 0;
|
|
send_str("Alive\r\n");
|
|
HAL_GPIO_TogglePin(RUN_GPIO_Port,RUN_Pin);
|
|
uint32_t tick_start = HAL_GetTick();
|
|
while (!blive_received && (HAL_GetTick() - tick_start < 5000)) {
|
|
usb_poll_rx();
|
|
}
|
|
if (!blive_received) {
|
|
break;
|
|
}
|
|
}
|
|
HAL_Delay(10);
|
|
/* USER CODE END WHILE */
|
|
|
|
/* USER CODE BEGIN 3 */
|
|
}
|
|
HAL_GPIO_WritePin(RUN_GPIO_Port,RUN_Pin,GPIO_PIN_SET);
|
|
HAL_GPIO_WritePin(SHUT_GPIO_Port,SHUT_Pin,GPIO_PIN_RESET);
|
|
send_str("SHUTDOWN\r\n");
|
|
uint32_t tick_start = HAL_GetTick();
|
|
|
|
while (!ok_received && (HAL_GetTick() - tick_start < 60000))
|
|
{
|
|
usb_poll_rx();
|
|
}
|
|
HAL_Delay(10000);
|
|
HAL_GPIO_WritePin(Relay_out_GPIO_Port,Relay_out_Pin,GPIO_PIN_SET);
|
|
/* USER CODE END 3 */
|
|
}
|
|
|
|
/**
|
|
* @brief System Clock Configuration
|
|
* @retval None
|
|
*/
|
|
void SystemClock_Config(void)
|
|
{
|
|
RCC_OscInitTypeDef RCC_OscInitStruct = {0};
|
|
RCC_ClkInitTypeDef RCC_ClkInitStruct = {0};
|
|
RCC_PeriphCLKInitTypeDef PeriphClkInit = {0};
|
|
|
|
/** Initializes the RCC Oscillators according to the specified parameters
|
|
* in the RCC_OscInitTypeDef structure.
|
|
*/
|
|
RCC_OscInitStruct.OscillatorType = RCC_OSCILLATORTYPE_HSE;
|
|
RCC_OscInitStruct.HSEState = RCC_HSE_ON;
|
|
RCC_OscInitStruct.HSEPredivValue = RCC_HSE_PREDIV_DIV1;
|
|
RCC_OscInitStruct.HSIState = RCC_HSI_ON;
|
|
RCC_OscInitStruct.PLL.PLLState = RCC_PLL_ON;
|
|
RCC_OscInitStruct.PLL.PLLSource = RCC_PLLSOURCE_HSE;
|
|
RCC_OscInitStruct.PLL.PLLMUL = RCC_PLL_MUL9;
|
|
if (HAL_RCC_OscConfig(&RCC_OscInitStruct) != HAL_OK)
|
|
{
|
|
Error_Handler();
|
|
}
|
|
|
|
/** Initializes the CPU, AHB and APB buses clocks
|
|
*/
|
|
RCC_ClkInitStruct.ClockType = RCC_CLOCKTYPE_HCLK|RCC_CLOCKTYPE_SYSCLK
|
|
|RCC_CLOCKTYPE_PCLK1|RCC_CLOCKTYPE_PCLK2;
|
|
RCC_ClkInitStruct.SYSCLKSource = RCC_SYSCLKSOURCE_PLLCLK;
|
|
RCC_ClkInitStruct.AHBCLKDivider = RCC_SYSCLK_DIV1;
|
|
RCC_ClkInitStruct.APB1CLKDivider = RCC_HCLK_DIV2;
|
|
RCC_ClkInitStruct.APB2CLKDivider = RCC_HCLK_DIV1;
|
|
|
|
if (HAL_RCC_ClockConfig(&RCC_ClkInitStruct, FLASH_LATENCY_2) != HAL_OK)
|
|
{
|
|
Error_Handler();
|
|
}
|
|
PeriphClkInit.PeriphClockSelection = RCC_PERIPHCLK_USB;
|
|
PeriphClkInit.UsbClockSelection = RCC_USBCLKSOURCE_PLL_DIV1_5;
|
|
if (HAL_RCCEx_PeriphCLKConfig(&PeriphClkInit) != HAL_OK)
|
|
{
|
|
Error_Handler();
|
|
}
|
|
}
|
|
|
|
/* USER CODE BEGIN 4 */
|
|
void HAL_UART_RxCpltCallback(UART_HandleTypeDef *huart)
|
|
{
|
|
if (huart->Instance == USART1)
|
|
{
|
|
if (uart_rx_byte == '\r' || uart_rx_byte == '\n')
|
|
{
|
|
uart_rx_buf[uart_rx_idx] = '\0';
|
|
if (strcmp((char *)uart_rx_buf, "ok") == 0)
|
|
{
|
|
ok_received = 1;
|
|
}
|
|
else if (strcmp((char *)uart_rx_buf, "blive") == 0)
|
|
{
|
|
blive_received = 1;
|
|
}
|
|
else if (strcmp((char *)uart_rx_buf, "start") == 0)
|
|
{
|
|
if (!start_received)
|
|
{
|
|
start_received = 1;
|
|
use_usb = 0;
|
|
}
|
|
}
|
|
else {
|
|
blive_received = 1;
|
|
}
|
|
uart_rx_idx = 0;
|
|
}
|
|
else
|
|
{
|
|
if (uart_rx_idx < sizeof(uart_rx_buf) - 1)
|
|
{
|
|
uart_rx_buf[uart_rx_idx] = uart_rx_byte;
|
|
uart_rx_idx++;
|
|
}
|
|
else
|
|
{
|
|
uart_rx_idx = 0;
|
|
}
|
|
}
|
|
HAL_UART_Receive_IT(&huart1, &uart_rx_byte, 1);
|
|
}
|
|
}
|
|
|
|
void HAL_UART_ErrorCallback(UART_HandleTypeDef *huart)
|
|
{
|
|
if (huart->Instance == USART1)
|
|
{
|
|
__HAL_UART_CLEAR_OREFLAG(huart);
|
|
huart->RxState = HAL_UART_STATE_READY;
|
|
uart_rx_idx = 0;
|
|
HAL_UART_Receive_IT(&huart1, &uart_rx_byte, 1);
|
|
}
|
|
}
|
|
/* USER CODE END 4 */
|
|
|
|
/**
|
|
* @brief This function is executed in case of error occurrence.
|
|
* @retval None
|
|
*/
|
|
void Error_Handler(void)
|
|
{
|
|
/* USER CODE BEGIN Error_Handler_Debug */
|
|
/* User can add his own implementation to report the HAL error return state */
|
|
__disable_irq();
|
|
while (1)
|
|
{
|
|
}
|
|
/* USER CODE END Error_Handler_Debug */
|
|
}
|
|
#ifdef USE_FULL_ASSERT
|
|
/**
|
|
* @brief Reports the name of the source file and the source line number
|
|
* where the assert_param error has occurred.
|
|
* @param file: pointer to the source file name
|
|
* @param line: assert_param error line number
|
|
* @retval None
|
|
*/
|
|
void assert_failed(uint8_t *file, uint32_t line)
|
|
{
|
|
/* USER CODE BEGIN 6 */
|
|
/* User can add his own implementation to report the file name and the number,
|
|
ex: printf("Wrong parameters value: file %s on line %d\r\n", file, line) */
|
|
/* USER CODE END 6 */
|
|
}
|
|
#endif /* USE_FULL_ASSERT */
|