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