#!/bin/bash
#Program:  要求以 root用户执行此脚本程序。执行完此脚本后，再行去字符界面
#         执行安装即可。
#            字符界面安装命令为： ./NVIDIA*.run -k $(uname -r)
#            其中假定驱动文件（.run文件）是存放在当前目录下，并且文件名为NVIDIA
#            开头。
#Author : Kailash
#Contact: http://www.ninthday.net
PATH=/bin:/sbin:/usr/bin:/usr/sbin:/usr/local/bin:/usr/local/sbin:~/bin
export PATH
# 检测是否安装了 kernel-devel与kernel-headers.
echo "正在检测环境,此过程需要联网并可能花费几分钟..."
dev_stat=`rpm -qa | grep "^kernel-devel-$(uname -r)"`
if [ -z "$dev_stat" ] ; then
    yum install kernel-devel -y
    temp=`echo $?`
    if [ "$temp" -gt "0" ] ; then
        echo "软件kernel-devel没有安装成功，可能是网络问题."
        exit 0
    fi
fi
head_stat=`rpm -qa | grep "^kernel-headers-$(uname -r)"`
if [ -z "$head_stat" ] ; then
    yum install kernel-headers -y
    temp=`echo $?`
    if [ "$temp" -gt "0" ] ; then
        echo "软件kernel-headers没有安装成功，可能是网络问题."
        exit 0
    fi
fi
#配置文件
clear
echo "现在配置需要的修改的文件..."
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img
dracut /boot/initramfs-$(uname -r).img $(uname -r)
#修改grub.conf文件
echo "修改/boot/grub/grub.conf文件..."
sed "s/quiet/& rdblacklist=nouveau\ vga=0x361/"g /etc/grub.conf > /etc/grub.conf
#修改/etc/modprobe.d/blacklist.conf文件
echo "修改/etc/modprobe.d/blacklist.conf文件..."
sed '/nvidia/'d /etc/modprobe.d/blacklist.conf > /etc/modprobe.d/blacklist.conf
#禁止SeLinux阻止驱动安装
echo "禁止SeLinux阻止驱动安装..."
setsebool -P allow_execstack on
echo "安装前配置完成！现在请重启电脑进入字符界面执行显卡驱动安装。"
exit 0