73 lines
2.0 KiB
C
73 lines
2.0 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 GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), 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 GNU Affero 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 <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <netinet/in.h>
|
|
#include <errno.h>
|
|
#include "fastcommon/shared_func.h"
|
|
#include "fastcommon/logger.h"
|
|
#include "fastcommon/sockopt.h"
|
|
#include "fastcommon/http_func.h"
|
|
#include "sf_define.h"
|
|
#include "sf_global.h"
|
|
#include "sf_func.h"
|
|
|
|
int sf_connect_to_server(const char *ip_addr, const int port, int *sock)
|
|
{
|
|
int result;
|
|
*sock = socket(AF_INET, SOCK_STREAM, 0);
|
|
if(*sock < 0) {
|
|
return errno != 0 ? errno : ENOMEM;
|
|
}
|
|
tcpsetserveropt(*sock, g_sf_global_vars.net_buffer_cfg.network_timeout);
|
|
|
|
if ((result=tcpsetnonblockopt(*sock)) != 0) {
|
|
close(*sock);
|
|
*sock = -1;
|
|
return result;
|
|
}
|
|
FC_SET_CLOEXEC(*sock);
|
|
|
|
if ((result=connectserverbyip_nb(*sock, ip_addr, port, g_sf_global_vars.
|
|
net_buffer_cfg.connect_timeout)) != 0)
|
|
{
|
|
close(*sock);
|
|
*sock = -1;
|
|
return result;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
static void sf_memory_oom_notify_callback(const size_t curr_size)
|
|
{
|
|
logCrit("file: "__FILE__", line: %d, "
|
|
"alloc %"PRId64" bytes fail, exiting ...",
|
|
__LINE__, (int64_t)curr_size);
|
|
sf_terminate_myself();
|
|
}
|
|
|
|
void sf_enable_exit_on_oom()
|
|
{
|
|
g_oom_notify = sf_memory_oom_notify_callback;
|
|
}
|