#
# Author:: Kurt Yoder (ktyopscode@yoderhome.com)
# Copyright:: Copyright (c) 2010 Kurt Yoder
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
# 
#     http://www.apache.org/licenses/LICENSE-2.0
# 
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#

Ohai.plugin(:DMI) do

  collect_data(:solaris2) do
    # if we already have a "dmi" with keys (presumably from dmidecode), don't try smbios
    # note that a single key just means dmidecode exited with its version
    if (dmi.class.to_s == 'Mash') and (dmi.keys.length > 1) 
      Ohai::Log.debug('skipping smbios output, since DMI information has already been provided')
      return
    end

    dmi Mash.new

    # bad Solaris shows strings defined by system instead of SMB IDs 
    # this is what the *real* IDs are:
    # pulled from http://src.opensolaris.org/source/xref/nwam/nwam1/usr/src/uts/common/sys/smbios.h
    smb_to_id = {
      'SMB_TYPE_BIOS' =>         0, # BIOS information (R)
      'SMB_TYPE_SYSTEM' =>       1, # system information (R)
      'SMB_TYPE_BASEBOARD' =>    2, # base board
      'SMB_TYPE_CHASSIS' =>      3, # system enclosure or chassis (R)
      'SMB_TYPE_PROCESSOR' =>    4, # processor (R)
      'SMB_TYPE_MEMCTL' =>       5, # memory controller (O)
      'SMB_TYPE_MEMMOD' =>       6, # memory module (O)
      'SMB_TYPE_CACHE' =>        7, # processor cache (R)
      'SMB_TYPE_PORT' =>         8, # port connector
      'SMB_TYPE_SLOT' =>         9, # upgradeable system slot (R)
      'SMB_TYPE_OBDEVS' =>       10, # on-board devices
      'SMB_TYPE_OEMSTR' =>       11, # OEM string table
      'SMB_TYPE_SYSCONFSTR' =>   12, # system configuration string table
      'SMB_TYPE_LANG' =>         13, # BIOS language information
      'SMB_TYPE_GROUP' =>        14, # group associations
      'SMB_TYPE_EVENTLOG' =>     15, # system event log
      'SMB_TYPE_MEMARRAY' =>     16, # physical memory array (R)
      'SMB_TYPE_MEMDEVICE' =>    17, # memory device (R)
      'SMB_TYPE_MEMERR32' =>     18, # 32-bit memory error information
      'SMB_TYPE_MEMARRAYMAP' =>  19, # memory array mapped address (R)
      'SMB_TYPE_MEMDEVICEMAP' => 20, # memory device mapped address (R)
      'SMB_TYPE_POINTDEV' =>     21, # built-in pointing device
      'SMB_TYPE_BATTERY' =>      22, # portable battery
      'SMB_TYPE_RESET' =>        23, # system reset settings
      'SMB_TYPE_SECURITY' =>     24, # hardware security settings
      'SMB_TYPE_POWERCTL' =>     25, # system power controls
      'SMB_TYPE_VPROBE' =>       26, # voltage probe
      'SMB_TYPE_COOLDEV' =>      27, # cooling device
      'SMB_TYPE_TPROBE' =>       28, # temperature probe
      'SMB_TYPE_IPROBE' =>       29, # current probe
      'SMB_TYPE_OOBRA' =>        30, # out-of-band remote access facility
      'SMB_TYPE_BIS' =>          31, # boot integrity services
      'SMB_TYPE_BOOT' =>         32, # system boot status (R)
      'SMB_TYPE_MEMERR64' =>     33, # 64-bit memory error information
      'SMB_TYPE_MGMTDEV' =>      34, # management device
      'SMB_TYPE_MGMTDEVCP' =>    35, # management device component
      'SMB_TYPE_MGMTDEVDATA' =>  36, # management device threshold data
      'SMB_TYPE_MEMCHAN' =>      37, # memory channel
      'SMB_TYPE_IPMIDEV' =>      38, # IPMI device information
      'SMB_TYPE_POWERSUP' =>     39, # system power supply
      'SMB_TYPE_INACTIVE' =>     126, # inactive table entry
      'SMB_TYPE_EOT' =>          127, # end of table
      'SMB_TYPE_OEM_LO' =>       128, # start of OEM-specific type range
      'SUN_OEM_EXT_PROCESSOR' => 132, # processor extended info
      'SUN_OEM_PCIEXRC' =>       138, # PCIE RootComplex/RootPort info
      'SUN_OEM_EXT_MEMARRAY' =>  144, # phys memory array extended info
      'SUN_OEM_EXT_MEMDEVICE' => 145, # memory device extended info
      'SMB_TYPE_OEM_HI' =>       256, # end of OEM-specific type range
    }

    # all output lines should fall within one of these patterns
    header_type_line = /^ID\s+SIZE\s+TYPE/
    header_information_line = /^(\d+)\s+(\d+)\s+(\S+)\s+\(([^\)]+)\)/
    blank_line = /^\s*$/
    data_key_value_line = /^  ([^:]+): (.*)/
    data_key_only_line = /^  (\S.*)(:\s*)?$/
    extended_data_line = /^\t(\S+) \((.+)\)/

    dmi_record = nil
    field = nil

    so = shell_out("smbios")
    # ==== EXAMPLE: ====
    # ID    SIZE TYPE
    # 0     40   SMB_TYPE_BIOS (BIOS information)
    # 
    #   Vendor: HP
    #   Version String: 2.16  
    # ... similar lines trimmed
    #   Characteristics: 0x7fc9da80
    #         SMB_BIOSFL_PCI (PCI is supported)
    # ... similar lines trimmed
    # note the second level of indentation is via a *tab*
    so.stdout.lines do |raw_line|
      next if header_type_line.match(raw_line)
      next if blank_line.match(raw_line)

      # remove/replace any characters that don't fall inside permissible ASCII range, or whitespace
      line = raw_line.gsub(/[^\x20-\x7E\n\t\r]/, '.')
      if (line != raw_line)
        Ohai::Log.debug("converted characters from line:\n#{raw_line}")
      end

      if header_information = header_information_line.match(line)
        dmi_record = {}

        # look up SMB ID
        if smb_to_id.has_key?(header_information[3])
          dmi_record[:type] = DMI.id_lookup(smb_to_id[header_information[3]])

        else
          dmi_record[:type] = header_information[3].downcase
          Ohai::Log.debug("unrecognized header type; falling back to #{dmi_record[:type]}")
        end

        dmi[dmi_record[:type]] = Mash.new unless dmi.has_key?(dmi_record[:type])
        dmi[dmi_record[:type]][:all_records] = [] unless dmi[dmi_record[:type]].has_key?(:all_records)
        dmi_record[:position] = dmi[dmi_record[:type]][:all_records].length
        dmi[dmi_record[:type]][:all_records].push(Mash.new)
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][:record_id] = header_information[1]
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][:size] = header_information[2]
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][:application_identifier] = header_information[4]
        field = nil

      elsif data = data_key_value_line.match(line)
        if dmi_record == nil
          Ohai::Log.debug("unexpected data line found before header; discarding:\n#{line}")
          next
        end
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][data[1]] = data[2]
        field = data[1]

      elsif data = data_key_only_line.match(line)
        if dmi_record == nil
          Ohai::Log.debug("unexpected data line found before header; discarding:\n#{line}")
          next
        end
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][data[1]] = ''
        field = data[1]

      elsif extended_data = extended_data_line.match(line)
        if dmi_record == nil
          Ohai::Log.debug("unexpected extended data line found before header; discarding:\n#{line}")
          next
        end
        if field == nil
          Ohai::Log.debug("unexpected extended data line found outside data section; discarding:\n#{line}")
          next
        end
        # overwrite "raw" value with a new Mash
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][field] = Mash.new unless dmi[dmi_record[:type]][:all_records][dmi_record[:position]][field].class.to_s == 'Mash'
        dmi[dmi_record[:type]][:all_records][dmi_record[:position]][field][extended_data[1]] = extended_data[2]

      else
        Ohai::Log.debug("unrecognized output line; discarding:\n#{line}")

      end
    end

    DMI.convenience_keys(dmi)
  end
end