#!/usr/bin/env bash
#系统初始化脚本

function set_tencent_yum(){
#替换默认yum源为腾讯yum源
    mv /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.backup
    wget -O /etc/yum.repos.d/CentOS-Base.repo http://mirrors.cloud.tencent.com/repo/centos7_base.repo
    mv /etc/yum.repos.d/CentOS-Epel.repo /etc/yum.repos.d/CentOS-Epel.repo.backup
    wget -O /etc/yum.repos.d/epel.repo http://mirrors.cloud.tencent.com/repo/epel-7.repo
    yum clean all
    yum makecache
    echo -e "\033[32m [yum源(tencent) 配置] ==> OK \033[0m"
    echo -e "\033[32m [EPEL源(tencent) 配置] ==> OK \033[0m"
}

function set_init() {
#安装必要的软件,创建目录
    yum install -y vim wget lrzsz telnet traceroute iotop tree rsync
    yum install -y git zlib-devel openssl-devel unzip xz libxslt-devel libxml2-devel libcurl-devel
    #yum -y install ipset  ipset-service  >/dev/null 2>&1
    yum -y install net-tools.x86_64 vim lsof
    yum install -y openssl openssl-devel openssh openssh-server
    yum install -y ntpdate crontabs
    echo -e "\033[32m [安装常用工具] ==> OK \033[0m"
}

function set_ulimits(){
#系统限制更改函数
#修改系统限制
cat <<EOF >>/etc/security/limits.conf
* soft nofile 1048576
* hard nofile 1048576
EOF

cat <<EOF >/etc/security/limits.d/20-nproc.conf
*             soft    nproc     1048576
*             soft    nproc     1048576
EOF

echo -e "\033[32m [ulimits 配置] ==> OK \033[0m"
}

# 增加操作系统记录数量
function set_history(){
#更改history命令记录函数
if ! grep "HISTTIMEFORMAT" /etc/profile >/dev/null 2>&1
then echo '
UserIP=$(who -u am i | cut -d"("  -f 2 | sed -e "s/[()]//g")
export HISTTIMEFORMAT="[%F %T] [`whoami`] [${UserIP}] " ' >> /etc/profile;
fi
sed -i "s/HISTSIZE=1000/HISTSIZE=999999999/" /etc/profile

echo -e "\033[32m [history 优化] ==> OK \033[0m"
}

function set_kernel(){
#系统内核优化
cat <<EOF > /etc/sysctl.conf
fs.file-max = 6553560
net.core.netdev_max_backlog = 32768
net.core.rmem_default = 8388608
net.core.rmem_max = 16777216
net.core.somaxconn = 32768
net.core.wmem_default = 8388608
net.core.wmem_max = 16777216
net.ipv4.conf.all.arp_ignore = 0
net.ipv4.conf.lo.arp_announce = 0
net.ipv4.conf.lo.arp_ignore = 0
net.ipv4.ip_local_port_range = 5000 65000
net.ipv4.tcp_fin_timeout = 30
net.ipv4.tcp_keepalive_intvl = 30
net.ipv4.tcp_keepalive_probes = 3
net.ipv4.tcp_keepalive_time = 300
net.ipv4.tcp_max_orphans = 3276800
net.ipv4.tcp_max_syn_backlog = 65536
net.ipv4.tcp_max_tw_buckets = 5000
net.ipv4.tcp_mem = 94500000 915000000 927000000
net.ipv4.tcp_syn_retries = 2
net.ipv4.tcp_synack_retries = 2
net.ipv4.tcp_syncookies = 1
net.ipv4.tcp_timestamps = 0
net.ipv4.tcp_tw_recycle = 1
net.ipv4.tcp_tw_reuse = 1
vm.overcommit_memory = 1
EOF

sysctl -p >/dev/null 2>&1
echo -e "\033[32m [内核 优化] ==> OK \033[0m"
}

# 稍后测试一下看看
function set_security() {
#关闭selinux
    sed -i 's/SELINUX=enforcing/SELINUX=disabled/g' /etc/selinux/config
    sed -i 's/SELINUX=permissive/SELINUX=disabled/g' /etc/selinux/config
    setenforce 0 >/dev/null 2>&1
    #systemctl stop firewalld.service
    #systemctl disable firewalld.service
    yum -y remove firewalld
    yum -y install iptables-services  iptables
    echo -e "\033[32m [安全配置] ==> OK \033[0m"
}

function set_timezone() {
#同步系统时间,设置时区
    rm -f /etc/localtime
    ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime
    ntpdate ntp.ntsc.ac.cn >/dev/null 2>&1

    #计划任务,每5分钟同步系统时间
    echo "*/5 * * * * flock -xn /tmp/stargate.lock -c '/usr/sbin/ntpdate ntp.ntsc.ac.cn  > /tmp/ntp.log 2>&1 &'" >> /var/spool/cron/root

    echo -e "\033[32m [时区设置] ==> OK \033[0m"
}

#声明函数变量
export -f set_tencent_yum
export -f set_init
export -f set_ulimits
export -f set_ssh
export -f set_history
export -f set_kernel
export -f set_security
export -f set_timezone

#执行
set_tencent_yum
set_init
set_ulimits
set_history
set_security
set_timezone

#!/usr/bin/env bash
#检查系统环境脚本

#检查selinux
function check_selinux(){
    status=`sestatus | awk '{print $3}'`
    if [ $status == 'disabled' ];then
        echo -e  "\033[32m [INFO] SELinux ==> OK \033[0m"
    else
        echo -e  "\033[31m [ERROR]: SELinux --> $status \033[0m"
    fi
}

#检查firewall
function check_firewalld(){
    status=`rpm -qa firewalld`
    if [[ $status == '' ]];then
        echo -e  "\033[32m [INFO] firewalld ==> OK \033[0m"
    else
        echo -e  "\033[31m [ERROR]: firewalld --> $status \033[0m"
    fi
}

#检查是否存在rsync
function check_rsync(){
    status=`which rsync`
    if [[ $status == '/usr/bin/rsync' ]];then
        echo -e  "\033[32m [INFO] rsync ==> OK \033[0m"
    else
        echo -e  "\033[31m [ERROR]: rsync --> $status \033[0m"
    fi
}

#检查是否存在全局代理
function check_proxy(){
    status=`echo "$http_proxy" "$https_proxy"`
    if [[ $status == ' ' ]];then
        echo -e  "\033[32m [INFO] proxy ==> OK \033[0m"
    else
        echo -e  "\033[31m [ERROR]: proxy --> $status \033[0m"
    fi
}

#检查dns文件权限
function check_resolv(){
    status=`lsattr /etc/resolv.conf | awk '{print $1}'`
    if [[ $status == '-------------e--' ]];then
        echo -e  "\033[32m [INFO] resolv ==> OK \033[0m"
    else
        echo -e  "\033[31m [ERROR]: resolv --> $status \033[0m"
    fi
}

export -f check_selinux
export -f check_firewalld
export -f check_rsync
export -f check_proxy
export -f check_resolv
check_selinux
check_firewalld
check_rsync
check_proxy
check_resolv