1#!/usr/bin/python
2# The TCM v4 multi-protocol fabric module generation script for drivers/target/$NEW_MOD
3#
4# Copyright (c) 2010 Rising Tide Systems
5# Copyright (c) 2010 Linux-iSCSI.org
6#
7# Author: nab@kernel.org
8#
9import os, sys
10import subprocess as sub
11import string
12import re
13import optparse
14
15tcm_dir = ""
16
17fabric_ops = []
18fabric_mod_dir = ""
19fabric_mod_port = ""
20fabric_mod_init_port = ""
21
22def tcm_mod_err(msg):
23	print msg
24	sys.exit(1)
25
26def tcm_mod_create_module_subdir(fabric_mod_dir_var):
27
28	if os.path.isdir(fabric_mod_dir_var) == True:
29		return 1
30
31	print "Creating fabric_mod_dir: " + fabric_mod_dir_var
32	ret = os.mkdir(fabric_mod_dir_var)
33	if ret:
34		tcm_mod_err("Unable to mkdir " + fabric_mod_dir_var)
35
36	return
37
38def tcm_mod_build_FC_include(fabric_mod_dir_var, fabric_mod_name):
39	global fabric_mod_port
40	global fabric_mod_init_port
41	buf = ""
42
43	f = fabric_mod_dir_var + "/" + fabric_mod_name + "_base.h"
44	print "Writing file: " + f
45
46	p = open(f, 'w');
47	if not p:
48		tcm_mod_err("Unable to open file: " + f)
49
50	buf = "#define " + fabric_mod_name.upper() + "_VERSION	\"v0.1\"\n"
51	buf += "#define " + fabric_mod_name.upper() + "_NAMELEN	32\n"
52	buf += "\n"
53	buf += "struct " + fabric_mod_name + "_tpg {\n"
54	buf += "	/* FC lport target portal group tag for TCM */\n"
55	buf += "	u16 lport_tpgt;\n"
56	buf += "	/* Pointer back to " + fabric_mod_name + "_lport */\n"
57	buf += "	struct " + fabric_mod_name + "_lport *lport;\n"
58	buf += "	/* Returned by " + fabric_mod_name + "_make_tpg() */\n"
59	buf += "	struct se_portal_group se_tpg;\n"
60	buf += "};\n"
61	buf += "\n"
62	buf += "struct " + fabric_mod_name + "_lport {\n"
63	buf += "	/* Binary World Wide unique Port Name for FC Target Lport */\n"
64	buf += "	u64 lport_wwpn;\n"
65	buf += "	/* ASCII formatted WWPN for FC Target Lport */\n"
66	buf += "	char lport_name[" + fabric_mod_name.upper() + "_NAMELEN];\n"
67	buf += "	/* Returned by " + fabric_mod_name + "_make_lport() */\n"
68	buf += "	struct se_wwn lport_wwn;\n"
69	buf += "};\n"
70
71	ret = p.write(buf)
72	if ret:
73		tcm_mod_err("Unable to write f: " + f)
74
75	p.close()
76
77	fabric_mod_port = "lport"
78	fabric_mod_init_port = "nport"
79
80	return
81
82def tcm_mod_build_SAS_include(fabric_mod_dir_var, fabric_mod_name):
83	global fabric_mod_port
84	global fabric_mod_init_port
85	buf = ""
86
87	f = fabric_mod_dir_var + "/" + fabric_mod_name + "_base.h"
88	print "Writing file: " + f
89
90	p = open(f, 'w');
91	if not p:
92		tcm_mod_err("Unable to open file: " + f)
93
94	buf = "#define " + fabric_mod_name.upper() + "_VERSION  \"v0.1\"\n"
95	buf += "#define " + fabric_mod_name.upper() + "_NAMELEN 32\n"
96	buf += "\n"
97	buf += "struct " + fabric_mod_name + "_tpg {\n"
98	buf += "	/* SAS port target portal group tag for TCM */\n"
99	buf += "	u16 tport_tpgt;\n"
100	buf += "	/* Pointer back to " + fabric_mod_name + "_tport */\n"
101	buf += "	struct " + fabric_mod_name + "_tport *tport;\n"
102	buf += "	/* Returned by " + fabric_mod_name + "_make_tpg() */\n"
103	buf += "	struct se_portal_group se_tpg;\n"
104	buf += "};\n\n"
105	buf += "struct " + fabric_mod_name + "_tport {\n"
106	buf += "	/* Binary World Wide unique Port Name for SAS Target port */\n"
107	buf += "	u64 tport_wwpn;\n"
108	buf += "	/* ASCII formatted WWPN for SAS Target port */\n"
109	buf += "	char tport_name[" + fabric_mod_name.upper() + "_NAMELEN];\n"
110	buf += "	/* Returned by " + fabric_mod_name + "_make_tport() */\n"
111	buf += "	struct se_wwn tport_wwn;\n"
112	buf += "};\n"
113
114	ret = p.write(buf)
115	if ret:
116		tcm_mod_err("Unable to write f: " + f)
117
118	p.close()
119
120	fabric_mod_port = "tport"
121	fabric_mod_init_port = "iport"
122
123	return
124
125def tcm_mod_build_iSCSI_include(fabric_mod_dir_var, fabric_mod_name):
126	global fabric_mod_port
127	global fabric_mod_init_port
128	buf = ""
129
130	f = fabric_mod_dir_var + "/" + fabric_mod_name + "_base.h"
131	print "Writing file: " + f
132
133	p = open(f, 'w');
134	if not p:
135		tcm_mod_err("Unable to open file: " + f)
136
137	buf = "#define " + fabric_mod_name.upper() + "_VERSION  \"v0.1\"\n"
138	buf += "#define " + fabric_mod_name.upper() + "_NAMELEN 32\n"
139	buf += "\n"
140	buf += "struct " + fabric_mod_name + "_tpg {\n"
141	buf += "	/* iSCSI target portal group tag for TCM */\n"
142	buf += "	u16 tport_tpgt;\n"
143	buf += "	/* Pointer back to " + fabric_mod_name + "_tport */\n"
144	buf += "	struct " + fabric_mod_name + "_tport *tport;\n"
145	buf += "	/* Returned by " + fabric_mod_name + "_make_tpg() */\n"
146	buf += "	struct se_portal_group se_tpg;\n"
147	buf += "};\n\n"
148	buf += "struct " + fabric_mod_name + "_tport {\n"
149	buf += "	/* ASCII formatted TargetName for IQN */\n"
150	buf += "	char tport_name[" + fabric_mod_name.upper() + "_NAMELEN];\n"
151	buf += "	/* Returned by " + fabric_mod_name + "_make_tport() */\n"
152	buf += "	struct se_wwn tport_wwn;\n"
153	buf += "};\n"
154
155	ret = p.write(buf)
156	if ret:
157		tcm_mod_err("Unable to write f: " + f)
158
159	p.close()
160
161	fabric_mod_port = "tport"
162	fabric_mod_init_port = "iport"
163
164	return
165
166def tcm_mod_build_base_includes(proto_ident, fabric_mod_dir_val, fabric_mod_name):
167
168	if proto_ident == "FC":
169		tcm_mod_build_FC_include(fabric_mod_dir_val, fabric_mod_name)
170	elif proto_ident == "SAS":
171		tcm_mod_build_SAS_include(fabric_mod_dir_val, fabric_mod_name)
172	elif proto_ident == "iSCSI":
173		tcm_mod_build_iSCSI_include(fabric_mod_dir_val, fabric_mod_name)
174	else:
175		print "Unsupported proto_ident: " + proto_ident
176		sys.exit(1)
177
178	return
179
180def tcm_mod_build_configfs(proto_ident, fabric_mod_dir_var, fabric_mod_name):
181	buf = ""
182
183	f = fabric_mod_dir_var + "/" + fabric_mod_name + "_configfs.c"
184	print "Writing file: " + f
185
186        p = open(f, 'w');
187        if not p:
188                tcm_mod_err("Unable to open file: " + f)
189
190	buf = "#include <linux/module.h>\n"
191	buf += "#include <linux/moduleparam.h>\n"
192	buf += "#include <linux/version.h>\n"
193	buf += "#include <generated/utsrelease.h>\n"
194	buf += "#include <linux/utsname.h>\n"
195	buf += "#include <linux/init.h>\n"
196	buf += "#include <linux/slab.h>\n"
197	buf += "#include <linux/kthread.h>\n"
198	buf += "#include <linux/types.h>\n"
199	buf += "#include <linux/string.h>\n"
200	buf += "#include <linux/configfs.h>\n"
201	buf += "#include <linux/ctype.h>\n"
202	buf += "#include <asm/unaligned.h>\n"
203	buf += "#include <scsi/scsi_proto.h>\n\n"
204	buf += "#include <target/target_core_base.h>\n"
205	buf += "#include <target/target_core_fabric.h>\n"
206	buf += "#include \"" + fabric_mod_name + "_base.h\"\n"
207	buf += "#include \"" + fabric_mod_name + "_fabric.h\"\n\n"
208
209	buf += "static const struct target_core_fabric_ops " + fabric_mod_name + "_ops;\n\n"
210
211	buf += "static struct se_portal_group *" + fabric_mod_name + "_make_tpg(\n"
212	buf += "	struct se_wwn *wwn,\n"
213	buf += "	struct config_group *group,\n"
214	buf += "	const char *name)\n"
215	buf += "{\n"
216	buf += "	struct " + fabric_mod_name + "_" + fabric_mod_port + "*" + fabric_mod_port + " = container_of(wwn,\n"
217	buf += "			struct " + fabric_mod_name + "_" + fabric_mod_port + ", " + fabric_mod_port + "_wwn);\n\n"
218	buf += "	struct " + fabric_mod_name + "_tpg *tpg;\n"
219	buf += "	unsigned long tpgt;\n"
220	buf += "	int ret;\n\n"
221	buf += "	if (strstr(name, \"tpgt_\") != name)\n"
222	buf += "		return ERR_PTR(-EINVAL);\n"
223	buf += "	if (kstrtoul(name + 5, 10, &tpgt) || tpgt > UINT_MAX)\n"
224	buf += "		return ERR_PTR(-EINVAL);\n\n"
225	buf += "	tpg = kzalloc(sizeof(struct " + fabric_mod_name + "_tpg), GFP_KERNEL);\n"
226	buf += "	if (!tpg) {\n"
227	buf += "		printk(KERN_ERR \"Unable to allocate struct " + fabric_mod_name + "_tpg\");\n"
228	buf += "		return ERR_PTR(-ENOMEM);\n"
229	buf += "	}\n"
230	buf += "	tpg->" + fabric_mod_port + " = " + fabric_mod_port + ";\n"
231	buf += "	tpg->" + fabric_mod_port + "_tpgt = tpgt;\n\n"
232
233	if proto_ident == "FC":
234		buf += "	ret = core_tpg_register(wwn, &tpg->se_tpg, SCSI_PROTOCOL_FCP);\n"
235	elif proto_ident == "SAS":
236		buf += "	ret = core_tpg_register(wwn, &tpg->se_tpg, SCSI_PROTOCOL_SAS);\n"
237	elif proto_ident == "iSCSI":
238		buf += "	ret = core_tpg_register(wwn, &tpg->se_tpg, SCSI_PROTOCOL_ISCSI);\n"
239
240	buf += "	if (ret < 0) {\n"
241	buf += "		kfree(tpg);\n"
242	buf += "		return NULL;\n"
243	buf += "	}\n"
244	buf += "	return &tpg->se_tpg;\n"
245	buf += "}\n\n"
246	buf += "static void " + fabric_mod_name + "_drop_tpg(struct se_portal_group *se_tpg)\n"
247	buf += "{\n"
248	buf += "	struct " + fabric_mod_name + "_tpg *tpg = container_of(se_tpg,\n"
249	buf += "				struct " + fabric_mod_name + "_tpg, se_tpg);\n\n"
250	buf += "	core_tpg_deregister(se_tpg);\n"
251	buf += "	kfree(tpg);\n"
252	buf += "}\n\n"
253
254	buf += "static struct se_wwn *" + fabric_mod_name + "_make_" + fabric_mod_port + "(\n"
255	buf += "	struct target_fabric_configfs *tf,\n"
256	buf += "	struct config_group *group,\n"
257	buf += "	const char *name)\n"
258	buf += "{\n"
259	buf += "	struct " + fabric_mod_name + "_" + fabric_mod_port + " *" + fabric_mod_port + ";\n"
260
261	if proto_ident == "FC" or proto_ident == "SAS":
262		buf += "	u64 wwpn = 0;\n\n"
263
264	buf += "	/* if (" + fabric_mod_name + "_parse_wwn(name, &wwpn, 1) < 0)\n"
265	buf += "		return ERR_PTR(-EINVAL); */\n\n"
266	buf += "	" + fabric_mod_port + " = kzalloc(sizeof(struct " + fabric_mod_name + "_" + fabric_mod_port + "), GFP_KERNEL);\n"
267	buf += "	if (!" + fabric_mod_port + ") {\n"
268	buf += "		printk(KERN_ERR \"Unable to allocate struct " + fabric_mod_name + "_" + fabric_mod_port + "\");\n"
269	buf += "		return ERR_PTR(-ENOMEM);\n"
270	buf += "	}\n"
271
272	if proto_ident == "FC" or proto_ident == "SAS":
273		buf += "	" + fabric_mod_port + "->" + fabric_mod_port + "_wwpn = wwpn;\n"
274
275	buf += "	/* " + fabric_mod_name + "_format_wwn(&" + fabric_mod_port + "->" + fabric_mod_port + "_name[0], " + fabric_mod_name.upper() + "_NAMELEN, wwpn); */\n\n"
276	buf += "	return &" + fabric_mod_port + "->" + fabric_mod_port + "_wwn;\n"
277	buf += "}\n\n"
278	buf += "static void " + fabric_mod_name + "_drop_" + fabric_mod_port + "(struct se_wwn *wwn)\n"
279	buf += "{\n"
280	buf += "	struct " + fabric_mod_name + "_" + fabric_mod_port + " *" + fabric_mod_port + " = container_of(wwn,\n"
281	buf += "				struct " + fabric_mod_name + "_" + fabric_mod_port + ", " + fabric_mod_port + "_wwn);\n"
282	buf += "	kfree(" + fabric_mod_port + ");\n"
283	buf += "}\n\n"
284
285	buf += "static const struct target_core_fabric_ops " + fabric_mod_name + "_ops = {\n"
286	buf += "	.module				= THIS_MODULE,\n"
287	buf += "	.name				= \"" + fabric_mod_name + "\",\n"
288	buf += "	.get_fabric_name		= " + fabric_mod_name + "_get_fabric_name,\n"
289	buf += "	.tpg_get_wwn			= " + fabric_mod_name + "_get_fabric_wwn,\n"
290	buf += "	.tpg_get_tag			= " + fabric_mod_name + "_get_tag,\n"
291	buf += "	.tpg_check_demo_mode		= " + fabric_mod_name + "_check_false,\n"
292	buf += "	.tpg_check_demo_mode_cache	= " + fabric_mod_name + "_check_true,\n"
293	buf += "	.tpg_check_demo_mode_write_protect = " + fabric_mod_name + "_check_true,\n"
294	buf += "	.tpg_check_prod_mode_write_protect = " + fabric_mod_name + "_check_false,\n"
295	buf += "	.tpg_get_inst_index		= " + fabric_mod_name + "_tpg_get_inst_index,\n"
296	buf += "	.release_cmd			= " + fabric_mod_name + "_release_cmd,\n"
297	buf += "	.sess_get_index			= " + fabric_mod_name + "_sess_get_index,\n"
298	buf += "	.sess_get_initiator_sid		= NULL,\n"
299	buf += "	.write_pending			= " + fabric_mod_name + "_write_pending,\n"
300	buf += "	.set_default_node_attributes	= " + fabric_mod_name + "_set_default_node_attrs,\n"
301	buf += "	.get_cmd_state			= " + fabric_mod_name + "_get_cmd_state,\n"
302	buf += "	.queue_data_in			= " + fabric_mod_name + "_queue_data_in,\n"
303	buf += "	.queue_status			= " + fabric_mod_name + "_queue_status,\n"
304	buf += "	.queue_tm_rsp			= " + fabric_mod_name + "_queue_tm_rsp,\n"
305	buf += "	.aborted_task			= " + fabric_mod_name + "_aborted_task,\n"
306	buf += "	/*\n"
307	buf += "	 * Setup function pointers for generic logic in target_core_fabric_configfs.c\n"
308	buf += "	 */\n"
309	buf += "	.fabric_make_wwn		= " + fabric_mod_name + "_make_" + fabric_mod_port + ",\n"
310	buf += "	.fabric_drop_wwn		= " + fabric_mod_name + "_drop_" + fabric_mod_port + ",\n"
311	buf += "	.fabric_make_tpg		= " + fabric_mod_name + "_make_tpg,\n"
312	buf += "	.fabric_drop_tpg		= " + fabric_mod_name + "_drop_tpg,\n"
313	buf += "};\n\n"
314
315	buf += "static int __init " + fabric_mod_name + "_init(void)\n"
316	buf += "{\n"
317	buf += "	return target_register_template(&" + fabric_mod_name + "_ops);\n"
318	buf += "};\n\n"
319
320	buf += "static void __exit " + fabric_mod_name + "_exit(void)\n"
321	buf += "{\n"
322	buf += "	target_unregister_template(&" + fabric_mod_name + "_ops);\n"
323	buf += "};\n\n"
324
325	buf += "MODULE_DESCRIPTION(\"" + fabric_mod_name.upper() + " series fabric driver\");\n"
326	buf += "MODULE_LICENSE(\"GPL\");\n"
327	buf += "module_init(" + fabric_mod_name + "_init);\n"
328	buf += "module_exit(" + fabric_mod_name + "_exit);\n"
329
330	ret = p.write(buf)
331	if ret:
332		tcm_mod_err("Unable to write f: " + f)
333
334	p.close()
335
336	return
337
338def tcm_mod_scan_fabric_ops(tcm_dir):
339
340	fabric_ops_api = tcm_dir + "include/target/target_core_fabric.h"
341
342	print "Using tcm_mod_scan_fabric_ops: " + fabric_ops_api
343	process_fo = 0;
344
345	p = open(fabric_ops_api, 'r')
346
347	line = p.readline()
348	while line:
349		if process_fo == 0 and re.search('struct target_core_fabric_ops {', line):
350			line = p.readline()
351			continue
352
353		if process_fo == 0:
354			process_fo = 1;
355			line = p.readline()
356			# Search for function pointer
357			if not re.search('\(\*', line):
358				continue
359
360			fabric_ops.append(line.rstrip())
361			continue
362
363		line = p.readline()
364		# Search for function pointer
365		if not re.search('\(\*', line):
366			continue
367
368		fabric_ops.append(line.rstrip())
369
370	p.close()
371	return
372
373def tcm_mod_dump_fabric_ops(proto_ident, fabric_mod_dir_var, fabric_mod_name):
374	buf = ""
375	bufi = ""
376
377	f = fabric_mod_dir_var + "/" + fabric_mod_name + "_fabric.c"
378	print "Writing file: " + f
379
380	p = open(f, 'w')
381	if not p:
382		tcm_mod_err("Unable to open file: " + f)
383
384	fi = fabric_mod_dir_var + "/" + fabric_mod_name + "_fabric.h"
385	print "Writing file: " + fi
386
387	pi = open(fi, 'w')
388	if not pi:
389		tcm_mod_err("Unable to open file: " + fi)
390
391	buf = "#include <linux/slab.h>\n"
392	buf += "#include <linux/kthread.h>\n"
393	buf += "#include <linux/types.h>\n"
394	buf += "#include <linux/list.h>\n"
395	buf += "#include <linux/types.h>\n"
396	buf += "#include <linux/string.h>\n"
397	buf += "#include <linux/ctype.h>\n"
398	buf += "#include <asm/unaligned.h>\n"
399	buf += "#include <scsi/scsi_common.h>\n"
400	buf += "#include <scsi/scsi_proto.h>\n"
401	buf += "#include <target/target_core_base.h>\n"
402	buf += "#include <target/target_core_fabric.h>\n"
403	buf += "#include \"" + fabric_mod_name + "_base.h\"\n"
404	buf += "#include \"" + fabric_mod_name + "_fabric.h\"\n\n"
405
406	buf += "int " + fabric_mod_name + "_check_true(struct se_portal_group *se_tpg)\n"
407	buf += "{\n"
408	buf += "	return 1;\n"
409	buf += "}\n\n"
410	bufi += "int " + fabric_mod_name + "_check_true(struct se_portal_group *);\n"
411
412	buf += "int " + fabric_mod_name + "_check_false(struct se_portal_group *se_tpg)\n"
413	buf += "{\n"
414	buf += "	return 0;\n"
415	buf += "}\n\n"
416	bufi += "int " + fabric_mod_name + "_check_false(struct se_portal_group *);\n"
417
418	total_fabric_ops = len(fabric_ops)
419	i = 0
420
421	while i < total_fabric_ops:
422		fo = fabric_ops[i]
423		i += 1
424#		print "fabric_ops: " + fo
425
426		if re.search('get_fabric_name', fo):
427			buf += "char *" + fabric_mod_name + "_get_fabric_name(void)\n"
428			buf += "{\n"
429			buf += "	return \"" + fabric_mod_name + "\";\n"
430			buf += "}\n\n"
431			bufi += "char *" + fabric_mod_name + "_get_fabric_name(void);\n"
432			continue
433
434		if re.search('get_wwn', fo):
435			buf += "char *" + fabric_mod_name + "_get_fabric_wwn(struct se_portal_group *se_tpg)\n"
436			buf += "{\n"
437			buf += "	struct " + fabric_mod_name + "_tpg *tpg = container_of(se_tpg,\n"
438			buf += "				struct " + fabric_mod_name + "_tpg, se_tpg);\n"
439			buf += "	struct " + fabric_mod_name + "_" + fabric_mod_port + " *" + fabric_mod_port + " = tpg->" + fabric_mod_port + ";\n\n"
440			buf += "	return &" + fabric_mod_port + "->" + fabric_mod_port + "_name[0];\n"
441			buf += "}\n\n"
442			bufi += "char *" + fabric_mod_name + "_get_fabric_wwn(struct se_portal_group *);\n"
443
444		if re.search('get_tag', fo):
445			buf += "u16 " + fabric_mod_name + "_get_tag(struct se_portal_group *se_tpg)\n"
446			buf += "{\n"
447			buf += "	struct " + fabric_mod_name + "_tpg *tpg = container_of(se_tpg,\n"
448			buf += "				struct " + fabric_mod_name + "_tpg, se_tpg);\n"
449			buf += "	return tpg->" + fabric_mod_port + "_tpgt;\n"
450			buf += "}\n\n"
451			bufi += "u16 " + fabric_mod_name + "_get_tag(struct se_portal_group *);\n"
452
453		if re.search('tpg_get_inst_index\)\(', fo):
454			buf += "u32 " + fabric_mod_name + "_tpg_get_inst_index(struct se_portal_group *se_tpg)\n"
455			buf += "{\n"
456			buf += "	return 1;\n"
457			buf += "}\n\n"
458			bufi += "u32 " + fabric_mod_name + "_tpg_get_inst_index(struct se_portal_group *);\n"
459
460		if re.search('\*release_cmd\)\(', fo):
461			buf += "void " + fabric_mod_name + "_release_cmd(struct se_cmd *se_cmd)\n"
462			buf += "{\n"
463			buf += "	return;\n"
464			buf += "}\n\n"
465			bufi += "void " + fabric_mod_name + "_release_cmd(struct se_cmd *);\n"
466
467		if re.search('sess_get_index\)\(', fo):
468			buf += "u32 " + fabric_mod_name + "_sess_get_index(struct se_session *se_sess)\n"
469			buf += "{\n"
470			buf += "	return 0;\n"
471			buf += "}\n\n"
472			bufi += "u32 " + fabric_mod_name + "_sess_get_index(struct se_session *);\n"
473
474		if re.search('write_pending\)\(', fo):
475			buf += "int " + fabric_mod_name + "_write_pending(struct se_cmd *se_cmd)\n"
476			buf += "{\n"
477			buf += "	return 0;\n"
478			buf += "}\n\n"
479			bufi += "int " + fabric_mod_name + "_write_pending(struct se_cmd *);\n"
480
481		if re.search('set_default_node_attributes\)\(', fo):
482			buf += "void " + fabric_mod_name + "_set_default_node_attrs(struct se_node_acl *nacl)\n"
483			buf += "{\n"
484			buf += "	return;\n"
485			buf += "}\n\n"
486			bufi += "void " + fabric_mod_name + "_set_default_node_attrs(struct se_node_acl *);\n"
487
488		if re.search('get_cmd_state\)\(', fo):
489			buf += "int " + fabric_mod_name + "_get_cmd_state(struct se_cmd *se_cmd)\n"
490			buf += "{\n"
491			buf += "	return 0;\n"
492			buf += "}\n\n"
493			bufi += "int " + fabric_mod_name + "_get_cmd_state(struct se_cmd *);\n"
494
495		if re.search('queue_data_in\)\(', fo):
496			buf += "int " + fabric_mod_name + "_queue_data_in(struct se_cmd *se_cmd)\n"
497			buf += "{\n"
498			buf += "	return 0;\n"
499			buf += "}\n\n"
500			bufi += "int " + fabric_mod_name + "_queue_data_in(struct se_cmd *);\n"
501
502		if re.search('queue_status\)\(', fo):
503			buf += "int " + fabric_mod_name + "_queue_status(struct se_cmd *se_cmd)\n"
504			buf += "{\n"
505			buf += "	return 0;\n"
506			buf += "}\n\n"
507			bufi += "int " + fabric_mod_name + "_queue_status(struct se_cmd *);\n"
508
509		if re.search('queue_tm_rsp\)\(', fo):
510			buf += "void " + fabric_mod_name + "_queue_tm_rsp(struct se_cmd *se_cmd)\n"
511			buf += "{\n"
512			buf += "	return;\n"
513			buf += "}\n\n"
514			bufi += "void " + fabric_mod_name + "_queue_tm_rsp(struct se_cmd *);\n"
515
516		if re.search('aborted_task\)\(', fo):
517			buf += "void " + fabric_mod_name + "_aborted_task(struct se_cmd *se_cmd)\n"
518			buf += "{\n"
519			buf += "	return;\n"
520			buf += "}\n\n"
521			bufi += "void " + fabric_mod_name + "_aborted_task(struct se_cmd *);\n"
522
523	ret = p.write(buf)
524	if ret:
525		tcm_mod_err("Unable to write f: " + f)
526
527	p.close()
528
529	ret = pi.write(bufi)
530	if ret:
531		tcm_mod_err("Unable to write fi: " + fi)
532
533	pi.close()
534	return
535
536def tcm_mod_build_kbuild(fabric_mod_dir_var, fabric_mod_name):
537
538	buf = ""
539	f = fabric_mod_dir_var + "/Makefile"
540	print "Writing file: " + f
541
542	p = open(f, 'w')
543	if not p:
544		tcm_mod_err("Unable to open file: " + f)
545
546	buf += fabric_mod_name + "-objs			:= " + fabric_mod_name + "_fabric.o \\\n"
547	buf += "					   " + fabric_mod_name + "_configfs.o\n"
548	buf += "obj-$(CONFIG_" + fabric_mod_name.upper() + ")		+= " + fabric_mod_name + ".o\n"
549
550	ret = p.write(buf)
551	if ret:
552		tcm_mod_err("Unable to write f: " + f)
553
554	p.close()
555	return
556
557def tcm_mod_build_kconfig(fabric_mod_dir_var, fabric_mod_name):
558
559	buf = ""
560	f = fabric_mod_dir_var + "/Kconfig"
561	print "Writing file: " + f
562
563	p = open(f, 'w')
564	if not p:
565		tcm_mod_err("Unable to open file: " + f)
566
567	buf = "config " + fabric_mod_name.upper() + "\n"
568	buf += "	tristate \"" + fabric_mod_name.upper() + " fabric module\"\n"
569	buf += "	depends on TARGET_CORE && CONFIGFS_FS\n"
570	buf += "	default n\n"
571	buf += "	---help---\n"
572	buf += "	Say Y here to enable the " + fabric_mod_name.upper() + " fabric module\n"
573
574	ret = p.write(buf)
575	if ret:
576		tcm_mod_err("Unable to write f: " + f)
577
578	p.close()
579	return
580
581def tcm_mod_add_kbuild(tcm_dir, fabric_mod_name):
582	buf = "obj-$(CONFIG_" + fabric_mod_name.upper() + ")	+= " + fabric_mod_name.lower() + "/\n"
583	kbuild = tcm_dir + "/drivers/target/Makefile"
584
585	f = open(kbuild, 'a')
586	f.write(buf)
587	f.close()
588	return
589
590def tcm_mod_add_kconfig(tcm_dir, fabric_mod_name):
591	buf = "source \"drivers/target/" + fabric_mod_name.lower() + "/Kconfig\"\n"
592	kconfig = tcm_dir + "/drivers/target/Kconfig"
593
594	f = open(kconfig, 'a')
595	f.write(buf)
596	f.close()
597	return
598
599def main(modname, proto_ident):
600#	proto_ident = "FC"
601#	proto_ident = "SAS"
602#	proto_ident = "iSCSI"
603
604	tcm_dir = os.getcwd();
605	tcm_dir += "/../../"
606	print "tcm_dir: " + tcm_dir
607	fabric_mod_name = modname
608	fabric_mod_dir = tcm_dir + "drivers/target/" + fabric_mod_name
609	print "Set fabric_mod_name: " + fabric_mod_name
610	print "Set fabric_mod_dir: " + fabric_mod_dir
611	print "Using proto_ident: " + proto_ident
612
613	if proto_ident != "FC" and proto_ident != "SAS" and proto_ident != "iSCSI":
614		print "Unsupported proto_ident: " + proto_ident
615		sys.exit(1)
616
617	ret = tcm_mod_create_module_subdir(fabric_mod_dir)
618	if ret:
619		print "tcm_mod_create_module_subdir() failed because module already exists!"
620		sys.exit(1)
621
622	tcm_mod_build_base_includes(proto_ident, fabric_mod_dir, fabric_mod_name)
623	tcm_mod_scan_fabric_ops(tcm_dir)
624	tcm_mod_dump_fabric_ops(proto_ident, fabric_mod_dir, fabric_mod_name)
625	tcm_mod_build_configfs(proto_ident, fabric_mod_dir, fabric_mod_name)
626	tcm_mod_build_kbuild(fabric_mod_dir, fabric_mod_name)
627	tcm_mod_build_kconfig(fabric_mod_dir, fabric_mod_name)
628
629	input = raw_input("Would you like to add " + fabric_mod_name + " to drivers/target/Makefile..? [yes,no]: ")
630	if input == "yes" or input == "y":
631		tcm_mod_add_kbuild(tcm_dir, fabric_mod_name)
632
633	input = raw_input("Would you like to add " + fabric_mod_name + " to drivers/target/Kconfig..? [yes,no]: ")
634	if input == "yes" or input == "y":
635		tcm_mod_add_kconfig(tcm_dir, fabric_mod_name)
636
637	return
638
639parser = optparse.OptionParser()
640parser.add_option('-m', '--modulename', help='Module name', dest='modname',
641		action='store', nargs=1, type='string')
642parser.add_option('-p', '--protoident', help='Protocol Ident', dest='protoident',
643		action='store', nargs=1, type='string')
644
645(opts, args) = parser.parse_args()
646
647mandatories = ['modname', 'protoident']
648for m in mandatories:
649	if not opts.__dict__[m]:
650		print "mandatory option is missing\n"
651		parser.print_help()
652		exit(-1)
653
654if __name__ == "__main__":
655
656	main(str(opts.modname), opts.protoident)
657