290 lines
7.1 KiB
C
290 lines
7.1 KiB
C
/*
|
||
* Copyright (c) 2020 YuQing <384681@qq.com>
|
||
*
|
||
* This program is free software: you can use, redistribute, and/or modify
|
||
* it under the terms of the Lesser GNU General Public License, version 3
|
||
* or later ("LGPL"), as published by the Free Software Foundation.
|
||
*
|
||
* This program is distributed in the hope that it will be useful, but WITHOUT
|
||
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
||
* FITNESS FOR A PARTICULAR PURPOSE.
|
||
*
|
||
* You should have received a copy of the Lesser GNU General Public License
|
||
* along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
*/
|
||
|
||
#include <time.h>
|
||
#include <stdio.h>
|
||
#include <stdlib.h>
|
||
#include <string.h>
|
||
#include <ctype.h>
|
||
#include <unistd.h>
|
||
#include <fcntl.h>
|
||
#include <sys/types.h>
|
||
#include <sys/stat.h>
|
||
#include <netinet/in.h>
|
||
#include "logger.h"
|
||
#include "shared_func.h"
|
||
#include "local_ip_func.h"
|
||
#include "id_generator.h"
|
||
|
||
int id_generator_init_extra_ex(struct idg_context *context, const char *filename,
|
||
const int machine_id, const int mid_bits, const int extra_bits,
|
||
const int sn_bits, const mode_t mode)
|
||
{
|
||
int result;
|
||
int mid;
|
||
if (mid_bits < 2 || mid_bits > 20)
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"invalid bits of machine id: %d",
|
||
__LINE__, mid_bits);
|
||
context->fd = -1;
|
||
return EINVAL;
|
||
}
|
||
if (extra_bits < 0 || extra_bits > 16)
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"invalid bits of extra data: %d",
|
||
__LINE__, extra_bits);
|
||
context->fd = -1;
|
||
return EINVAL;
|
||
}
|
||
if (sn_bits < 8)
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"invalid bits of serial no: %d < 8",
|
||
__LINE__, sn_bits);
|
||
context->fd = -1;
|
||
return EINVAL;
|
||
}
|
||
if (mid_bits + extra_bits + sn_bits > 32)
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"invalid mid_bits + extra_bits + sn_bits: %d > 32",
|
||
__LINE__, mid_bits + extra_bits + sn_bits);
|
||
context->fd = -1;
|
||
return EINVAL;
|
||
}
|
||
|
||
if (machine_id < 0 || machine_id >= (1 << mid_bits))
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"invalid machine id: %d",
|
||
__LINE__, machine_id);
|
||
context->fd = -1;
|
||
return EINVAL;
|
||
}
|
||
if (machine_id != 0)
|
||
{
|
||
mid = machine_id;
|
||
}
|
||
else
|
||
{
|
||
const char *local_ip;
|
||
const char *private_ip;
|
||
struct in_addr ip_addr;
|
||
struct in6_addr ip6_addr;
|
||
|
||
private_ip = get_first_local_private_ip();
|
||
if (private_ip != NULL)
|
||
{
|
||
local_ip = private_ip;
|
||
}
|
||
else
|
||
{
|
||
local_ip = get_first_local_ip();
|
||
if (local_ip == NULL)
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"can't get local ip address", __LINE__);
|
||
context->fd = -1;
|
||
return ENOENT;
|
||
}
|
||
else if (strcmp(local_ip, LOCAL_LOOPBACK_IPv4) == 0)
|
||
{
|
||
/* 注意，当系统存在IPv6回环地址时，为了简化系统的改动，
|
||
会将IPv6回环地址修改成IPv4回环地址返回
|
||
此处错误打印时，需要带上IPv6回环地址
|
||
*/
|
||
logWarning("file: "__FILE__", line: %d, "
|
||
"can't get local ip address, set to %s or %s",
|
||
__LINE__, LOCAL_LOOPBACK_IPv4, LOCAL_LOOPBACK_IPv6);
|
||
}
|
||
}
|
||
|
||
if (inet_pton(AF_INET, local_ip, &ip_addr) == 1)
|
||
{
|
||
//do nothing
|
||
}
|
||
else if(inet_pton(AF_INET6, local_ip, &ip6_addr) == 1)
|
||
{
|
||
ip_addr.s_addr = *((in_addr_64_t *)&ip6_addr);
|
||
}
|
||
else
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"invalid local ip address: %s",
|
||
__LINE__, local_ip);
|
||
context->fd = -1;
|
||
return EINVAL;
|
||
}
|
||
|
||
logDebug("ip_addr: %s, s_addr: 0x%08X",
|
||
local_ip, ip_addr.s_addr);
|
||
|
||
mid = ntohl(ip_addr.s_addr) & ((1 << mid_bits) - 1);
|
||
}
|
||
|
||
if ((context->fd = open(filename, O_RDWR | O_CLOEXEC)) < 0)
|
||
{
|
||
if (errno == ENOENT)
|
||
{
|
||
mode_t old_mode;
|
||
old_mode = umask(0);
|
||
if ((context->fd=open(filename, O_RDWR | O_CREAT |
|
||
O_CLOEXEC, mode)) < 0)
|
||
{
|
||
result = errno != 0 ? errno : EACCES;
|
||
}
|
||
else
|
||
{
|
||
result = 0;
|
||
}
|
||
umask(old_mode);
|
||
}
|
||
else
|
||
{
|
||
result = errno != 0 ? errno : EACCES;
|
||
}
|
||
|
||
if (result != 0)
|
||
{
|
||
logError("file: "__FILE__", line: %d, "
|
||
"open file \"%s\" fail, "
|
||
"errno: %d, error info: %s", __LINE__,
|
||
filename, result, STRERROR(result));
|
||
return result;
|
||
}
|
||
}
|
||
|
||
context->machine_id = mid;
|
||
context->mid_bits = mid_bits;
|
||
context->extra_bits = extra_bits;
|
||
context->sn_bits = sn_bits;
|
||
context->mes_bits_sum = mid_bits + extra_bits + sn_bits;
|
||
context->masked_mid = ((int64_t)mid) << (extra_bits + sn_bits);
|
||
context->extra_mask = ((int64_t)1 << (extra_bits + sn_bits)) -
|
||
((int64_t)1 << sn_bits);
|
||
context->sn_mask = ((int64_t)1 << sn_bits) - 1;
|
||
|
||
logDebug("mid: 0x%08X, masked_mid: 0x%08"PRIX64", extra_mask: 0x%08"PRIX64", "
|
||
"sn_mask: 0x%08"PRIX64, mid, context->masked_mid,
|
||
context->extra_mask, context->sn_mask);
|
||
|
||
return 0;
|
||
}
|
||
|
||
int id_generator_init_extra(struct idg_context *context, const char *filename,
|
||
const int machine_id, const int mid_bits, const int extra_bits,
|
||
const int sn_bits)
|
||
{
|
||
return id_generator_init_extra_ex(context, filename,
|
||
machine_id, mid_bits, extra_bits, sn_bits,
|
||
ID_GENERATOR_DEFAULT_FILE_MODE);
|
||
}
|
||
|
||
void id_generator_destroy(struct idg_context *context)
|
||
{
|
||
if (context->fd >= 0)
|
||
{
|
||
close(context->fd);
|
||
context->fd = -1;
|
||
}
|
||
}
|
||
|
||
int id_generator_next_extra_ptr(struct idg_context *context,
|
||
const int *extra, int64_t *id)
|
||
{
|
||
int result;
|
||
int len;
|
||
int bytes;
|
||
int new_extra;
|
||
int64_t sn;
|
||
char buff[32];
|
||
char *endptr;
|
||
|
||
if ((result=file_write_lock(context->fd)) != 0)
|
||
{
|
||
*id = 0;
|
||
return result;
|
||
}
|
||
|
||
do
|
||
{
|
||
if (lseek(context->fd, 0L, SEEK_SET) == -1)
|
||
{
|
||
result = errno != 0 ? errno : EACCES;
|
||
logError("file: "__FILE__", line: %d, "
|
||
"file lseek fail, "
|
||
"errno: %d, error info: %s", __LINE__,
|
||
result, STRERROR(result));
|
||
sn = 0;
|
||
break;
|
||
}
|
||
|
||
if ((bytes=read(context->fd, buff, sizeof(buff) - 1)) < 0)
|
||
{
|
||
result = errno != 0 ? errno : EACCES;
|
||
logError("file: "__FILE__", line: %d, "
|
||
"file read fail, "
|
||
"errno: %d, error info: %s", __LINE__,
|
||
result, STRERROR(result));
|
||
sn = 0;
|
||
break;
|
||
}
|
||
*(buff + bytes) = '\0';
|
||
|
||
sn = strtoll(buff, &endptr, 10);
|
||
++sn;
|
||
|
||
if (lseek(context->fd, 0L, SEEK_SET) == -1)
|
||
{
|
||
result = errno != 0 ? errno : EACCES;
|
||
logError("file: "__FILE__", line: %d, "
|
||
"cal lseek fail, "
|
||
"errno: %d, error info: %s", __LINE__,
|
||
result, STRERROR(result));
|
||
break;
|
||
}
|
||
|
||
len = sprintf(buff, "%-20"PRId64, sn);
|
||
if ((bytes=write(context->fd, buff, len)) != len)
|
||
{
|
||
result = errno != 0 ? errno : EACCES;
|
||
logError("file: "__FILE__", line: %d, "
|
||
"file write %d bytes fail, written: %d bytes, "
|
||
"errno: %d, error info: %s", __LINE__,
|
||
len, bytes, result, STRERROR(result));
|
||
break;
|
||
}
|
||
} while (0);
|
||
|
||
file_unlock(context->fd);
|
||
|
||
if (extra == NULL)
|
||
{
|
||
new_extra = sn % (1 << context->extra_bits);
|
||
}
|
||
else
|
||
{
|
||
new_extra = *extra;
|
||
}
|
||
|
||
*id = (((int64_t)time(NULL)) << context->mes_bits_sum) |
|
||
context->masked_mid |
|
||
((new_extra << context->sn_bits) & context->extra_mask) |
|
||
(sn & context->sn_mask);
|
||
return result;
|
||
}
|