From 5022a8615668c86219644791a780792449dcc729 Mon Sep 17 00:00:00 2001
From: erpalma <palmarini@unive.it>
Date: Sat, 7 Apr 2018 19:09:06 +0200
Subject: [PATCH] add automatic msr module loading

---
 lenovo_fix.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/lenovo_fix.py b/lenovo_fix.py
index 19ce6b7..cc93ce2 100755
--- a/lenovo_fix.py
+++ b/lenovo_fix.py
@@ -4,6 +4,7 @@ import ConfigParser
 import glob
 import os
 import struct
+import subprocess
 
 from collections import defaultdict
 from periphery import MMIO
@@ -21,7 +22,10 @@ def writemsr(msr, val):
         os.write(f, struct.pack('Q', val))
         os.close(f)
     if not n:
-        raise OSError("msr module not loaded (run modprobe msr)")
+        try:
+            subprocess.check_call(('modprobe', 'msr'))
+        except subprocess.CalledProcessError:
+            raise OSError("Unable to load msr module.")
 
 
 def is_on_battery():