Skip to content

Commit

Permalink
[Arista]: Add a reboot script for all remaining platforms. (#1706)
Browse files Browse the repository at this point in the history
Adds a platform_reboot script for 7050QX-32, 7050QX-32S and 7060CX-32S.
This allow a proper cold reboot to happen.
  • Loading branch information
Staphylo authored and lguohan committed May 19, 2018
1 parent 20d0531 commit 4faf96f
Show file tree
Hide file tree
Showing 3 changed files with 192 additions and 0 deletions.
87 changes: 87 additions & 0 deletions device/arista/x86_64-arista_7050_qx32/platform_reboot
Original file line number Diff line number Diff line change
@@ -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()
87 changes: 87 additions & 0 deletions device/arista/x86_64-arista_7050_qx32s/platform_reboot
Original file line number Diff line number Diff line change
@@ -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()
18 changes: 18 additions & 0 deletions device/arista/x86_64-arista_7060_cx32s/platform_reboot
Original file line number Diff line number Diff line change
@@ -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()

0 comments on commit 4faf96f

Please sign in to comment.