- Continued on the new C parser.
[wine] / tools / winapi / winapi_fixup_statements.pm
1 package winapi_fixup_statements;
2
3 use strict;
4
5 use vars qw($VERSION @ISA @EXPORT @EXPORT_OK);
6 require Exporter;
7
8 @ISA = qw(Exporter);
9 @EXPORT = qw();
10 @EXPORT_OK = qw(&fixup_statements);
11
12 use config qw($wine_dir);
13 use options qw($options);
14 use output qw($output);
15
16 use c_parser;
17 use winapi_module_user qw(
18     &get_message_result_kind
19     &get_message_wparam_kind
20     &get_message_lparam_kind
21 );
22
23 ########################################################################
24 # fixup_function_call
25
26 sub fixup_function_call {
27     my $name = shift;
28     my @arguments = @{(shift)};;
29
30     return "$name(" . join(", ", @arguments) . ")";
31 }
32
33 ########################################################################
34 # _parse_makelong
35
36 sub _parse_makelong {
37     local $_ = shift;
38
39     my $low;
40     my $high;
41
42     my $name;
43     my @arguments;
44     my @argument_lines;
45     my @argument_columns;
46
47     my $parser = new c_parser;
48
49     my $line = 1;
50     my $column = 0;
51     if($parser->parse_c_function_call(\$_, \$line, \$column, \$name, \@arguments, \@argument_lines, \@argument_columns) &&
52        $name =~ /^MAKE(?:LONG|LPARAM|LRESULT|WPARAM)$/) 
53     {
54         $low = $arguments[0];
55         $high = $arguments[1];
56     } elsif(/^(?:\(\w+\)\s*)?0L?$/) {
57         $low = "0";
58         $high = "0";
59     } else {
60         $low = "($_) & 0xffff";
61         $high = "($_) << 16";
62     }
63
64     $low =~ s/^\s*(.*?)\s*$/$1/;
65     $high =~ s/^\s*(.*?)\s*$/$1/;
66
67     return ($low, $high);
68 }
69
70 ########################################################################
71 # fixup_function_call_2_windowsx
72
73 sub fixup_user_message_2_windowsx {
74     my $name = shift;
75     (my $hwnd, my $msg, my $wparam, my $lparam) = @{(shift)};
76
77     if($msg !~ /^WM_/) {
78         return undef;
79     } elsif($msg =~ /^(?:WM_BEGINDRAG|WM_ENTERMENULOOP|WM_EXITMENULOOP|WM_HELP|
80                        WM_ISACTIVEICON|WM_LBTRACKPOINT|WM_NEXTMENU)$/x) 
81     {
82         return undef;
83     } elsif($msg =~ /^WM_(?:GET|SET)TEXT$/) {
84         return undef;
85     }
86
87     my $suffix;
88     $name =~ /([AW])?$/;
89     if(defined($1)) {
90         $suffix = $1;
91     } else {
92         $suffix = "";
93     }
94
95     $wparam =~ s/^\(WPARAM\)\s*//;
96     $lparam =~ s/^\(LPARAM\)\s*//;
97
98     my @arguments;
99     if(0) {
100         # Nothing
101     } elsif($msg =~ /^WM_COMMAND$/) {
102         (my $id, my $code_notify) = _parse_makelong($wparam);
103         my $hwndctl = $lparam;
104         @arguments = ($id, $hwndctl, $code_notify);
105     } elsif($msg =~ /^WM_(?:COPY|CUT|PASTE)$/) {
106         @arguments = ();
107     } elsif($msg =~ /^WM_(?:CHARTO|VKEYTO)ITEM$/) {
108         (my $key, my $caret) = _parse_makelong($wparam);
109         my $hwndctl = $lparam;
110         @arguments = ($key, $hwndctl, $caret);
111     } elsif($msg =~ /^WM_(?:COMPARE|DELETE|DRAW|MEASURE)ITEM$/) {
112         @arguments = ($lparam);
113     } elsif($msg =~ s/^WM_GETTEXT$/$&$suffix/) {
114         @arguments = ($wparam, $lparam);
115     } elsif($msg =~ /^WM_INITMENU$/) {
116         my $hmenu = $wparam;
117         @arguments =  ($hmenu);
118     } elsif($msg =~ /^WM_INITMENUPOPUP$/) {
119         my $hmenu = $wparam;
120         (my $item, my $system_menu) = _parse_makelong($lparam);
121         @arguments =  ($hmenu, $item, $system_menu);
122     } elsif($msg =~ /^WM_MENUCHAR$/) {
123         (my $ch, my $flags) = _parse_makelong($wparam);
124         my $hmenu = $lparam;
125         @arguments = ($ch, $flags, $hmenu);
126     } elsif($msg =~ /^WM_MENUSELECT$/) {
127         (my $item, my $flags) = _parse_makelong($wparam);
128         my $hmenu = $lparam;
129         my $hmenu_popup = "NULL"; # FIXME: Is this really correct?
130         @arguments = ($hmenu, $item, $hmenu_popup, $flags);
131     } elsif($msg =~ s/^WM_(NC)?LBUTTONDBLCLK$/WM_$1LBUTTONDOWN/) {
132         my $double_click = "TRUE";
133         my $key_flags = $wparam;
134         (my $x, my $y) = _parse_makelong($lparam);
135         @arguments = ($double_click, $x, $y, $key_flags);
136     } elsif($msg =~ /^WM_(NC)?LBUTTONDOWN$/) {
137         my $double_click = "FALSE";
138         my $key_flags = $wparam;
139         (my $x, my $y) = _parse_makelong($lparam);
140         @arguments = ($double_click, $x, $y, $key_flags);
141     } elsif($msg =~ /^WM_LBUTTONUP$/) {
142         my $key_flags = $wparam;
143         (my $x, my $y) = _parse_makelong($lparam);
144         @arguments = ($x, $y, $key_flags);
145     } elsif($msg =~ /^WM_SETCURSOR$/) {
146         my $hwnd_cursor = $wparam;
147         (my $code_hit_test, my $msg2) = _parse_makelong($lparam);
148         @arguments = ($hwnd_cursor, $code_hit_test, $msg2);
149     } elsif($msg =~ s/^WM_SETTEXT$/$&$suffix/) {
150         my $text = $lparam;
151         @arguments = ($text);
152     } elsif($msg =~ /^WM_(?:SYS)?KEYDOWN$/) {
153         my $vk = $wparam;
154         (my $repeat, my $flags) = _parse_makelong($lparam);
155         @arguments = ($vk, $repeat, $flags);
156     } else {
157         @arguments = ($wparam, $lparam);
158     }
159     unshift @arguments, $hwnd;
160
161     return "FORWARD_" . $msg . "(" . join(", ", @arguments) . ", $name)";
162 }
163
164 ########################################################################
165 # _fixup_user_message
166
167 sub _get_messages {
168     local $_ = shift;
169
170     if(/^WM_\w+$/) {
171         return ($_)
172     } elsif(/^(.*?)\s*\?\s*(WM_\w+)\s*:\s*(WM_\w+)$/) {
173         return ($2, $3);
174     } elsif(/^\w+$/) {
175         return ();
176     } else {
177         $output->write("_fixup_user_message: '$_'\n");
178         exit 1;
179     }
180 }
181
182 ########################################################################
183 # _fixup_user_message
184
185 sub _fixup_user_message {
186     my $name = shift;
187     (my $hwnd, my $msg, my $wparam, my $lparam) = @{(shift)};
188
189     my $modified = 0;
190
191     my $wkind;
192     my $lkind;
193     foreach my $msg (_get_messages($msg)) {
194         my $new_wkind = &get_message_wparam_kind($msg);
195         if(defined($wkind) && $new_wkind ne $wkind) {
196             $output->write("messsages used together do not have the same type\n");
197         } else {
198             $wkind = $new_wkind;
199         }
200
201         my $new_lkind = &get_message_lparam_kind($msg);
202         if(defined($lkind) && $new_lkind ne $lkind) {
203             $output->write("messsages used together do not have the same type\n");
204         } else {
205             $lkind = $new_lkind;
206         }
207     }
208
209     my @entries = (
210         [ \$wparam, $wkind, "W", "w" ],
211         [ \$lparam, $lkind, "L", "l" ]
212     );
213     foreach my $entry (@entries) {
214         (my $refparam, my $kind, my $upper, my $lower) = @$entry;
215
216         if(!defined($kind)) {
217             if($msg =~ /^WM_/) {
218                 $output->write("messsage $msg not defined\n");
219             }
220         } elsif($kind eq "ptr") {
221             if($$refparam =~ /^(\(${upper}PARAM\))?\s*($lower[pP]aram)$/) {
222                 if(defined($1)) {
223                     $$refparam = $2;
224                     $modified = 1;
225                 }
226             } elsif($$refparam =~ /^(\(${upper}PARAM\))?\s*0$/) {
227                 $$refparam = "(${upper}PARAM) NULL";
228                 $modified = 1;
229             } elsif($$refparam !~ /^\(${upper}PARAM\)\s*/) {
230                 $$refparam = "(${upper}PARAM) $$refparam";
231                 $modified = 1;
232             }
233         } elsif($kind eq "long") {
234             if($$refparam =~ s/^\(${upper}PARAM\)\s*//) {
235                 $modified = 1;
236             }
237         }
238     }
239
240     if($modified) {
241         my @arguments = ($hwnd, $msg, $wparam, $lparam);
242         return "$name(" . join(", ", @arguments) . ")";
243     } else {
244         return undef;
245     }
246 }
247
248 ########################################################################
249 # fixup_statements
250
251 sub fixup_statements {
252     my $function = shift;
253     my $editor = shift;
254
255     my $file = $function->file;
256     my $linkage = $function->linkage;
257     my $name = $function->name;
258     my $statements_line = $function->statements_line;
259     my $statements_column = $function->statements_column;
260     my $statements = $function->statements;
261    
262     if(!defined($statements)) {
263         return;
264     }
265
266     if(0 && $statements_line > 490) {
267         $output->write("$statements_line: \\\n");
268         my $line = $statements_line;
269         foreach my $statement (split(/\n/, $statements)) {
270             $output->write("$line: $statement\n");
271             $line++;
272         }
273     }
274
275     my $parser = new c_parser($file);
276     
277     my $found_function_call = sub {
278         my $begin_line = shift;
279         my $begin_column = shift;
280         my $end_line = shift;
281         my $end_column = shift;
282         my $name = shift;
283         my $arguments = shift;
284         
285         foreach my $argument (@$arguments) {
286             $argument =~ s/^\s*(.*?)\s*$/$1/;
287         }
288
289         my $fixup_function_call;
290         if($name =~ /^(?:DefWindowProc|SendMessage)[AW]$/)
291         {
292             if($options->statements_windowsx) {
293                 $fixup_function_call = \&fixup_user_message_2_windowsx;
294             } else {
295                 $fixup_function_call = \&_fixup_user_message;
296             }
297         } 
298
299         if(defined($fixup_function_call)) {
300             my $replace = &$fixup_function_call($name, $arguments);
301
302             if(defined($replace)) {
303                 $editor->replace($begin_line, $begin_column, $end_line, $end_column, $replace);
304             }
305         }  elsif(0 || $options->debug) {
306             $output->write("$begin_line.$begin_column-$end_line.$end_column: " .
307                            "$name(" . join(", ", @$arguments) . ")\n");
308         }
309
310         return 0;
311     };
312     
313     $parser->set_found_function_call_callback($found_function_call);
314     
315     my $line = $statements_line;
316     my $column = 0;     
317     if(!$parser->parse_c_statements(\$statements, \$line, \$column)) {
318         $output->write("error: can't parse statements\n");
319     }
320 }
321
322 1;