Replace similar code with a common routine.

svn path=/trunk/; revision=53519

1 files changed, 2 insertions, 21 deletions

diff --git a/ui/qt/main_window.cpp b/ui/qt/main_window.cpp
index cbd010aa8e..d3c827406d 100644
--- a/ui/qt/main_window.cpp
+++ b/ui/qt/main_window.cpp
@@ -1366,9 +1366,6 @@ void MainWindow::setTitlebarForCaptureInProgress()
 
 void MainWindow::setMenusForFollowStream()
 {
-    wmem_list_frame_t* protos;
-    int proto_id;
-    const char* proto_name;
     gboolean is_tcp = FALSE, is_udp = FALSE;
 
     if (!cap_file_)
@@ -1381,24 +1378,8 @@ void MainWindow::setMenusForFollowStream()
     main_ui_->actionAnalyzeFollowUDPStream->setEnabled(false);
     main_ui_->actionAnalyzeFollowSSLStream->setEnabled(false);
 
-    protos = wmem_list_head(cap_file_->edt->pi.layers);
-
-    /* walk the list of a available protocols in the packet to
-        figure out if any of them affect context sensitivity */
-    while (protos != NULL)
-    {
-        proto_id = GPOINTER_TO_INT(wmem_list_frame_data(protos));
-        proto_name = proto_get_protocol_filter_name(proto_id);
-
-        if (!strcmp(proto_name, "tcp")) {
-            is_tcp = TRUE;
-        } else if (!strcmp(proto_name, "udp")) {
-            is_udp = TRUE;
-        }
-
-        protos = wmem_list_frame_next(protos);
-    }
-
+    proto_get_frame_protocols(cap_file_->edt->pi.layers, NULL, &is_tcp, &is_udp, NULL);
+    
     if (is_tcp)
     {
         main_ui_->actionAnalyzeFollowTCPStream->setEnabled(true);