100 lines
3.1 KiB
C
100 lines
3.1 KiB
C
#include "common.h"
|
||
|
||
//有些头文件不声明memmem
|
||
void *memmem(const void *haystack, size_t haystacklen, const void *needle, size_t needlelen);
|
||
|
||
void error(const char *error_info)
|
||
{
|
||
fprintf(stderr, "%s\n\n", error_info);
|
||
exit(1);
|
||
}
|
||
|
||
int8_t copy_new_mem(char *src, int src_len, char **dest)
|
||
{
|
||
*dest = (char *)malloc(src_len + 1);
|
||
if (*dest == NULL)
|
||
return 1;
|
||
memcpy(*dest, src, src_len);
|
||
*((*dest) + src_len) = '\0';
|
||
|
||
return 0;
|
||
}
|
||
|
||
/* 字符串替换，replace_memory为可以用free释放的指针 */
|
||
char *replace(char *replace_memory, int *replace_memory_len, const char *src, const int src_len, const char *dest, const int dest_len)
|
||
{
|
||
if (!replace_memory || !src || !dest)
|
||
return replace_memory;
|
||
|
||
char *p;
|
||
int diff;
|
||
|
||
if (src_len == dest_len) {
|
||
for (p = memmem(replace_memory, *replace_memory_len, src, src_len); p; p = memmem(p, *replace_memory_len - (p - replace_memory), src, src_len)) {
|
||
memcpy(p, dest, dest_len);
|
||
p += dest_len;
|
||
}
|
||
} else if (src_len < dest_len) {
|
||
int before_len;
|
||
char *before_end, *new_replace_memory;
|
||
|
||
diff = dest_len - src_len;
|
||
for (p = memmem(replace_memory, *replace_memory_len, src, src_len); p; p = memmem(p, *replace_memory_len - (p - replace_memory), src, src_len)) {
|
||
*replace_memory_len += diff;
|
||
before_len = p - replace_memory;
|
||
new_replace_memory = (char *)realloc(replace_memory, *replace_memory_len + 1);
|
||
if (new_replace_memory == NULL) {
|
||
free(replace_memory);
|
||
return NULL;
|
||
}
|
||
replace_memory = new_replace_memory;
|
||
before_end = replace_memory + before_len;
|
||
p = before_end + dest_len;
|
||
memmove(p, p - diff, *replace_memory_len - (p - replace_memory));
|
||
memcpy(before_end, dest, dest_len);
|
||
}
|
||
} else if (src_len > dest_len) {
|
||
diff = src_len - dest_len;
|
||
for (p = memmem(replace_memory, *replace_memory_len, src, src_len); p; p = memmem(p, *replace_memory_len - (p - replace_memory), src, src_len)) {
|
||
*replace_memory_len -= diff;
|
||
memcpy(p, dest, dest_len);
|
||
p += dest_len;
|
||
memmove(p, p + diff, *replace_memory_len - (p - replace_memory));
|
||
}
|
||
}
|
||
|
||
replace_memory[*replace_memory_len] = '\0';
|
||
return replace_memory;
|
||
}
|
||
|
||
/* 对数据进行编码 */
|
||
void dataEncode(char *data, int data_len, unsigned code)
|
||
{
|
||
while (data_len-- > 0)
|
||
data[data_len] ^= code;
|
||
}
|
||
|
||
/* 监听一个UDP接口 */
|
||
int udp_listen(char *ip, int port)
|
||
{
|
||
struct sockaddr_in addr;
|
||
int fd, opt = 1;
|
||
|
||
if ((fd = socket(AF_INET, SOCK_DGRAM, 0)) < 0) {
|
||
perror("udp socket");
|
||
exit(1);
|
||
}
|
||
setsockopt(fd, SOL_IP, IP_TRANSPARENT, &opt, sizeof(opt));
|
||
setsockopt(fd, SOL_IP, IP_RECVORIGDSTADDR, &opt, sizeof(opt));
|
||
memset(&addr, 0, sizeof(addr));
|
||
addr.sin_addr.s_addr = inet_addr(ip);
|
||
addr.sin_port = htons(port);
|
||
addr.sin_family = AF_INET;
|
||
if (bind(fd, (struct sockaddr *)&addr, sizeof(addr)) != 0) {
|
||
perror("udp bind");
|
||
exit(1);
|
||
}
|
||
|
||
return fd;
|
||
}
|