perlasm/x86* update: support for 3 and 4 argument instructions.
[oweals/openssl.git] / crypto / perlasm / x86masm.pl
1 #!/usr/bin/env perl
2
3 package x86masm;
4
5 *out=\@::out;
6
7 $::lbdecor="\$L";       # local label decoration
8 $nmdecor="_";           # external name decoration
9
10 $initseg="";
11 $segment="";
12
13 sub ::generic
14 { my ($opcode,@arg)=@_;
15
16     # fix hexadecimal constants
17     for (@arg) { s/0x([0-9a-f]+)/0$1h/oi; }
18
19     # fix xmm references
20     $arg[0] =~ s/\b[A-Z]+WORD\s+PTR/XMMWORD PTR/i if ($arg[1]=~/\bxmm[0-7]\b/i);
21     $arg[1] =~ s/\b[A-Z]+WORD\s+PTR/XMMWORD PTR/i if ($arg[0]=~/\bxmm[0-7]\b/i);
22
23     &::emit($opcode,@arg);
24   1;
25 }
26 #
27 # opcodes not covered by ::generic above, mostly inconsistent namings...
28 #
29 sub ::call      { &::emit("call",(&::islabel($_[0]) or "$nmdecor$_[0]")); }
30 sub ::call_ptr  { &::emit("call",@_);   }
31 sub ::jmp_ptr   { &::emit("jmp",@_);    }
32
33 sub get_mem
34 { my($size,$addr,$reg1,$reg2,$idx)=@_;
35   my($post,$ret);
36
37     $ret .= "$size PTR " if ($size ne "");
38
39     $addr =~ s/^\s+//;
40     # prepend global references with optional underscore
41     $addr =~ s/^([^\+\-0-9][^\+\-]*)/&::islabel($1) or "$nmdecor$1"/ige;
42     # put address arithmetic expression in parenthesis
43     $addr="($addr)" if ($addr =~ /^.+[\-\+].+$/);
44
45     if (($addr ne "") && ($addr ne 0))
46     {   if ($addr !~ /^-/)      { $ret .= "$addr";  }
47         else                    { $post=$addr;      }
48     }
49     $ret .= "[";
50
51     if ($reg2 ne "")
52     {   $idx!=0 or $idx=1;
53         $ret .= "$reg2*$idx";
54         $ret .= "+$reg1" if ($reg1 ne "");
55     }
56     else
57     {   $ret .= "$reg1";   }
58
59     $ret .= "$post]";
60     $ret =~ s/\+\]/]/; # in case $addr was the only argument
61     $ret =~ s/\[\s*\]//;
62
63   $ret;
64 }
65 sub ::BP        { &get_mem("BYTE",@_);  }
66 sub ::DWP       { &get_mem("DWORD",@_); }
67 sub ::QWP       { &get_mem("QWORD",@_); }
68 sub ::BC        { "@_";  }
69 sub ::DWC       { "@_"; }
70
71 sub ::file
72 { my $tmp=<<___;
73 TITLE   $_[0].asm
74 IF \@Version LT 800
75 ECHO MASM version 8.00 or later is strongly recommended.
76 ENDIF
77 .486
78 .MODEL  FLAT
79 OPTION  DOTNAME
80 IF \@Version LT 800
81 .text\$ SEGMENT PAGE 'CODE'
82 ELSE
83 .text\$ SEGMENT ALIGN(64) 'CODE'
84 ENDIF
85 ___
86     push(@out,$tmp);
87     $segment = ".text\$";
88 }
89
90 sub ::function_begin_B
91 { my $func=shift;
92   my $global=($func !~ /^_/);
93   my $begin="${::lbdecor}_${func}_begin";
94
95     &::LABEL($func,$global?"$begin":"$nmdecor$func");
96     $func="ALIGN\t16\n".$nmdecor.$func."\tPROC";
97
98     if ($global)    { $func.=" PUBLIC\n${begin}::\n"; }
99     else            { $func.=" PRIVATE\n";            }
100     push(@out,$func);
101     $::stack=4;
102 }
103 sub ::function_end_B
104 { my $func=shift;
105
106     push(@out,"$nmdecor$func ENDP\n");
107     $::stack=0;
108     &::wipe_labels();
109 }
110
111 sub ::file_end
112 { my $xmmheader=<<___;
113 .686
114 .XMM
115 IF \@Version LT 800
116 XMMWORD STRUCT 16
117 DQ      2 dup (?)
118 XMMWORD ENDS
119 ENDIF
120 ___
121     if (grep {/\b[x]?mm[0-7]\b/i} @out) {
122         grep {s/\.[3-7]86/$xmmheader/} @out;
123     }
124
125     push(@out,"$segment ENDS\n");
126
127     if (grep {/\b${nmdecor}OPENSSL_ia32cap_P\b/i} @out)
128     {   my $comm=<<___;
129 .bss    SEGMENT
130 COMM    ${nmdecor}OPENSSL_ia32cap_P:DWORD
131 .bss    ENDS
132 ___
133         # comment out OPENSSL_ia32cap_P declarations
134         grep {s/(^EXTERN\s+${nmdecor}OPENSSL_ia32cap_P)/\;$1/} @out;
135         push (@out,$comm);
136     }
137     push (@out,$initseg) if ($initseg);
138     push (@out,"END\n");
139 }
140
141 sub ::comment {   foreach (@_) { push(@out,"\t; $_\n"); }   }
142
143 *::set_label_B = sub
144 { my $l=shift; push(@out,$l.($l=~/^\Q${::lbdecor}\E[0-9]{3}/?":\n":"::\n")); };
145
146 sub ::external_label
147 {   foreach(@_)
148     {   push(@out, "EXTERN\t".&::LABEL($_,$nmdecor.$_).":NEAR\n");   }
149 }
150
151 sub ::public_label
152 {   push(@out,"PUBLIC\t".&::LABEL($_[0],$nmdecor.$_[0])."\n");   }
153
154 sub ::data_byte
155 {   push(@out,("DB\t").join(',',@_)."\n");      }
156
157 sub ::data_word
158 {   push(@out,("DD\t").join(',',@_)."\n");      }
159
160 sub ::align
161 {   push(@out,"ALIGN\t$_[0]\n");        }
162
163 sub ::picmeup
164 { my($dst,$sym)=@_;
165     &::lea($dst,&::DWP($sym));
166 }
167
168 sub ::initseg
169 { my $f=$nmdecor.shift;
170
171     $initseg.=<<___;
172 .CRT\$XCU       SEGMENT DWORD PUBLIC 'DATA'
173 EXTERN  $f:NEAR
174 DD      $f
175 .CRT\$XCU       ENDS
176 ___
177 }
178
179 sub ::dataseg
180 {   push(@out,"$segment\tENDS\n_DATA\tSEGMENT\n"); $segment="_DATA";   }
181
182 1;