From c1e37a6be9fbf437bb09fc31fdb96d0c94b5a631 Mon Sep 17 00:00:00 2001
From: Samuel Angebault <angebault.samuel@gmail.com>
Date: Tue, 15 May 2018 05:27:10 -0700
Subject: [PATCH] [Arista]: Add a reboot script for all remaining platforms.
 (#1706)

Adds a platform_reboot script for 7050QX-32, 7050QX-32S and 7060CX-32S.
This allow a proper cold reboot to happen.
---
 .../x86_64-arista_7050_qx32/platform_reboot   | 87 +++++++++++++++++++
 .../x86_64-arista_7050_qx32s/platform_reboot  | 87 +++++++++++++++++++
 .../x86_64-arista_7060_cx32s/platform_reboot  | 18 ++++
 3 files changed, 192 insertions(+)
 create mode 100755 device/arista/x86_64-arista_7050_qx32/platform_reboot
 create mode 100755 device/arista/x86_64-arista_7050_qx32s/platform_reboot
 create mode 100755 device/arista/x86_64-arista_7060_cx32s/platform_reboot

diff --git a/device/arista/x86_64-arista_7050_qx32/platform_reboot b/device/arista/x86_64-arista_7050_qx32/platform_reboot
new file mode 100755
index 000000000000..da438b8fe52b
--- /dev/null
+++ b/device/arista/x86_64-arista_7050_qx32/platform_reboot
@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+# Copyright (c) 2018 Arista Networks, Inc.  All rights reserved.
+# Arista Networks, Inc. Confidential and Proprietary.
+
+# Reboot script for 7050QX-32
+
+from __future__ import print_function
+import sys
+import mmap, os
+import subprocess
+from struct import pack, unpack
+
+class MmapResource( object ):
+   """Resource implementation for a directly-mapped memory region."""
+
+   def __init__( self, path ):
+      try:
+         fd = os.open( path, os.O_RDWR )
+      except EnvironmentError:
+         print( "FAIL can not open scd memory-map resource file" )
+         print( "FAIL are you running on the proper platform?" )
+         sys.exit( 1 )
+      try:
+         size = os.fstat( fd ).st_size
+      except EnvironmentError:
+         print( "FAIL can not fstat scd memory-map resource file" )
+         print( "FAIL are you running on the proper platform?" )
+         sys.exit( 1 )
+      try:
+         self.mmap_ = mmap.mmap( fd, size, mmap.MAP_SHARED,
+                      mmap.PROT_READ | mmap.PROT_WRITE )
+      except EnvironmentError:
+         print( "FAIL can not map scd memory-map file" )
+         print( "FAIL are you running on the proper platform?" )
+         sys.exit( 1 )
+      finally:
+         try:
+            # Note that closing the file descriptor has no effect on the memory map
+            os.close( fd )
+         except EnvironmentError:
+            print( "FAIL failed to close scd memory-map file" )
+            sys.exit( 1 )
+
+   def read32( self, addr ):
+      return unpack( '<L', self.mmap_[ addr : addr + 4 ] )[ 0 ]
+
+   def write32( self, addr, value ):
+      self.mmap_[ addr: addr + 4 ] = pack( '<L', value )
+
+def scdRegTest( scd, offset, val1, count ):
+   scd.write32( offset, val1 )
+   val2 = scd.read32( offset )
+   if val1 != val2:
+      print( "FAIL: scd write 0x%08x but read back 0x%08x in iter %d" %
+             ( val1, val2, count ) )
+      sys.exit( 17 )
+
+def scdScrRegTest( scd ):
+   scrOffset = 0x0130
+   for i in range( 0, 3 ):
+      scdRegTest( scd, scrOffset, 0xdeadbeef, i )
+      scdRegTest( scd, scrOffset, 0xa5a5a5a5, i )
+      scdRegTest( scd, scrOffset, 0x00000000, i )
+
+def reboot( scd ):
+   # reboot the system by writing to register 0x7000
+   print( "Rebooting" )
+   scd.write32( 0x7000, 0xDEAD )
+   print( "REBOOTED" )
+
+def main():
+   busName = "/sys/bus/pci/devices/0000:04:00.0/resource0"
+
+   subprocess.call( [ 'modprobe', 'scd' ] )
+
+   scd = MmapResource( busName )
+
+   #
+   # verify that we can read/write scd scratch register
+   #
+   scdScrRegTest( scd )
+
+   # reboot the system
+   reboot( scd )
+
+if __name__ == "__main__":
+   main()
diff --git a/device/arista/x86_64-arista_7050_qx32s/platform_reboot b/device/arista/x86_64-arista_7050_qx32s/platform_reboot
new file mode 100755
index 000000000000..6e9da639509d
--- /dev/null
+++ b/device/arista/x86_64-arista_7050_qx32s/platform_reboot
@@ -0,0 +1,87 @@
+#!/usr/bin/env python
+# Copyright (c) 2018 Arista Networks, Inc.  All rights reserved.
+# Arista Networks, Inc. Confidential and Proprietary.
+
+# Reboot script for 7050QX-32S
+
+from __future__ import print_function
+import sys
+import mmap, os
+import subprocess
+from struct import pack, unpack
+
+class MmapResource( object ):
+   """Resource implementation for a directly-mapped memory region."""
+
+   def __init__( self, path ):
+      try:
+         fd = os.open( path, os.O_RDWR )
+      except EnvironmentError:
+         print( "FAIL can not open scd memory-map resource file" )
+         print( "FAIL are you running on the proper platform?" )
+         sys.exit( 1 )
+      try:
+         size = os.fstat( fd ).st_size
+      except EnvironmentError:
+         print( "FAIL can not fstat scd memory-map resource file" )
+         print( "FAIL are you running on the proper platform?" )
+         sys.exit( 1 )
+      try:
+         self.mmap_ = mmap.mmap( fd, size, mmap.MAP_SHARED,
+                      mmap.PROT_READ | mmap.PROT_WRITE )
+      except EnvironmentError:
+         print( "FAIL can not map scd memory-map file" )
+         print( "FAIL are you running on the proper platform?" )
+         sys.exit( 1 )
+      finally:
+         try:
+            # Note that closing the file descriptor has no effect on the memory map
+            os.close( fd )
+         except EnvironmentError:
+            print( "FAIL failed to close scd memory-map file" )
+            sys.exit( 1 )
+
+   def read32( self, addr ):
+      return unpack( '<L', self.mmap_[ addr : addr + 4 ] )[ 0 ]
+
+   def write32( self, addr, value ):
+      self.mmap_[ addr: addr + 4 ] = pack( '<L', value )
+
+def scdRegTest( scd, offset, val1, count ):
+   scd.write32( offset, val1 )
+   val2 = scd.read32( offset )
+   if val1 != val2:
+      print( "FAIL: scd write 0x%08x but read back 0x%08x in iter %d" %
+             ( val1, val2, count ) )
+      sys.exit( 17 )
+
+def scdScrRegTest( scd ):
+   scrOffset = 0x0130
+   for i in range( 0, 3 ):
+      scdRegTest( scd, scrOffset, 0xdeadbeef, i )
+      scdRegTest( scd, scrOffset, 0xa5a5a5a5, i )
+      scdRegTest( scd, scrOffset, 0x00000000, i )
+
+def reboot( scd ):
+   # reboot the system by writing to register 0x7000
+   print( "Rebooting" )
+   scd.write32( 0x7000, 0xDEAD )
+   print( "REBOOTED" )
+
+def main():
+   busName = "/sys/bus/pci/devices/0000:02:00.0/resource0"
+
+   subprocess.call( [ 'modprobe', 'scd' ] )
+
+   scd = MmapResource( busName )
+
+   #
+   # verify that we can read/write scd scratch register
+   #
+   scdScrRegTest( scd )
+
+   # reboot the system
+   reboot( scd )
+
+if __name__ == "__main__":
+   main()
diff --git a/device/arista/x86_64-arista_7060_cx32s/platform_reboot b/device/arista/x86_64-arista_7060_cx32s/platform_reboot
new file mode 100755
index 000000000000..0e801683faec
--- /dev/null
+++ b/device/arista/x86_64-arista_7060_cx32s/platform_reboot
@@ -0,0 +1,18 @@
+#!/usr/bin/env python
+# Copyright (c) 2018 Arista Networks, Inc.  All rights reserved.
+# Arista Networks, Inc. Confidential and Proprietary.
+
+# Reboot script for 7060CX-32
+
+from __future__ import print_function
+
+import smbus
+
+def main():
+   print( "Rebooting" )
+   bus = smbus.SMBus( 1 )
+   bus.write_byte_data( 0x23, 0x04, 0xde )
+   print( "REBOOTED" )
+
+if __name__ == "__main__":
+   main()