2007-12-26 Igor Zelmanovich <igorz@mainsoft.com>

[mono.git] / mcs / class / System.Web / System.Web / TraceContext.cs

diff --git a/mcs/class/System.Web/System.Web/TraceContext.cs b/mcs/class/System.Web/System.Web/TraceContext.cs
index 02db0ed0819717c65fd421f3da0a58b3a5d59384..960dd6c9a5724d50b575e7ae71d0bd2f963e0e17 100644 (file)
--- a/mcs/class/System.Web/System.Web/TraceContext.cs
+++ b/mcs/class/System.Web/System.Web/TraceContext.cs
@@ -45,6 +45,9 @@ namespace System.Web {
                bool data_saved;
                bool _haveTrace;
                Hashtable view_states;
+#if NET_2_0
+               Hashtable control_states;
+#endif
                Hashtable sizes;
 
                public TraceContext (HttpContext Context)
@@ -143,7 +146,11 @@ namespace System.Web {
 
                        SetRequestDetails ();
                        if (_Context.Handler is Page)
+#if NET_2_0
+                               data.AddControlTree ((Page) _Context.Handler, view_states, control_states, sizes);
+#else
                                data.AddControlTree ((Page) _Context.Handler, view_states, sizes);
+#endif
 
                        AddCookies ();
                        AddHeaders ();
@@ -160,6 +167,15 @@ namespace System.Web {
                        view_states [ctrl] = vs;
                }
 
+#if NET_2_0
+               internal void SaveControlState (Control ctrl, object vs) {
+                       if (control_states == null)
+                               control_states = new Hashtable ();
+
+                       control_states [ctrl] = vs;
+               }
+#endif
+
                internal void SaveSize (Control ctrl, int size)
                {
                        if (sizes == null)