uint8_t* random_advdata(array_size) { uint8_t* ykAdvData1 = (uint8_t*)malloc(sizeof(uint8_t) * array_size); //动态分配内存 if(ykAdvData1 == NULL) { printf("Memory allocation failed!"); return NULL; } uint8_t ykAdvData1[] = { 0x04, GAP_ADTYPE_LOCAL_NAME_SHORT, 'S', 'M', 'C', 0x09, GAP_ADTYPE_MANUFACTURER_SPECIFIC, //Company Identifier 0x06, 0xff, //Additional Data 0x00, 0x00, 0x00, 0x00, 0x3B, }; //Generate random number seeds srand((unsigned)time(NULL)); //Generate 4 random numbers and store them in ykAdvData for(int i = 0; i < 4; i++) { int random_num = rand(); ykAdvData1[9+i] = random_num & 0xFF; Log_info2("ykAdvData[9+%d] = 0x%x", i, ykAdvData1[9+i]); } return ykAdvData1; } 代码优化
时间: 2024-02-14 21:07:41 浏览: 76
以下是对代码的优化建议:
1. 在函数外部定义广告数据的数组,并将其作为参数传递到函数中进行修改,避免使用动态内存分配。这样可以避免内存泄漏和频繁的内存分配和释放操作。
2. 在生成随机数之前,可以使用宏定义或常量来定义随机数的个数,避免在循环中频繁调用rand()函数。
3. 为了可读性和可维护性,可以将广告数据的格式定义为一个结构体或数组,然后使用循环和结构体成员来填充随机数。
4. 引入错误处理,例如检查malloc()函数的返回值是否为NULL,以及在生成随机数时检查rand()函数是否返回正确的值。
以下是优化后的代码示例:
```
#include <stdlib.h>
#include <time.h>
#define ADV_DATA_SIZE 14
#define RAND_NUM_COUNT 4
typedef struct {
uint8_t length;
uint8_t type;
uint8_t data[ADV_DATA_SIZE - 2];
} AdvData;
void generate_random_nums(uint8_t* nums, int count) {
for (int i = 0; i < count; i++) {
int random_num = rand();
nums[i] = random_num & 0xFF;
Log_info2("Random number[%d] = 0x%x", i, nums[i]);
}
}
void generate_advdata(AdvData* advdata) {
advdata[0].length = 3;
advdata[0].type = GAP_ADTYPE_LOCAL_NAME_SHORT;
advdata[0].data[0] = 'S';
advdata[0].data[1] = 'M';
advdata[0].data[2] = 'C';
advdata[1].length = 7;
advdata[1].type = GAP_ADTYPE_MANUFACTURER_SPECIFIC;
advdata[1].data[0] = 0x06;
advdata[1].data[1] = 0xff;
advdata[1].data[2] = 0x00;
advdata[1].data[3] = 0x00;
generate_random_nums(&advdata[1].data[4], RAND_NUM_COUNT);
}
// Example usage
int main() {
AdvData advdata[2];
generate_advdata(advdata);
// Do something with advdata...
return 0;
}
```
阅读全文