1#!/bin/sh
2# SPDX-License-Identifier: GPL-2.0
3# Linux kernel symbol namespace import generator
4#
5# This script requires a minimum spatch version.
6SPATCH_REQ_VERSION="1.0.4"
7
8DIR="$(dirname $(readlink -f $0))/.."
9SPATCH="`which ${SPATCH:=spatch}`"
10if [ ! -x "$SPATCH" ]; then
11	echo 'spatch is part of the Coccinelle project and is available at http://coccinelle.lip6.fr/'
12	exit 1
13fi
14
15SPATCH_REQ_VERSION_NUM=$(echo $SPATCH_REQ_VERSION | ${DIR}/scripts/ld-version.sh)
16SPATCH_VERSION=$($SPATCH --version | head -1 | awk '{print $3}')
17SPATCH_VERSION_NUM=$(echo $SPATCH_VERSION | ${DIR}/scripts/ld-version.sh)
18
19if [ "$SPATCH_VERSION_NUM" -lt "$SPATCH_REQ_VERSION_NUM" ] ; then
20	echo "spatch needs to be version $SPATCH_REQ_VERSION or higher"
21	exit 1
22fi
23
24if [ "$KBUILD_EXTMOD" ]; then
25	src_prefix=
26else
27	src_prefix=$srctree/
28fi
29
30generate_deps_for_ns() {
31	$SPATCH --very-quiet --in-place --sp-file \
32		$srctree/scripts/coccinelle/misc/add_namespace.cocci -D nsdeps -D ns=$1 $2
33}
34
35generate_deps() {
36	local mod=${1%.ko:}
37	shift
38	local namespaces="$*"
39	local mod_source_files="`cat $mod.mod | sed -n 1p                      \
40					      | sed -e 's/\.o/\.c/g'           \
41					      | sed "s|[^ ]* *|${src_prefix}&|g"`"
42	for ns in $namespaces; do
43		echo "Adding namespace $ns to module $mod.ko."
44		generate_deps_for_ns $ns "$mod_source_files"
45		# sort the imports
46		for source_file in $mod_source_files; do
47			sed '/MODULE_IMPORT_NS/Q' $source_file > ${source_file}.tmp
48			offset=$(wc -l ${source_file}.tmp | awk '{print $1;}')
49			cat $source_file | grep MODULE_IMPORT_NS | LANG=C sort -u >> ${source_file}.tmp
50			tail -n +$((offset +1)) ${source_file} | grep -v MODULE_IMPORT_NS >> ${source_file}.tmp
51			if ! diff -q ${source_file} ${source_file}.tmp; then
52				mv ${source_file}.tmp ${source_file}
53			else
54				rm ${source_file}.tmp
55			fi
56		done
57	done
58}
59
60while read line
61do
62	generate_deps $line
63done < $MODULES_NSDEPS
64