#include <net/netfilter/nf_conntrack_tuple.h>
 
+struct nf_ct_udp {
+       unsigned long   stream_ts;
+};
+
 /* per conntrack: protocol private data */
 union nf_conntrack_proto {
        /* insert conntrack proto private data here */
        struct nf_ct_dccp dccp;
        struct ip_ct_sctp sctp;
        struct ip_ct_tcp tcp;
+       struct nf_ct_udp udp;
        struct nf_ct_gre gre;
        unsigned int tmpl_padto;
 };
 
        if (!timeouts)
                timeouts = udp_get_timeouts(nf_ct_net(ct));
 
+       if (!nf_ct_is_confirmed(ct))
+               ct->proto.udp.stream_ts = 2 * HZ + jiffies;
+
        /* If we've seen traffic both ways, this is some kind of UDP
-          stream.  Extend timeout. */
+        * stream. Set Assured.
+        */
        if (test_bit(IPS_SEEN_REPLY_BIT, &ct->status)) {
-               nf_ct_refresh_acct(ct, ctinfo, skb,
-                                  timeouts[UDP_CT_REPLIED]);
+               unsigned long extra = timeouts[UDP_CT_UNREPLIED];
+
+               /* Still active after two seconds? Extend timeout. */
+               if (time_after(jiffies, ct->proto.udp.stream_ts))
+                       extra = timeouts[UDP_CT_REPLIED];
+
+               nf_ct_refresh_acct(ct, ctinfo, skb, extra);
+
                /* Also, more likely to be important, and not a probe */
                if (!test_and_set_bit(IPS_ASSURED_BIT, &ct->status))
                        nf_conntrack_event_cache(IPCT_ASSURED, ct);