Start of a PCNFSD dissector as the RPC dissector 150001.
[obnox/wireshark/wip.git] / packet-imap.c
index d7f1155f1b72ec1868b798e7b1d395018264fabc..8a1c506f429076476a318a5e09961461b50aaac1 100644 (file)
@@ -2,7 +2,7 @@
  * Routines for imap packet dissection
  * Copyright 1999, Richard Sharpe <rsharpe@ns.aus.com>
  *
- * $Id: packet-imap.c,v 1.5 2000/04/08 07:07:19 guy Exp $
+ * $Id: packet-imap.c,v 1.15 2001/06/18 02:17:47 guy Exp $
  *
  * Ethereal - Network traffic analyzer
  * By Gerald Combs <gerald@zing.org>
@@ -42,6 +42,7 @@
 #include <string.h>
 #include <glib.h>
 #include "packet.h"
+#include "strutil.h"
 
 static int proto_imap = -1;
 static int hf_imap_response = -1;
@@ -52,64 +53,105 @@ static gint ett_imap = -1;
 #define TCP_PORT_IMAP                  143
 
 static void
-dissect_imap(const u_char *pd, int offset, frame_data *fd, proto_tree *tree)
+dissect_imap(tvbuff_t *tvb, packet_info *pinfo, proto_tree *tree)
 {
+        gboolean        is_request;
         proto_tree      *imap_tree, *ti;
-       gchar          rr[50], rd[1500];
-       int i1 = (u_char *)strchr(pd + offset, ' ') - (pd + offset); /* Where is that space */
-       int i2;
-       int max_data = pi.captured_len - offset;
-
-       memset(rr, '\0', sizeof(rr));
-       memset(rd, '\0', sizeof(rd));
-
-       if ((i1 > max_data) || (i1 <= 0)) {
-         
-         i1 = max_data;
-         strncpy(rr, pd + offset, MIN(max_data - 2, sizeof(rr) - 1));
-
-       }
-       else {
-
-         strncpy(rr, pd + offset, MIN(i1, sizeof(rr) - 1));
-         i2 = ((u_char *)strchr(pd + offset + i1 + 1, '\r') - (pd + offset)) - i1 - 1;
-         if (i2 > max_data - i1 - 1 || i2 <= 0) {
-           i2 = ((u_char *)strchr(pd + offset + i1 + 1, '\n') - (pd + offset)) - i1 - 1;
-           if (i2 > max_data - i1 - 1 || i2 <= 0)
-             i2 = max_data - i1 - 1;
-         }
-         strncpy(rd, pd + offset + i1 + 1, MIN(i2, sizeof(rd) - 1));
-       }
-
-       if (check_col(fd, COL_PROTOCOL))
-               col_add_str(fd, COL_PROTOCOL, "IMAP");
-
-       if (check_col(fd, COL_INFO)) {
-
-         col_add_fstr(fd, COL_INFO, "%s: %s %s", (pi.match_port == pi.destport)? "Request" : "Response", rr, rd);        
+       gint            offset = 0;
+       const u_char    *line;
+       gint            next_offset;
+       int             linelen;
+       int             tokenlen;
+       const u_char    *next_token;
+
+       if (check_col(pinfo->fd, COL_PROTOCOL))
+               col_set_str(pinfo->fd, COL_PROTOCOL, "IMAP");
+
+       /*
+        * Find the end of the first line.
+        *
+        * Note that "tvb_find_line_end()" will return a value that is
+        * not longer than what's in the buffer, so the "tvb_get_ptr()"
+        * call won't throw an exception.
+        */
+       linelen = tvb_find_line_end(tvb, offset, -1, &next_offset);
+       line = tvb_get_ptr(tvb, offset, linelen);
+
+       if (pinfo->match_port == pinfo->destport)
+               is_request = TRUE;
+       else
+               is_request = FALSE;
+
+       if (check_col(pinfo->fd, COL_INFO)) {
+               /*
+                * Put the first line from the buffer into the summary
+                * (but leave out the line terminator).
+                */
+               col_add_fstr(pinfo->fd, COL_INFO, "%s: %s",
+                   is_request ? "Request" : "Response",
+                   format_text(line, linelen));
        }
 
        if (tree) {
-
-         ti = proto_tree_add_item(tree, proto_imap, offset, END_OF_FRAME, NULL);
-         imap_tree = proto_item_add_subtree(ti, ett_imap);
-
-         if (pi.match_port == pi.destport) { /* Request */
-
-           proto_tree_add_item_hidden(imap_tree, hf_imap_request, offset, i1, TRUE);
-           proto_tree_add_text(imap_tree, offset, i1, "Request Tag: %s", rr);
-
-           proto_tree_add_text(imap_tree, offset + i1 + 1, END_OF_FRAME, "Request: %s", rd);
-
-         }
-         else {
-
-           proto_tree_add_item_hidden(imap_tree, hf_imap_response, offset, i1, TRUE);
-           proto_tree_add_text(imap_tree, offset, i1, "Response Tag: %s", rr);
-
-           proto_tree_add_text(imap_tree, offset + i1 + 1, END_OF_FRAME, "Response: %s", rd);
-         }
-
+               ti = proto_tree_add_item(tree, proto_imap, tvb, offset,
+                   tvb_length_remaining(tvb, offset), FALSE);
+               imap_tree = proto_item_add_subtree(ti, ett_imap);
+
+               if (is_request) {
+                       proto_tree_add_boolean_hidden(imap_tree,
+                           hf_imap_request, tvb, 0, 0, TRUE);
+               } else {
+                       proto_tree_add_boolean_hidden(imap_tree,
+                           hf_imap_response, tvb, 0, 0, TRUE);
+               }
+
+               /*
+                * Show the first line as tags + requests or replies.
+                */
+
+               /*
+                * Extract the first token, and, if there is a first
+                * token, add it as the request or reply tag.
+                */
+               tokenlen = get_token_len(line, line + linelen, &next_token);
+               if (tokenlen != 0) {
+                       if (is_request) {
+                               proto_tree_add_text(imap_tree, tvb, offset,
+                                   tokenlen, "Request Tag: %s",
+                                   format_text(line, tokenlen));
+                       } else {
+                               proto_tree_add_text(imap_tree, tvb, offset,
+                                   tokenlen, "Response Tag: %s",
+                                   format_text(line, tokenlen));
+                       }
+                       offset += next_token - line;
+                       linelen -= next_token - line;
+                       line = next_token;
+               }
+
+               /*
+                * Add the rest of the line as request or reply data.
+                */
+               if (linelen != 0) {
+                       if (is_request) {
+                               proto_tree_add_text(imap_tree, tvb, offset,
+                                   linelen, "Request: %s",
+                                   format_text(line, linelen));
+                       } else {
+                               proto_tree_add_text(imap_tree, tvb, offset,
+                                   linelen, "Response: %s",
+                                   format_text(line, linelen));
+                       }
+               }
+
+               /*
+                * XXX - show the rest of the frame; this requires that
+                * we handle literals, quoted strings, continuation
+                * responses, etc..
+                *
+                * This involves a state machine, and attaching
+                * state information to the packets.
+                */
        }
 }
 
@@ -120,19 +162,19 @@ proto_register_imap(void)
     { &hf_imap_response,
       { "Response",           "imap.response",
        FT_BOOLEAN, BASE_NONE, NULL, 0x0,
-       "TRUE if IMAP response" }},
+       "TRUE if IMAP response", HFILL }},
 
     { &hf_imap_request,
       { "Request",            "imap.request",
        FT_BOOLEAN, BASE_NONE, NULL, 0x0,
-       "TRUE if IMAP request" }}
+       "TRUE if IMAP request", HFILL }}
   };
   static gint *ett[] = {
     &ett_imap,
   };
 
   proto_imap = proto_register_protocol("Internet Message Access Protocol", 
-                                      "imap");
+                                      "IMAP", "imap");
   proto_register_field_array(proto_imap, hf, array_length(hf));
   proto_register_subtree_array(ett, array_length(ett));
 }
@@ -140,5 +182,5 @@ proto_register_imap(void)
 void
 proto_reg_handoff_imap(void)
 {
-  dissector_add("tcp.port", TCP_PORT_IMAP, dissect_imap);
+  dissector_add("tcp.port", TCP_PORT_IMAP, dissect_imap, proto_imap);
 }