#!/bin/bash
#Program: 要求以 root用户执行此脚本程序。执行完此脚本后，再行去字符界面
#           执行安装即可。
#            字符界面安装命令为： ./NVIDIA*.run -k $(uname -r)
#            其中假定驱动文件（.run文件）是存放在当前目录下，并且文件名为NVIDIA
#            开头。
#Author : Dawnson
#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文件..."
line=`sed -n "/vmlinuz-$(uname -r)/=" /etc/grub.conf`
split -l $line /etc/grub.conf
mv ./xab ./temp_2
split -l `expr $line - 1` ./xaa
echo "rdblacklist=nouveau vga=0x361" > ./temp
paste -d ' '  ./xab ./temp > ./temp_1
cat ./temp_2 >>./temp_1
cat ./temp_1 >>./xaa
rm -f ./xaa ./xab ./temp ./temp_1 ./temp_2
#修改/etc/modprobe.d/blacklist.conf文件
echo "修改/etc/modprobe.d/blacklist.conf文件..."
line_2=`sed -n "/nvidia/=" /etc/modprobe.d/blacklist.conf`
split -l $line_2 /etc/modprobe.d/blacklist.conf
mv ./xab ./temp
split -l `expr $line_2 - 1` ./xaa
echo "blacklist nouveau">>./temp
cat ./temp >>./xaa
mv -f ./xaa /etc/modprobe.d/blacklist.conf
rm -f ./temp ./xaa ./xab
#禁止SeLinux阻止驱动安装
echo "禁止SeLinux阻止驱动安装..."
setsebool -P allow_execstack on
echo "安装前配置完成！现在请重启电脑进入字符界面执行显卡驱动安装。"
exit 0