[SCSI] iscsi: add target discvery event to transport class
[pandora-kernel.git] / drivers / scsi / scsi_transport_iscsi.c
1 /*
2  * iSCSI transport class definitions
3  *
4  * Copyright (C) IBM Corporation, 2004
5  * Copyright (C) Mike Christie, 2004 - 2005
6  * Copyright (C) Dmitry Yusupov, 2004 - 2005
7  * Copyright (C) Alex Aizman, 2004 - 2005
8  *
9  * This program is free software; you can redistribute it and/or modify
10  * it under the terms of the GNU General Public License as published by
11  * the Free Software Foundation; either version 2 of the License, or
12  * (at your option) any later version.
13  *
14  * This program is distributed in the hope that it will be useful,
15  * but WITHOUT ANY WARRANTY; without even the implied warranty of
16  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17  * GNU General Public License for more details.
18  *
19  * You should have received a copy of the GNU General Public License
20  * along with this program; if not, write to the Free Software
21  * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
22  */
23 #include <linux/module.h>
24 #include <linux/mempool.h>
25 #include <linux/mutex.h>
26 #include <net/tcp.h>
27 #include <scsi/scsi.h>
28 #include <scsi/scsi_host.h>
29 #include <scsi/scsi_device.h>
30 #include <scsi/scsi_transport.h>
31 #include <scsi/scsi_transport_iscsi.h>
32 #include <scsi/iscsi_if.h>
33
34 #define ISCSI_SESSION_ATTRS 11
35 #define ISCSI_CONN_ATTRS 11
36 #define ISCSI_HOST_ATTRS 0
37
38 struct iscsi_internal {
39         int daemon_pid;
40         struct scsi_transport_template t;
41         struct iscsi_transport *iscsi_transport;
42         struct list_head list;
43         struct class_device cdev;
44
45         struct class_device_attribute *host_attrs[ISCSI_HOST_ATTRS + 1];
46         struct transport_container conn_cont;
47         struct class_device_attribute *conn_attrs[ISCSI_CONN_ATTRS + 1];
48         struct transport_container session_cont;
49         struct class_device_attribute *session_attrs[ISCSI_SESSION_ATTRS + 1];
50 };
51
52 static int iscsi_session_nr;    /* sysfs session id for next new session */
53
54 /*
55  * list of registered transports and lock that must
56  * be held while accessing list. The iscsi_transport_lock must
57  * be acquired after the rx_queue_mutex.
58  */
59 static LIST_HEAD(iscsi_transports);
60 static DEFINE_SPINLOCK(iscsi_transport_lock);
61
62 #define to_iscsi_internal(tmpl) \
63         container_of(tmpl, struct iscsi_internal, t)
64
65 #define cdev_to_iscsi_internal(_cdev) \
66         container_of(_cdev, struct iscsi_internal, cdev)
67
68 static void iscsi_transport_release(struct class_device *cdev)
69 {
70         struct iscsi_internal *priv = cdev_to_iscsi_internal(cdev);
71         kfree(priv);
72 }
73
74 /*
75  * iscsi_transport_class represents the iscsi_transports that are
76  * registered.
77  */
78 static struct class iscsi_transport_class = {
79         .name = "iscsi_transport",
80         .release = iscsi_transport_release,
81 };
82
83 static ssize_t
84 show_transport_handle(struct class_device *cdev, char *buf)
85 {
86         struct iscsi_internal *priv = cdev_to_iscsi_internal(cdev);
87         return sprintf(buf, "%llu\n", (unsigned long long)iscsi_handle(priv->iscsi_transport));
88 }
89 static CLASS_DEVICE_ATTR(handle, S_IRUGO, show_transport_handle, NULL);
90
91 #define show_transport_attr(name, format)                               \
92 static ssize_t                                                          \
93 show_transport_##name(struct class_device *cdev, char *buf)             \
94 {                                                                       \
95         struct iscsi_internal *priv = cdev_to_iscsi_internal(cdev);     \
96         return sprintf(buf, format"\n", priv->iscsi_transport->name);   \
97 }                                                                       \
98 static CLASS_DEVICE_ATTR(name, S_IRUGO, show_transport_##name, NULL);
99
100 show_transport_attr(caps, "0x%x");
101 show_transport_attr(max_lun, "%d");
102 show_transport_attr(max_conn, "%d");
103 show_transport_attr(max_cmd_len, "%d");
104
105 static struct attribute *iscsi_transport_attrs[] = {
106         &class_device_attr_handle.attr,
107         &class_device_attr_caps.attr,
108         &class_device_attr_max_lun.attr,
109         &class_device_attr_max_conn.attr,
110         &class_device_attr_max_cmd_len.attr,
111         NULL,
112 };
113
114 static struct attribute_group iscsi_transport_group = {
115         .attrs = iscsi_transport_attrs,
116 };
117
118 static int iscsi_setup_host(struct transport_container *tc, struct device *dev,
119                             struct class_device *cdev)
120 {
121         struct Scsi_Host *shost = dev_to_shost(dev);
122         struct iscsi_host *ihost = shost->shost_data;
123
124         memset(ihost, 0, sizeof(*ihost));
125         INIT_LIST_HEAD(&ihost->sessions);
126         mutex_init(&ihost->mutex);
127         return 0;
128 }
129
130 static DECLARE_TRANSPORT_CLASS(iscsi_host_class,
131                                "iscsi_host",
132                                iscsi_setup_host,
133                                NULL,
134                                NULL);
135
136 static DECLARE_TRANSPORT_CLASS(iscsi_session_class,
137                                "iscsi_session",
138                                NULL,
139                                NULL,
140                                NULL);
141
142 static DECLARE_TRANSPORT_CLASS(iscsi_connection_class,
143                                "iscsi_connection",
144                                NULL,
145                                NULL,
146                                NULL);
147
148 static struct sock *nls;
149 static DEFINE_MUTEX(rx_queue_mutex);
150
151 struct mempool_zone {
152         mempool_t *pool;
153         atomic_t allocated;
154         int size;
155         int hiwat;
156         struct list_head freequeue;
157         spinlock_t freelock;
158 };
159
160 static struct mempool_zone *z_reply;
161
162 /*
163  * Z_MAX_* - actual mempool size allocated at the mempool_zone_init() time
164  * Z_HIWAT_* - zone's high watermark when if_error bit will be set to -ENOMEM
165  *             so daemon will notice OOM on NETLINK tranposrt level and will
166  *             be able to predict or change operational behavior
167  */
168 #define Z_MAX_REPLY     8
169 #define Z_HIWAT_REPLY   6
170 #define Z_MAX_PDU       8
171 #define Z_HIWAT_PDU     6
172 #define Z_MAX_ERROR     16
173 #define Z_HIWAT_ERROR   12
174
175 static LIST_HEAD(sesslist);
176 static DEFINE_SPINLOCK(sesslock);
177 static LIST_HEAD(connlist);
178 static DEFINE_SPINLOCK(connlock);
179
180 static uint32_t iscsi_conn_get_sid(struct iscsi_cls_conn *conn)
181 {
182         struct iscsi_cls_session *sess = iscsi_dev_to_session(conn->dev.parent);
183         return sess->sid;
184 }
185
186 /*
187  * Returns the matching session to a given sid
188  */
189 static struct iscsi_cls_session *iscsi_session_lookup(uint32_t sid)
190 {
191         unsigned long flags;
192         struct iscsi_cls_session *sess;
193
194         spin_lock_irqsave(&sesslock, flags);
195         list_for_each_entry(sess, &sesslist, sess_list) {
196                 if (sess->sid == sid) {
197                         spin_unlock_irqrestore(&sesslock, flags);
198                         return sess;
199                 }
200         }
201         spin_unlock_irqrestore(&sesslock, flags);
202         return NULL;
203 }
204
205 /*
206  * Returns the matching connection to a given sid / cid tuple
207  */
208 static struct iscsi_cls_conn *iscsi_conn_lookup(uint32_t sid, uint32_t cid)
209 {
210         unsigned long flags;
211         struct iscsi_cls_conn *conn;
212
213         spin_lock_irqsave(&connlock, flags);
214         list_for_each_entry(conn, &connlist, conn_list) {
215                 if ((conn->cid == cid) && (iscsi_conn_get_sid(conn) == sid)) {
216                         spin_unlock_irqrestore(&connlock, flags);
217                         return conn;
218                 }
219         }
220         spin_unlock_irqrestore(&connlock, flags);
221         return NULL;
222 }
223
224 /*
225  * The following functions can be used by LLDs that allocate
226  * their own scsi_hosts or by software iscsi LLDs
227  */
228 static void iscsi_session_release(struct device *dev)
229 {
230         struct iscsi_cls_session *session = iscsi_dev_to_session(dev);
231         struct iscsi_transport *transport = session->transport;
232         struct Scsi_Host *shost;
233
234         shost = iscsi_session_to_shost(session);
235         scsi_host_put(shost);
236         kfree(session->targetname);
237         kfree(session);
238         module_put(transport->owner);
239 }
240
241 static int iscsi_is_session_dev(const struct device *dev)
242 {
243         return dev->release == iscsi_session_release;
244 }
245
246 static int iscsi_user_scan(struct Scsi_Host *shost, uint channel,
247                            uint id, uint lun)
248 {
249         struct iscsi_host *ihost = shost->shost_data;
250         struct iscsi_cls_session *session;
251
252         mutex_lock(&ihost->mutex);
253         list_for_each_entry(session, &ihost->sessions, host_list) {
254                 if ((channel == SCAN_WILD_CARD ||
255                      channel == session->channel) &&
256                     (id == SCAN_WILD_CARD || id == session->target_id))
257                         scsi_scan_target(&session->dev, session->channel,
258                                          session->target_id, lun, 1);
259         }
260         mutex_unlock(&ihost->mutex);
261
262         return 0;
263 }
264
265 static void session_recovery_timedout(void *data)
266 {
267         struct iscsi_cls_session *session = data;
268
269         dev_printk(KERN_INFO, &session->dev, "iscsi: session recovery timed "
270                   "out after %d secs\n", session->recovery_tmo);
271
272         if (session->transport->session_recovery_timedout)
273                 session->transport->session_recovery_timedout(session);
274
275         scsi_target_unblock(&session->dev);
276 }
277
278 void iscsi_unblock_session(struct iscsi_cls_session *session)
279 {
280         if (!cancel_delayed_work(&session->recovery_work))
281                 flush_scheduled_work();
282         scsi_target_unblock(&session->dev);
283 }
284 EXPORT_SYMBOL_GPL(iscsi_unblock_session);
285
286 void iscsi_block_session(struct iscsi_cls_session *session)
287 {
288         scsi_target_block(&session->dev);
289         schedule_delayed_work(&session->recovery_work,
290                              session->recovery_tmo * HZ);
291 }
292 EXPORT_SYMBOL_GPL(iscsi_block_session);
293
294 /**
295  * iscsi_create_session - create iscsi class session
296  * @shost: scsi host
297  * @transport: iscsi transport
298  *
299  * This can be called from a LLD or iscsi_transport.
300  **/
301 struct iscsi_cls_session *
302 iscsi_create_session(struct Scsi_Host *shost,
303                      struct iscsi_transport *transport, int channel)
304 {
305         struct iscsi_host *ihost;
306         struct iscsi_cls_session *session;
307         int err;
308
309         if (!try_module_get(transport->owner))
310                 return NULL;
311
312         session = kzalloc(sizeof(*session) + transport->sessiondata_size,
313                           GFP_KERNEL);
314         if (!session)
315                 goto module_put;
316         session->transport = transport;
317         session->recovery_tmo = 120;
318         INIT_WORK(&session->recovery_work, session_recovery_timedout, session);
319         INIT_LIST_HEAD(&session->host_list);
320         INIT_LIST_HEAD(&session->sess_list);
321
322         if (transport->sessiondata_size)
323                 session->dd_data = &session[1];
324
325         /* this is released in the dev's release function */
326         scsi_host_get(shost);
327         ihost = shost->shost_data;
328
329         session->sid = iscsi_session_nr++;
330         session->channel = channel;
331         session->target_id = ihost->next_target_id++;
332
333         snprintf(session->dev.bus_id, BUS_ID_SIZE, "session%u",
334                  session->sid);
335         session->dev.parent = &shost->shost_gendev;
336         session->dev.release = iscsi_session_release;
337         err = device_register(&session->dev);
338         if (err) {
339                 dev_printk(KERN_ERR, &session->dev, "iscsi: could not "
340                            "register session's dev\n");
341                 goto free_session;
342         }
343         transport_register_device(&session->dev);
344
345         mutex_lock(&ihost->mutex);
346         list_add(&session->host_list, &ihost->sessions);
347         mutex_unlock(&ihost->mutex);
348
349         return session;
350
351 free_session:
352         kfree(session);
353 module_put:
354         module_put(transport->owner);
355         return NULL;
356 }
357
358 EXPORT_SYMBOL_GPL(iscsi_create_session);
359
360 /**
361  * iscsi_destroy_session - destroy iscsi session
362  * @session: iscsi_session
363  *
364  * Can be called by a LLD or iscsi_transport. There must not be
365  * any running connections.
366  **/
367 int iscsi_destroy_session(struct iscsi_cls_session *session)
368 {
369         struct Scsi_Host *shost = iscsi_session_to_shost(session);
370         struct iscsi_host *ihost = shost->shost_data;
371
372         if (!cancel_delayed_work(&session->recovery_work))
373                 flush_scheduled_work();
374
375         mutex_lock(&ihost->mutex);
376         list_del(&session->host_list);
377         mutex_unlock(&ihost->mutex);
378
379         transport_unregister_device(&session->dev);
380         device_unregister(&session->dev);
381         return 0;
382 }
383
384 EXPORT_SYMBOL_GPL(iscsi_destroy_session);
385
386 static void iscsi_conn_release(struct device *dev)
387 {
388         struct iscsi_cls_conn *conn = iscsi_dev_to_conn(dev);
389         struct device *parent = conn->dev.parent;
390
391         kfree(conn->persistent_address);
392         kfree(conn);
393         put_device(parent);
394 }
395
396 static int iscsi_is_conn_dev(const struct device *dev)
397 {
398         return dev->release == iscsi_conn_release;
399 }
400
401 /**
402  * iscsi_create_conn - create iscsi class connection
403  * @session: iscsi cls session
404  * @cid: connection id
405  *
406  * This can be called from a LLD or iscsi_transport. The connection
407  * is child of the session so cid must be unique for all connections
408  * on the session.
409  *
410  * Since we do not support MCS, cid will normally be zero. In some cases
411  * for software iscsi we could be trying to preallocate a connection struct
412  * in which case there could be two connection structs and cid would be
413  * non-zero.
414  **/
415 struct iscsi_cls_conn *
416 iscsi_create_conn(struct iscsi_cls_session *session, uint32_t cid)
417 {
418         struct iscsi_transport *transport = session->transport;
419         struct iscsi_cls_conn *conn;
420         int err;
421
422         conn = kzalloc(sizeof(*conn) + transport->conndata_size, GFP_KERNEL);
423         if (!conn)
424                 return NULL;
425
426         if (transport->conndata_size)
427                 conn->dd_data = &conn[1];
428
429         INIT_LIST_HEAD(&conn->conn_list);
430         conn->transport = transport;
431         conn->cid = cid;
432
433         /* this is released in the dev's release function */
434         if (!get_device(&session->dev))
435                 goto free_conn;
436
437         snprintf(conn->dev.bus_id, BUS_ID_SIZE, "connection%d:%u",
438                  session->sid, cid);
439         conn->dev.parent = &session->dev;
440         conn->dev.release = iscsi_conn_release;
441         err = device_register(&conn->dev);
442         if (err) {
443                 dev_printk(KERN_ERR, &conn->dev, "iscsi: could not register "
444                            "connection's dev\n");
445                 goto release_parent_ref;
446         }
447         transport_register_device(&conn->dev);
448         return conn;
449
450 release_parent_ref:
451         put_device(&session->dev);
452 free_conn:
453         kfree(conn);
454         return NULL;
455 }
456
457 EXPORT_SYMBOL_GPL(iscsi_create_conn);
458
459 /**
460  * iscsi_destroy_conn - destroy iscsi class connection
461  * @session: iscsi cls session
462  *
463  * This can be called from a LLD or iscsi_transport.
464  **/
465 int iscsi_destroy_conn(struct iscsi_cls_conn *conn)
466 {
467         transport_unregister_device(&conn->dev);
468         device_unregister(&conn->dev);
469         return 0;
470 }
471
472 EXPORT_SYMBOL_GPL(iscsi_destroy_conn);
473
474 /*
475  * iscsi interface functions
476  */
477 static struct iscsi_internal *
478 iscsi_if_transport_lookup(struct iscsi_transport *tt)
479 {
480         struct iscsi_internal *priv;
481         unsigned long flags;
482
483         spin_lock_irqsave(&iscsi_transport_lock, flags);
484         list_for_each_entry(priv, &iscsi_transports, list) {
485                 if (tt == priv->iscsi_transport) {
486                         spin_unlock_irqrestore(&iscsi_transport_lock, flags);
487                         return priv;
488                 }
489         }
490         spin_unlock_irqrestore(&iscsi_transport_lock, flags);
491         return NULL;
492 }
493
494 static inline struct list_head *skb_to_lh(struct sk_buff *skb)
495 {
496         return (struct list_head *)&skb->cb;
497 }
498
499 static void*
500 mempool_zone_alloc_skb(gfp_t gfp_mask, void *pool_data)
501 {
502         struct mempool_zone *zone = pool_data;
503
504         return alloc_skb(zone->size, gfp_mask);
505 }
506
507 static void
508 mempool_zone_free_skb(void *element, void *pool_data)
509 {
510         kfree_skb(element);
511 }
512
513 static void
514 mempool_zone_complete(struct mempool_zone *zone)
515 {
516         unsigned long flags;
517         struct list_head *lh, *n;
518
519         spin_lock_irqsave(&zone->freelock, flags);
520         list_for_each_safe(lh, n, &zone->freequeue) {
521                 struct sk_buff *skb = (struct sk_buff *)((char *)lh -
522                                 offsetof(struct sk_buff, cb));
523                 if (!skb_shared(skb)) {
524                         list_del(skb_to_lh(skb));
525                         mempool_free(skb, zone->pool);
526                         atomic_dec(&zone->allocated);
527                 }
528         }
529         spin_unlock_irqrestore(&zone->freelock, flags);
530 }
531
532 static struct mempool_zone *
533 mempool_zone_init(unsigned max, unsigned size, unsigned hiwat)
534 {
535         struct mempool_zone *zp;
536
537         zp = kzalloc(sizeof(*zp), GFP_KERNEL);
538         if (!zp)
539                 return NULL;
540
541         zp->size = size;
542         zp->hiwat = hiwat;
543         INIT_LIST_HEAD(&zp->freequeue);
544         spin_lock_init(&zp->freelock);
545         atomic_set(&zp->allocated, 0);
546
547         zp->pool = mempool_create(max, mempool_zone_alloc_skb,
548                                   mempool_zone_free_skb, zp);
549         if (!zp->pool) {
550                 kfree(zp);
551                 return NULL;
552         }
553
554         return zp;
555 }
556
557 static void mempool_zone_destroy(struct mempool_zone *zp)
558 {
559         mempool_destroy(zp->pool);
560         kfree(zp);
561 }
562
563 static struct sk_buff*
564 mempool_zone_get_skb(struct mempool_zone *zone)
565 {
566         struct sk_buff *skb;
567
568         skb = mempool_alloc(zone->pool, GFP_ATOMIC);
569         if (skb)
570                 atomic_inc(&zone->allocated);
571         return skb;
572 }
573
574 static int
575 iscsi_unicast_skb(struct mempool_zone *zone, struct sk_buff *skb, int pid)
576 {
577         unsigned long flags;
578         int rc;
579
580         skb_get(skb);
581         rc = netlink_unicast(nls, skb, pid, MSG_DONTWAIT);
582         if (rc < 0) {
583                 mempool_free(skb, zone->pool);
584                 printk(KERN_ERR "iscsi: can not unicast skb (%d)\n", rc);
585                 return rc;
586         }
587
588         spin_lock_irqsave(&zone->freelock, flags);
589         INIT_LIST_HEAD(skb_to_lh(skb));
590         list_add(skb_to_lh(skb), &zone->freequeue);
591         spin_unlock_irqrestore(&zone->freelock, flags);
592
593         return 0;
594 }
595
596 int iscsi_recv_pdu(struct iscsi_cls_conn *conn, struct iscsi_hdr *hdr,
597                    char *data, uint32_t data_size)
598 {
599         struct nlmsghdr *nlh;
600         struct sk_buff *skb;
601         struct iscsi_uevent *ev;
602         char *pdu;
603         struct iscsi_internal *priv;
604         int len = NLMSG_SPACE(sizeof(*ev) + sizeof(struct iscsi_hdr) +
605                               data_size);
606
607         priv = iscsi_if_transport_lookup(conn->transport);
608         if (!priv)
609                 return -EINVAL;
610
611         mempool_zone_complete(conn->z_pdu);
612
613         skb = mempool_zone_get_skb(conn->z_pdu);
614         if (!skb) {
615                 iscsi_conn_error(conn, ISCSI_ERR_CONN_FAILED);
616                 dev_printk(KERN_ERR, &conn->dev, "iscsi: can not deliver "
617                            "control PDU: OOM\n");
618                 return -ENOMEM;
619         }
620
621         nlh = __nlmsg_put(skb, priv->daemon_pid, 0, 0, (len - sizeof(*nlh)), 0);
622         ev = NLMSG_DATA(nlh);
623         memset(ev, 0, sizeof(*ev));
624         ev->transport_handle = iscsi_handle(conn->transport);
625         ev->type = ISCSI_KEVENT_RECV_PDU;
626         if (atomic_read(&conn->z_pdu->allocated) >= conn->z_pdu->hiwat)
627                 ev->iferror = -ENOMEM;
628         ev->r.recv_req.cid = conn->cid;
629         ev->r.recv_req.sid = iscsi_conn_get_sid(conn);
630         pdu = (char*)ev + sizeof(*ev);
631         memcpy(pdu, hdr, sizeof(struct iscsi_hdr));
632         memcpy(pdu + sizeof(struct iscsi_hdr), data, data_size);
633
634         return iscsi_unicast_skb(conn->z_pdu, skb, priv->daemon_pid);
635 }
636 EXPORT_SYMBOL_GPL(iscsi_recv_pdu);
637
638 void iscsi_conn_error(struct iscsi_cls_conn *conn, enum iscsi_err error)
639 {
640         struct nlmsghdr *nlh;
641         struct sk_buff  *skb;
642         struct iscsi_uevent *ev;
643         struct iscsi_internal *priv;
644         int len = NLMSG_SPACE(sizeof(*ev));
645
646         priv = iscsi_if_transport_lookup(conn->transport);
647         if (!priv)
648                 return;
649
650         mempool_zone_complete(conn->z_error);
651
652         skb = mempool_zone_get_skb(conn->z_error);
653         if (!skb) {
654                 dev_printk(KERN_ERR, &conn->dev, "iscsi: gracefully ignored "
655                           "conn error (%d)\n", error);
656                 return;
657         }
658
659         nlh = __nlmsg_put(skb, priv->daemon_pid, 0, 0, (len - sizeof(*nlh)), 0);
660         ev = NLMSG_DATA(nlh);
661         ev->transport_handle = iscsi_handle(conn->transport);
662         ev->type = ISCSI_KEVENT_CONN_ERROR;
663         if (atomic_read(&conn->z_error->allocated) >= conn->z_error->hiwat)
664                 ev->iferror = -ENOMEM;
665         ev->r.connerror.error = error;
666         ev->r.connerror.cid = conn->cid;
667         ev->r.connerror.sid = iscsi_conn_get_sid(conn);
668
669         iscsi_unicast_skb(conn->z_error, skb, priv->daemon_pid);
670
671         dev_printk(KERN_INFO, &conn->dev, "iscsi: detected conn error (%d)\n",
672                    error);
673 }
674 EXPORT_SYMBOL_GPL(iscsi_conn_error);
675
676 static int
677 iscsi_if_send_reply(int pid, int seq, int type, int done, int multi,
678                       void *payload, int size)
679 {
680         struct sk_buff  *skb;
681         struct nlmsghdr *nlh;
682         int len = NLMSG_SPACE(size);
683         int flags = multi ? NLM_F_MULTI : 0;
684         int t = done ? NLMSG_DONE : type;
685
686         mempool_zone_complete(z_reply);
687
688         skb = mempool_zone_get_skb(z_reply);
689         /*
690          * FIXME:
691          * user is supposed to react on iferror == -ENOMEM;
692          * see iscsi_if_rx().
693          */
694         BUG_ON(!skb);
695
696         nlh = __nlmsg_put(skb, pid, seq, t, (len - sizeof(*nlh)), 0);
697         nlh->nlmsg_flags = flags;
698         memcpy(NLMSG_DATA(nlh), payload, size);
699         return iscsi_unicast_skb(z_reply, skb, pid);
700 }
701
702 static int
703 iscsi_if_get_stats(struct iscsi_transport *transport, struct nlmsghdr *nlh)
704 {
705         struct iscsi_uevent *ev = NLMSG_DATA(nlh);
706         struct iscsi_stats *stats;
707         struct sk_buff *skbstat;
708         struct iscsi_cls_conn *conn;
709         struct nlmsghdr *nlhstat;
710         struct iscsi_uevent *evstat;
711         struct iscsi_internal *priv;
712         int len = NLMSG_SPACE(sizeof(*ev) +
713                               sizeof(struct iscsi_stats) +
714                               sizeof(struct iscsi_stats_custom) *
715                               ISCSI_STATS_CUSTOM_MAX);
716         int err = 0;
717
718         priv = iscsi_if_transport_lookup(transport);
719         if (!priv)
720                 return -EINVAL;
721
722         conn = iscsi_conn_lookup(ev->u.get_stats.sid, ev->u.get_stats.cid);
723         if (!conn)
724                 return -EEXIST;
725
726         do {
727                 int actual_size;
728
729                 mempool_zone_complete(conn->z_pdu);
730
731                 skbstat = mempool_zone_get_skb(conn->z_pdu);
732                 if (!skbstat) {
733                         dev_printk(KERN_ERR, &conn->dev, "iscsi: can not "
734                                    "deliver stats: OOM\n");
735                         return -ENOMEM;
736                 }
737
738                 nlhstat = __nlmsg_put(skbstat, priv->daemon_pid, 0, 0,
739                                       (len - sizeof(*nlhstat)), 0);
740                 evstat = NLMSG_DATA(nlhstat);
741                 memset(evstat, 0, sizeof(*evstat));
742                 evstat->transport_handle = iscsi_handle(conn->transport);
743                 evstat->type = nlh->nlmsg_type;
744                 if (atomic_read(&conn->z_pdu->allocated) >= conn->z_pdu->hiwat)
745                         evstat->iferror = -ENOMEM;
746                 evstat->u.get_stats.cid =
747                         ev->u.get_stats.cid;
748                 evstat->u.get_stats.sid =
749                         ev->u.get_stats.sid;
750                 stats = (struct iscsi_stats *)
751                         ((char*)evstat + sizeof(*evstat));
752                 memset(stats, 0, sizeof(*stats));
753
754                 transport->get_stats(conn, stats);
755                 actual_size = NLMSG_SPACE(sizeof(struct iscsi_uevent) +
756                                           sizeof(struct iscsi_stats) +
757                                           sizeof(struct iscsi_stats_custom) *
758                                           stats->custom_length);
759                 actual_size -= sizeof(*nlhstat);
760                 actual_size = NLMSG_LENGTH(actual_size);
761                 skb_trim(skbstat, NLMSG_ALIGN(actual_size));
762                 nlhstat->nlmsg_len = actual_size;
763
764                 err = iscsi_unicast_skb(conn->z_pdu, skbstat, priv->daemon_pid);
765         } while (err < 0 && err != -ECONNREFUSED);
766
767         return err;
768 }
769
770 static int
771 iscsi_if_create_session(struct iscsi_internal *priv, struct iscsi_uevent *ev)
772 {
773         struct iscsi_transport *transport = priv->iscsi_transport;
774         struct iscsi_cls_session *session;
775         unsigned long flags;
776         uint32_t hostno;
777
778         session = transport->create_session(transport, &priv->t,
779                                             ev->u.c_session.initial_cmdsn,
780                                             &hostno);
781         if (!session)
782                 return -ENOMEM;
783
784         spin_lock_irqsave(&sesslock, flags);
785         list_add(&session->sess_list, &sesslist);
786         spin_unlock_irqrestore(&sesslock, flags);
787
788         ev->r.c_session_ret.host_no = hostno;
789         ev->r.c_session_ret.sid = session->sid;
790         return 0;
791 }
792
793 static int
794 iscsi_if_create_conn(struct iscsi_transport *transport, struct iscsi_uevent *ev)
795 {
796         struct iscsi_cls_conn *conn;
797         struct iscsi_cls_session *session;
798         unsigned long flags;
799
800         session = iscsi_session_lookup(ev->u.c_conn.sid);
801         if (!session) {
802                 printk(KERN_ERR "iscsi: invalid session %d\n",
803                        ev->u.c_conn.sid);
804                 return -EINVAL;
805         }
806
807         conn = transport->create_conn(session, ev->u.c_conn.cid);
808         if (!conn) {
809                 printk(KERN_ERR "iscsi: couldn't create a new "
810                            "connection for session %d\n",
811                            session->sid);
812                 return -ENOMEM;
813         }
814
815         conn->z_pdu = mempool_zone_init(Z_MAX_PDU,
816                         NLMSG_SPACE(sizeof(struct iscsi_uevent) +
817                                     sizeof(struct iscsi_hdr) +
818                                     DEFAULT_MAX_RECV_DATA_SEGMENT_LENGTH),
819                         Z_HIWAT_PDU);
820         if (!conn->z_pdu) {
821                 dev_printk(KERN_ERR, &conn->dev, "iscsi: can not allocate "
822                            "pdu zone for new conn\n");
823                 goto destroy_conn;
824         }
825
826         conn->z_error = mempool_zone_init(Z_MAX_ERROR,
827                         NLMSG_SPACE(sizeof(struct iscsi_uevent)),
828                         Z_HIWAT_ERROR);
829         if (!conn->z_error) {
830                 dev_printk(KERN_ERR, &conn->dev, "iscsi: can not allocate "
831                            "error zone for new conn\n");
832                 goto free_pdu_pool;
833         }
834
835         ev->r.c_conn_ret.sid = session->sid;
836         ev->r.c_conn_ret.cid = conn->cid;
837
838         spin_lock_irqsave(&connlock, flags);
839         list_add(&conn->conn_list, &connlist);
840         conn->active = 1;
841         spin_unlock_irqrestore(&connlock, flags);
842
843         return 0;
844
845 free_pdu_pool:
846         mempool_zone_destroy(conn->z_pdu);
847 destroy_conn:
848         if (transport->destroy_conn)
849                 transport->destroy_conn(conn->dd_data);
850         return -ENOMEM;
851 }
852
853 static int
854 iscsi_if_destroy_conn(struct iscsi_transport *transport, struct iscsi_uevent *ev)
855 {
856         unsigned long flags;
857         struct iscsi_cls_conn *conn;
858         struct mempool_zone *z_error, *z_pdu;
859
860         conn = iscsi_conn_lookup(ev->u.d_conn.sid, ev->u.d_conn.cid);
861         if (!conn)
862                 return -EINVAL;
863         spin_lock_irqsave(&connlock, flags);
864         conn->active = 0;
865         list_del(&conn->conn_list);
866         spin_unlock_irqrestore(&connlock, flags);
867
868         z_pdu = conn->z_pdu;
869         z_error = conn->z_error;
870
871         if (transport->destroy_conn)
872                 transport->destroy_conn(conn);
873
874         mempool_zone_destroy(z_pdu);
875         mempool_zone_destroy(z_error);
876
877         return 0;
878 }
879
880 static void
881 iscsi_copy_param(struct iscsi_uevent *ev, uint32_t *value, char *data)
882 {
883         if (ev->u.set_param.len != sizeof(uint32_t))
884                 BUG();
885         memcpy(value, data, min_t(uint32_t, sizeof(uint32_t),
886                 ev->u.set_param.len));
887 }
888
889 static int
890 iscsi_set_param(struct iscsi_transport *transport, struct iscsi_uevent *ev)
891 {
892         char *data = (char*)ev + sizeof(*ev);
893         struct iscsi_cls_conn *conn;
894         struct iscsi_cls_session *session;
895         int err = 0;
896         uint32_t value = 0;
897
898         session = iscsi_session_lookup(ev->u.set_param.sid);
899         conn = iscsi_conn_lookup(ev->u.set_param.sid, ev->u.set_param.cid);
900         if (!conn || !session)
901                 return -EINVAL;
902
903         switch (ev->u.set_param.param) {
904         case ISCSI_PARAM_SESS_RECOVERY_TMO:
905                 iscsi_copy_param(ev, &value, data);
906                 if (value != 0)
907                         session->recovery_tmo = value;
908                 break;
909         case ISCSI_PARAM_TARGET_NAME:
910                 /* this should not change between logins */
911                 if (session->targetname)
912                         return 0;
913
914                 session->targetname = kstrdup(data, GFP_KERNEL);
915                 if (!session->targetname)
916                         return -ENOMEM;
917                 break;
918         case ISCSI_PARAM_TPGT:
919                 iscsi_copy_param(ev, &value, data);
920                 session->tpgt = value;
921                 break;
922         case ISCSI_PARAM_PERSISTENT_PORT:
923                 iscsi_copy_param(ev, &value, data);
924                 conn->persistent_port = value;
925                 break;
926         case ISCSI_PARAM_PERSISTENT_ADDRESS:
927                 /*
928                  * this is the address returned in discovery so it should
929                  * not change between logins.
930                  */
931                 if (conn->persistent_address)
932                         return 0;
933
934                 conn->persistent_address = kstrdup(data, GFP_KERNEL);
935                 if (!conn->persistent_address)
936                         return -ENOMEM;
937                 break;
938         default:
939                 iscsi_copy_param(ev, &value, data);
940                 err = transport->set_param(conn, ev->u.set_param.param, value);
941         }
942
943         return err;
944 }
945
946 static int
947 iscsi_if_transport_ep(struct iscsi_transport *transport,
948                       struct iscsi_uevent *ev, int msg_type)
949 {
950         struct sockaddr *dst_addr;
951         int rc = 0;
952
953         switch (msg_type) {
954         case ISCSI_UEVENT_TRANSPORT_EP_CONNECT:
955                 if (!transport->ep_connect)
956                         return -EINVAL;
957
958                 dst_addr = (struct sockaddr *)((char*)ev + sizeof(*ev));
959                 rc = transport->ep_connect(dst_addr,
960                                            ev->u.ep_connect.non_blocking,
961                                            &ev->r.ep_connect_ret.handle);
962                 break;
963         case ISCSI_UEVENT_TRANSPORT_EP_POLL:
964                 if (!transport->ep_poll)
965                         return -EINVAL;
966
967                 ev->r.retcode = transport->ep_poll(ev->u.ep_poll.ep_handle,
968                                                    ev->u.ep_poll.timeout_ms);
969                 break;
970         case ISCSI_UEVENT_TRANSPORT_EP_DISCONNECT:
971                 if (!transport->ep_disconnect)
972                         return -EINVAL;
973
974                 transport->ep_disconnect(ev->u.ep_disconnect.ep_handle);
975                 break;
976         }
977         return rc;
978 }
979
980 static int
981 iscsi_tgt_dscvr(struct iscsi_transport *transport,
982                 struct iscsi_uevent *ev)
983 {
984         struct sockaddr *dst_addr;
985
986         if (!transport->tgt_dscvr)
987                 return -EINVAL;
988
989         dst_addr = (struct sockaddr *)((char*)ev + sizeof(*ev));
990         return transport->tgt_dscvr(ev->u.tgt_dscvr.type,
991                                     ev->u.tgt_dscvr.host_no,
992                                     ev->u.tgt_dscvr.enable, dst_addr);
993 }
994
995 static int
996 iscsi_if_recv_msg(struct sk_buff *skb, struct nlmsghdr *nlh)
997 {
998         int err = 0;
999         struct iscsi_uevent *ev = NLMSG_DATA(nlh);
1000         struct iscsi_transport *transport = NULL;
1001         struct iscsi_internal *priv;
1002         struct iscsi_cls_session *session;
1003         struct iscsi_cls_conn *conn;
1004         unsigned long flags;
1005
1006         priv = iscsi_if_transport_lookup(iscsi_ptr(ev->transport_handle));
1007         if (!priv)
1008                 return -EINVAL;
1009         transport = priv->iscsi_transport;
1010
1011         if (!try_module_get(transport->owner))
1012                 return -EINVAL;
1013
1014         priv->daemon_pid = NETLINK_CREDS(skb)->pid;
1015
1016         switch (nlh->nlmsg_type) {
1017         case ISCSI_UEVENT_CREATE_SESSION:
1018                 err = iscsi_if_create_session(priv, ev);
1019                 break;
1020         case ISCSI_UEVENT_DESTROY_SESSION:
1021                 session = iscsi_session_lookup(ev->u.d_session.sid);
1022                 if (session) {
1023                         spin_lock_irqsave(&sesslock, flags);
1024                         list_del(&session->sess_list);
1025                         spin_unlock_irqrestore(&sesslock, flags);
1026
1027                         transport->destroy_session(session);
1028                 } else
1029                         err = -EINVAL;
1030                 break;
1031         case ISCSI_UEVENT_CREATE_CONN:
1032                 err = iscsi_if_create_conn(transport, ev);
1033                 break;
1034         case ISCSI_UEVENT_DESTROY_CONN:
1035                 err = iscsi_if_destroy_conn(transport, ev);
1036                 break;
1037         case ISCSI_UEVENT_BIND_CONN:
1038                 session = iscsi_session_lookup(ev->u.b_conn.sid);
1039                 conn = iscsi_conn_lookup(ev->u.b_conn.sid, ev->u.b_conn.cid);
1040
1041                 if (session && conn)
1042                         ev->r.retcode = transport->bind_conn(session, conn,
1043                                         ev->u.b_conn.transport_eph,
1044                                         ev->u.b_conn.is_leading);
1045                 else
1046                         err = -EINVAL;
1047                 break;
1048         case ISCSI_UEVENT_SET_PARAM:
1049                 err = iscsi_set_param(transport, ev);
1050                 break;
1051         case ISCSI_UEVENT_START_CONN:
1052                 conn = iscsi_conn_lookup(ev->u.start_conn.sid, ev->u.start_conn.cid);
1053                 if (conn)
1054                         ev->r.retcode = transport->start_conn(conn);
1055                 else
1056                         err = -EINVAL;
1057                 break;
1058         case ISCSI_UEVENT_STOP_CONN:
1059                 conn = iscsi_conn_lookup(ev->u.stop_conn.sid, ev->u.stop_conn.cid);
1060                 if (conn)
1061                         transport->stop_conn(conn, ev->u.stop_conn.flag);
1062                 else
1063                         err = -EINVAL;
1064                 break;
1065         case ISCSI_UEVENT_SEND_PDU:
1066                 conn = iscsi_conn_lookup(ev->u.send_pdu.sid, ev->u.send_pdu.cid);
1067                 if (conn)
1068                         ev->r.retcode = transport->send_pdu(conn,
1069                                 (struct iscsi_hdr*)((char*)ev + sizeof(*ev)),
1070                                 (char*)ev + sizeof(*ev) + ev->u.send_pdu.hdr_size,
1071                                 ev->u.send_pdu.data_size);
1072                 else
1073                         err = -EINVAL;
1074                 break;
1075         case ISCSI_UEVENT_GET_STATS:
1076                 err = iscsi_if_get_stats(transport, nlh);
1077                 break;
1078         case ISCSI_UEVENT_TRANSPORT_EP_CONNECT:
1079         case ISCSI_UEVENT_TRANSPORT_EP_POLL:
1080         case ISCSI_UEVENT_TRANSPORT_EP_DISCONNECT:
1081                 err = iscsi_if_transport_ep(transport, ev, nlh->nlmsg_type);
1082                 break;
1083         case ISCSI_UEVENT_TGT_DSCVR:
1084                 err = iscsi_tgt_dscvr(transport, ev);
1085                 break;
1086         default:
1087                 err = -EINVAL;
1088                 break;
1089         }
1090
1091         module_put(transport->owner);
1092         return err;
1093 }
1094
1095 /*
1096  * Get message from skb (based on rtnetlink_rcv_skb).  Each message is
1097  * processed by iscsi_if_recv_msg.  Malformed skbs with wrong lengths or
1098  * invalid creds are discarded silently.
1099  */
1100 static void
1101 iscsi_if_rx(struct sock *sk, int len)
1102 {
1103         struct sk_buff *skb;
1104
1105         mutex_lock(&rx_queue_mutex);
1106         while ((skb = skb_dequeue(&sk->sk_receive_queue)) != NULL) {
1107                 if (NETLINK_CREDS(skb)->uid) {
1108                         skb_pull(skb, skb->len);
1109                         goto free_skb;
1110                 }
1111
1112                 while (skb->len >= NLMSG_SPACE(0)) {
1113                         int err;
1114                         uint32_t rlen;
1115                         struct nlmsghdr *nlh;
1116                         struct iscsi_uevent *ev;
1117
1118                         nlh = (struct nlmsghdr *)skb->data;
1119                         if (nlh->nlmsg_len < sizeof(*nlh) ||
1120                             skb->len < nlh->nlmsg_len) {
1121                                 break;
1122                         }
1123
1124                         ev = NLMSG_DATA(nlh);
1125                         rlen = NLMSG_ALIGN(nlh->nlmsg_len);
1126                         if (rlen > skb->len)
1127                                 rlen = skb->len;
1128
1129                         err = iscsi_if_recv_msg(skb, nlh);
1130                         if (err) {
1131                                 ev->type = ISCSI_KEVENT_IF_ERROR;
1132                                 ev->iferror = err;
1133                         }
1134                         do {
1135                                 /*
1136                                  * special case for GET_STATS:
1137                                  * on success - sending reply and stats from
1138                                  * inside of if_recv_msg(),
1139                                  * on error - fall through.
1140                                  */
1141                                 if (ev->type == ISCSI_UEVENT_GET_STATS && !err)
1142                                         break;
1143                                 err = iscsi_if_send_reply(
1144                                         NETLINK_CREDS(skb)->pid, nlh->nlmsg_seq,
1145                                         nlh->nlmsg_type, 0, 0, ev, sizeof(*ev));
1146                                 if (atomic_read(&z_reply->allocated) >=
1147                                                 z_reply->hiwat)
1148                                         ev->iferror = -ENOMEM;
1149                         } while (err < 0 && err != -ECONNREFUSED);
1150                         skb_pull(skb, rlen);
1151                 }
1152 free_skb:
1153                 kfree_skb(skb);
1154         }
1155         mutex_unlock(&rx_queue_mutex);
1156 }
1157
1158 #define iscsi_cdev_to_conn(_cdev) \
1159         iscsi_dev_to_conn(_cdev->dev)
1160
1161 #define ISCSI_CLASS_ATTR(_prefix,_name,_mode,_show,_store)              \
1162 struct class_device_attribute class_device_attr_##_prefix##_##_name =   \
1163         __ATTR(_name,_mode,_show,_store)
1164
1165 /*
1166  * iSCSI connection attrs
1167  */
1168 #define iscsi_conn_int_attr_show(param, format)                         \
1169 static ssize_t                                                          \
1170 show_conn_int_param_##param(struct class_device *cdev, char *buf)       \
1171 {                                                                       \
1172         uint32_t value = 0;                                             \
1173         struct iscsi_cls_conn *conn = iscsi_cdev_to_conn(cdev);         \
1174         struct iscsi_transport *t = conn->transport;                    \
1175                                                                         \
1176         t->get_conn_param(conn, param, &value);                         \
1177         return snprintf(buf, 20, format"\n", value);                    \
1178 }
1179
1180 #define iscsi_conn_int_attr(field, param, format)                       \
1181         iscsi_conn_int_attr_show(param, format)                         \
1182 static ISCSI_CLASS_ATTR(conn, field, S_IRUGO, show_conn_int_param_##param, \
1183                         NULL);
1184
1185 iscsi_conn_int_attr(max_recv_dlength, ISCSI_PARAM_MAX_RECV_DLENGTH, "%u");
1186 iscsi_conn_int_attr(max_xmit_dlength, ISCSI_PARAM_MAX_XMIT_DLENGTH, "%u");
1187 iscsi_conn_int_attr(header_digest, ISCSI_PARAM_HDRDGST_EN, "%d");
1188 iscsi_conn_int_attr(data_digest, ISCSI_PARAM_DATADGST_EN, "%d");
1189 iscsi_conn_int_attr(ifmarker, ISCSI_PARAM_IFMARKER_EN, "%d");
1190 iscsi_conn_int_attr(ofmarker, ISCSI_PARAM_OFMARKER_EN, "%d");
1191 iscsi_conn_int_attr(persistent_port, ISCSI_PARAM_PERSISTENT_PORT, "%d");
1192 iscsi_conn_int_attr(port, ISCSI_PARAM_CONN_PORT, "%d");
1193 iscsi_conn_int_attr(exp_statsn, ISCSI_PARAM_EXP_STATSN, "%u");
1194
1195 #define iscsi_conn_str_attr_show(param)                                 \
1196 static ssize_t                                                          \
1197 show_conn_str_param_##param(struct class_device *cdev, char *buf)       \
1198 {                                                                       \
1199         struct iscsi_cls_conn *conn = iscsi_cdev_to_conn(cdev);         \
1200         struct iscsi_transport *t = conn->transport;                    \
1201         return t->get_conn_str_param(conn, param, buf);                 \
1202 }
1203
1204 #define iscsi_conn_str_attr(field, param)                               \
1205         iscsi_conn_str_attr_show(param)                                 \
1206 static ISCSI_CLASS_ATTR(conn, field, S_IRUGO, show_conn_str_param_##param, \
1207                         NULL);
1208
1209 iscsi_conn_str_attr(persistent_address, ISCSI_PARAM_PERSISTENT_ADDRESS);
1210 iscsi_conn_str_attr(address, ISCSI_PARAM_CONN_ADDRESS);
1211
1212 #define iscsi_cdev_to_session(_cdev) \
1213         iscsi_dev_to_session(_cdev->dev)
1214
1215 /*
1216  * iSCSI session attrs
1217  */
1218 #define iscsi_session_int_attr_show(param, format)                      \
1219 static ssize_t                                                          \
1220 show_session_int_param_##param(struct class_device *cdev, char *buf)    \
1221 {                                                                       \
1222         uint32_t value = 0;                                             \
1223         struct iscsi_cls_session *session = iscsi_cdev_to_session(cdev);        \
1224         struct iscsi_transport *t = session->transport;                 \
1225                                                                         \
1226         t->get_session_param(session, param, &value);                   \
1227         return snprintf(buf, 20, format"\n", value);                    \
1228 }
1229
1230 #define iscsi_session_int_attr(field, param, format)                    \
1231         iscsi_session_int_attr_show(param, format)                      \
1232 static ISCSI_CLASS_ATTR(sess, field, S_IRUGO, show_session_int_param_##param, \
1233                         NULL);
1234
1235 iscsi_session_int_attr(initial_r2t, ISCSI_PARAM_INITIAL_R2T_EN, "%d");
1236 iscsi_session_int_attr(max_outstanding_r2t, ISCSI_PARAM_MAX_R2T, "%hu");
1237 iscsi_session_int_attr(immediate_data, ISCSI_PARAM_IMM_DATA_EN, "%d");
1238 iscsi_session_int_attr(first_burst_len, ISCSI_PARAM_FIRST_BURST, "%u");
1239 iscsi_session_int_attr(max_burst_len, ISCSI_PARAM_MAX_BURST, "%u");
1240 iscsi_session_int_attr(data_pdu_in_order, ISCSI_PARAM_PDU_INORDER_EN, "%d");
1241 iscsi_session_int_attr(data_seq_in_order, ISCSI_PARAM_DATASEQ_INORDER_EN, "%d");
1242 iscsi_session_int_attr(erl, ISCSI_PARAM_ERL, "%d");
1243 iscsi_session_int_attr(tpgt, ISCSI_PARAM_TPGT, "%d");
1244
1245 #define iscsi_session_str_attr_show(param)                              \
1246 static ssize_t                                                          \
1247 show_session_str_param_##param(struct class_device *cdev, char *buf)    \
1248 {                                                                       \
1249         struct iscsi_cls_session *session = iscsi_cdev_to_session(cdev); \
1250         struct iscsi_transport *t = session->transport;                 \
1251         return t->get_session_str_param(session, param, buf);           \
1252 }
1253
1254 #define iscsi_session_str_attr(field, param)                            \
1255         iscsi_session_str_attr_show(param)                              \
1256 static ISCSI_CLASS_ATTR(sess, field, S_IRUGO, show_session_str_param_##param, \
1257                         NULL);
1258
1259 iscsi_session_str_attr(targetname, ISCSI_PARAM_TARGET_NAME);
1260
1261 /*
1262  * Private session and conn attrs. userspace uses several iscsi values
1263  * to identify each session between reboots. Some of these values may not
1264  * be present in the iscsi_transport/LLD driver becuase userspace handles
1265  * login (and failback for login redirect) so for these type of drivers
1266  * the class manages the attrs and values for the iscsi_transport/LLD
1267  */
1268 #define iscsi_priv_session_attr_show(field, format)                     \
1269 static ssize_t                                                          \
1270 show_priv_session_##field(struct class_device *cdev, char *buf) \
1271 {                                                                       \
1272         struct iscsi_cls_session *session = iscsi_cdev_to_session(cdev); \
1273         return sprintf(buf, format"\n", session->field);                \
1274 }
1275
1276 #define iscsi_priv_session_attr(field, format)                          \
1277         iscsi_priv_session_attr_show(field, format)                     \
1278 static ISCSI_CLASS_ATTR(priv_sess, field, S_IRUGO, show_priv_session_##field, \
1279                         NULL)
1280 iscsi_priv_session_attr(targetname, "%s");
1281 iscsi_priv_session_attr(tpgt, "%d");
1282 iscsi_priv_session_attr(recovery_tmo, "%d");
1283
1284 #define iscsi_priv_conn_attr_show(field, format)                        \
1285 static ssize_t                                                          \
1286 show_priv_conn_##field(struct class_device *cdev, char *buf)            \
1287 {                                                                       \
1288         struct iscsi_cls_conn *conn = iscsi_cdev_to_conn(cdev);         \
1289         return sprintf(buf, format"\n", conn->field);                   \
1290 }
1291
1292 #define iscsi_priv_conn_attr(field, format)                             \
1293         iscsi_priv_conn_attr_show(field, format)                        \
1294 static ISCSI_CLASS_ATTR(priv_conn, field, S_IRUGO, show_priv_conn_##field, \
1295                         NULL)
1296 iscsi_priv_conn_attr(persistent_address, "%s");
1297 iscsi_priv_conn_attr(persistent_port, "%d");
1298
1299 #define SETUP_PRIV_SESSION_RD_ATTR(field)                               \
1300 do {                                                                    \
1301         priv->session_attrs[count] = &class_device_attr_priv_sess_##field; \
1302         count++;                                                        \
1303 } while (0)
1304
1305 #define SETUP_SESSION_RD_ATTR(field, param_flag)                        \
1306 do {                                                                    \
1307         if (tt->param_mask & param_flag) {                              \
1308                 priv->session_attrs[count] = &class_device_attr_sess_##field; \
1309                 count++;                                                \
1310         }                                                               \
1311 } while (0)
1312
1313 #define SETUP_PRIV_CONN_RD_ATTR(field)                                  \
1314 do {                                                                    \
1315         priv->conn_attrs[count] = &class_device_attr_priv_conn_##field; \
1316         count++;                                                        \
1317 } while (0)
1318
1319 #define SETUP_CONN_RD_ATTR(field, param_flag)                           \
1320 do {                                                                    \
1321         if (tt->param_mask & param_flag) {                              \
1322                 priv->conn_attrs[count] = &class_device_attr_conn_##field; \
1323                 count++;                                                \
1324         }                                                               \
1325 } while (0)
1326
1327 static int iscsi_session_match(struct attribute_container *cont,
1328                            struct device *dev)
1329 {
1330         struct iscsi_cls_session *session;
1331         struct Scsi_Host *shost;
1332         struct iscsi_internal *priv;
1333
1334         if (!iscsi_is_session_dev(dev))
1335                 return 0;
1336
1337         session = iscsi_dev_to_session(dev);
1338         shost = iscsi_session_to_shost(session);
1339         if (!shost->transportt)
1340                 return 0;
1341
1342         priv = to_iscsi_internal(shost->transportt);
1343         if (priv->session_cont.ac.class != &iscsi_session_class.class)
1344                 return 0;
1345
1346         return &priv->session_cont.ac == cont;
1347 }
1348
1349 static int iscsi_conn_match(struct attribute_container *cont,
1350                            struct device *dev)
1351 {
1352         struct iscsi_cls_session *session;
1353         struct iscsi_cls_conn *conn;
1354         struct Scsi_Host *shost;
1355         struct iscsi_internal *priv;
1356
1357         if (!iscsi_is_conn_dev(dev))
1358                 return 0;
1359
1360         conn = iscsi_dev_to_conn(dev);
1361         session = iscsi_dev_to_session(conn->dev.parent);
1362         shost = iscsi_session_to_shost(session);
1363
1364         if (!shost->transportt)
1365                 return 0;
1366
1367         priv = to_iscsi_internal(shost->transportt);
1368         if (priv->conn_cont.ac.class != &iscsi_connection_class.class)
1369                 return 0;
1370
1371         return &priv->conn_cont.ac == cont;
1372 }
1373
1374 static int iscsi_host_match(struct attribute_container *cont,
1375                             struct device *dev)
1376 {
1377         struct Scsi_Host *shost;
1378         struct iscsi_internal *priv;
1379
1380         if (!scsi_is_host_device(dev))
1381                 return 0;
1382
1383         shost = dev_to_shost(dev);
1384         if (!shost->transportt  ||
1385             shost->transportt->host_attrs.ac.class != &iscsi_host_class.class)
1386                 return 0;
1387
1388         priv = to_iscsi_internal(shost->transportt);
1389         return &priv->t.host_attrs.ac == cont;
1390 }
1391
1392 struct scsi_transport_template *
1393 iscsi_register_transport(struct iscsi_transport *tt)
1394 {
1395         struct iscsi_internal *priv;
1396         unsigned long flags;
1397         int count = 0, err;
1398
1399         BUG_ON(!tt);
1400
1401         priv = iscsi_if_transport_lookup(tt);
1402         if (priv)
1403                 return NULL;
1404
1405         priv = kzalloc(sizeof(*priv), GFP_KERNEL);
1406         if (!priv)
1407                 return NULL;
1408         INIT_LIST_HEAD(&priv->list);
1409         priv->iscsi_transport = tt;
1410         priv->t.user_scan = iscsi_user_scan;
1411
1412         priv->cdev.class = &iscsi_transport_class;
1413         snprintf(priv->cdev.class_id, BUS_ID_SIZE, "%s", tt->name);
1414         err = class_device_register(&priv->cdev);
1415         if (err)
1416                 goto free_priv;
1417
1418         err = sysfs_create_group(&priv->cdev.kobj, &iscsi_transport_group);
1419         if (err)
1420                 goto unregister_cdev;
1421
1422         /* host parameters */
1423         priv->t.host_attrs.ac.attrs = &priv->host_attrs[0];
1424         priv->t.host_attrs.ac.class = &iscsi_host_class.class;
1425         priv->t.host_attrs.ac.match = iscsi_host_match;
1426         priv->t.host_size = sizeof(struct iscsi_host);
1427         priv->host_attrs[0] = NULL;
1428         transport_container_register(&priv->t.host_attrs);
1429
1430         /* connection parameters */
1431         priv->conn_cont.ac.attrs = &priv->conn_attrs[0];
1432         priv->conn_cont.ac.class = &iscsi_connection_class.class;
1433         priv->conn_cont.ac.match = iscsi_conn_match;
1434         transport_container_register(&priv->conn_cont);
1435
1436         SETUP_CONN_RD_ATTR(max_recv_dlength, ISCSI_MAX_RECV_DLENGTH);
1437         SETUP_CONN_RD_ATTR(max_xmit_dlength, ISCSI_MAX_XMIT_DLENGTH);
1438         SETUP_CONN_RD_ATTR(header_digest, ISCSI_HDRDGST_EN);
1439         SETUP_CONN_RD_ATTR(data_digest, ISCSI_DATADGST_EN);
1440         SETUP_CONN_RD_ATTR(ifmarker, ISCSI_IFMARKER_EN);
1441         SETUP_CONN_RD_ATTR(ofmarker, ISCSI_OFMARKER_EN);
1442         SETUP_CONN_RD_ATTR(address, ISCSI_CONN_ADDRESS);
1443         SETUP_CONN_RD_ATTR(port, ISCSI_CONN_PORT);
1444         SETUP_CONN_RD_ATTR(exp_statsn, ISCSI_EXP_STATSN);
1445
1446         if (tt->param_mask & ISCSI_PERSISTENT_ADDRESS)
1447                 SETUP_CONN_RD_ATTR(persistent_address, ISCSI_PERSISTENT_ADDRESS);
1448         else
1449                 SETUP_PRIV_CONN_RD_ATTR(persistent_address);
1450
1451         if (tt->param_mask & ISCSI_PERSISTENT_PORT)
1452                 SETUP_CONN_RD_ATTR(persistent_port, ISCSI_PERSISTENT_PORT);
1453         else
1454                 SETUP_PRIV_CONN_RD_ATTR(persistent_port);
1455
1456         BUG_ON(count > ISCSI_CONN_ATTRS);
1457         priv->conn_attrs[count] = NULL;
1458         count = 0;
1459
1460         /* session parameters */
1461         priv->session_cont.ac.attrs = &priv->session_attrs[0];
1462         priv->session_cont.ac.class = &iscsi_session_class.class;
1463         priv->session_cont.ac.match = iscsi_session_match;
1464         transport_container_register(&priv->session_cont);
1465
1466         SETUP_SESSION_RD_ATTR(initial_r2t, ISCSI_INITIAL_R2T_EN);
1467         SETUP_SESSION_RD_ATTR(max_outstanding_r2t, ISCSI_MAX_R2T);
1468         SETUP_SESSION_RD_ATTR(immediate_data, ISCSI_IMM_DATA_EN);
1469         SETUP_SESSION_RD_ATTR(first_burst_len, ISCSI_FIRST_BURST);
1470         SETUP_SESSION_RD_ATTR(max_burst_len, ISCSI_MAX_BURST);
1471         SETUP_SESSION_RD_ATTR(data_pdu_in_order, ISCSI_PDU_INORDER_EN);
1472         SETUP_SESSION_RD_ATTR(data_seq_in_order, ISCSI_DATASEQ_INORDER_EN);
1473         SETUP_SESSION_RD_ATTR(erl, ISCSI_ERL);
1474         SETUP_PRIV_SESSION_RD_ATTR(recovery_tmo);
1475
1476         if (tt->param_mask & ISCSI_TARGET_NAME)
1477                 SETUP_SESSION_RD_ATTR(targetname, ISCSI_TARGET_NAME);
1478         else
1479                 SETUP_PRIV_SESSION_RD_ATTR(targetname);
1480
1481         if (tt->param_mask & ISCSI_TPGT)
1482                 SETUP_SESSION_RD_ATTR(tpgt, ISCSI_TPGT);
1483         else
1484                 SETUP_PRIV_SESSION_RD_ATTR(tpgt);
1485
1486         BUG_ON(count > ISCSI_SESSION_ATTRS);
1487         priv->session_attrs[count] = NULL;
1488
1489         spin_lock_irqsave(&iscsi_transport_lock, flags);
1490         list_add(&priv->list, &iscsi_transports);
1491         spin_unlock_irqrestore(&iscsi_transport_lock, flags);
1492
1493         printk(KERN_NOTICE "iscsi: registered transport (%s)\n", tt->name);
1494         return &priv->t;
1495
1496 unregister_cdev:
1497         class_device_unregister(&priv->cdev);
1498 free_priv:
1499         kfree(priv);
1500         return NULL;
1501 }
1502 EXPORT_SYMBOL_GPL(iscsi_register_transport);
1503
1504 int iscsi_unregister_transport(struct iscsi_transport *tt)
1505 {
1506         struct iscsi_internal *priv;
1507         unsigned long flags;
1508
1509         BUG_ON(!tt);
1510
1511         mutex_lock(&rx_queue_mutex);
1512
1513         priv = iscsi_if_transport_lookup(tt);
1514         BUG_ON (!priv);
1515
1516         spin_lock_irqsave(&iscsi_transport_lock, flags);
1517         list_del(&priv->list);
1518         spin_unlock_irqrestore(&iscsi_transport_lock, flags);
1519
1520         transport_container_unregister(&priv->conn_cont);
1521         transport_container_unregister(&priv->session_cont);
1522         transport_container_unregister(&priv->t.host_attrs);
1523
1524         sysfs_remove_group(&priv->cdev.kobj, &iscsi_transport_group);
1525         class_device_unregister(&priv->cdev);
1526         mutex_unlock(&rx_queue_mutex);
1527
1528         return 0;
1529 }
1530 EXPORT_SYMBOL_GPL(iscsi_unregister_transport);
1531
1532 static int
1533 iscsi_rcv_nl_event(struct notifier_block *this, unsigned long event, void *ptr)
1534 {
1535         struct netlink_notify *n = ptr;
1536
1537         if (event == NETLINK_URELEASE &&
1538             n->protocol == NETLINK_ISCSI && n->pid) {
1539                 struct iscsi_cls_conn *conn;
1540                 unsigned long flags;
1541
1542                 mempool_zone_complete(z_reply);
1543                 spin_lock_irqsave(&connlock, flags);
1544                 list_for_each_entry(conn, &connlist, conn_list) {
1545                         mempool_zone_complete(conn->z_error);
1546                         mempool_zone_complete(conn->z_pdu);
1547                 }
1548                 spin_unlock_irqrestore(&connlock, flags);
1549         }
1550
1551         return NOTIFY_DONE;
1552 }
1553
1554 static struct notifier_block iscsi_nl_notifier = {
1555         .notifier_call  = iscsi_rcv_nl_event,
1556 };
1557
1558 static __init int iscsi_transport_init(void)
1559 {
1560         int err;
1561
1562         err = class_register(&iscsi_transport_class);
1563         if (err)
1564                 return err;
1565
1566         err = transport_class_register(&iscsi_host_class);
1567         if (err)
1568                 goto unregister_transport_class;
1569
1570         err = transport_class_register(&iscsi_connection_class);
1571         if (err)
1572                 goto unregister_host_class;
1573
1574         err = transport_class_register(&iscsi_session_class);
1575         if (err)
1576                 goto unregister_conn_class;
1577
1578         err = netlink_register_notifier(&iscsi_nl_notifier);
1579         if (err)
1580                 goto unregister_session_class;
1581
1582         nls = netlink_kernel_create(NETLINK_ISCSI, 1, iscsi_if_rx,
1583                         THIS_MODULE);
1584         if (!nls) {
1585                 err = -ENOBUFS;
1586                 goto unregister_notifier;
1587         }
1588
1589         z_reply = mempool_zone_init(Z_MAX_REPLY,
1590                 NLMSG_SPACE(sizeof(struct iscsi_uevent)), Z_HIWAT_REPLY);
1591         if (z_reply)
1592                 return 0;
1593
1594         sock_release(nls->sk_socket);
1595 unregister_notifier:
1596         netlink_unregister_notifier(&iscsi_nl_notifier);
1597 unregister_session_class:
1598         transport_class_unregister(&iscsi_session_class);
1599 unregister_conn_class:
1600         transport_class_unregister(&iscsi_connection_class);
1601 unregister_host_class:
1602         transport_class_unregister(&iscsi_host_class);
1603 unregister_transport_class:
1604         class_unregister(&iscsi_transport_class);
1605         return err;
1606 }
1607
1608 static void __exit iscsi_transport_exit(void)
1609 {
1610         mempool_zone_destroy(z_reply);
1611         sock_release(nls->sk_socket);
1612         netlink_unregister_notifier(&iscsi_nl_notifier);
1613         transport_class_unregister(&iscsi_connection_class);
1614         transport_class_unregister(&iscsi_session_class);
1615         transport_class_unregister(&iscsi_host_class);
1616         class_unregister(&iscsi_transport_class);
1617 }
1618
1619 module_init(iscsi_transport_init);
1620 module_exit(iscsi_transport_exit);
1621
1622 MODULE_AUTHOR("Mike Christie <michaelc@cs.wisc.edu>, "
1623               "Dmitry Yusupov <dmitry_yus@yahoo.com>, "
1624               "Alex Aizman <itn780@yahoo.com>");
1625 MODULE_DESCRIPTION("iSCSI Transport Interface");
1626 MODULE_LICENSE("GPL");