X-Git-Url: https://scm.lunaixsky.com/lunaix-os.git/blobdiff_plain/9209afa5f69cffa48a04c4a9066357d5cba75926..04f32c3e67120f8498cdcf7926bab88e4e405258:/lunaix-os/scripts/gen_ksymtable.sh

diff --git a/lunaix-os/scripts/gen_ksymtable.sh b/lunaix-os/scripts/gen_ksymtable.sh
index 6b59e4d..1f974bd 100755
--- a/lunaix-os/scripts/gen_ksymtable.sh
+++ b/lunaix-os/scripts/gen_ksymtable.sh
@@ -1,12 +1,19 @@
-#!/bin/bash
+#!/usr/bin/env bash
 
 sym_types=$1
 bin=$2
 
 nm_out=$(nm -nfbsd "$bin")
+class_info=$(readelf -h "$bin" | grep 'Class:' | awk '{print $2}')
+
 allsyms=($nm_out)
 allsyms_len=${#allsyms[@]}
 
+dtype="4byte"
+if [ "$class_info" == 'ELF64' ]; then
+    dtype="8byte"
+fi
+
 syms_idx=()
 
 for (( i=0; i<allsyms_len; i+=3));
@@ -24,7 +31,7 @@ declare -A assoc_array
 echo '.section .ksymtable, "a", @progbits'
 echo "    .global __lunaix_ksymtable"
 echo "    __lunaix_ksymtable:"
-echo "        .long $syms_len"
+echo "        .$dtype $syms_len"
 echo "        .align 8"
 
 for i in "${syms_idx[@]}"
@@ -33,22 +40,20 @@ do
     type=${allsyms[$i + 1]}
     sym=${allsyms[$i + 2]}
 
-    echo "$(cat <<EOF
-        .long 0x$addr
-        .long __S$sym
+    cat <<EOF
+        .$dtype 0x$addr
+        .$dtype __S$sym
         .align 8
 
 EOF
-)"
     assoc_array["$sym"]=1
 done
 
 for sym_str in "${!assoc_array[@]}"
 do
-    echo "$(cat <<EOF
+    cat <<EOF
     __S$sym_str:
         .asciz "$sym_str"
         .align 8
 EOF
-)"
 done
\ No newline at end of file