esx_vi_generator.py 61.9 KB
Newer Older
1
#!/usr/bin/env python
2 3 4 5

#
# esx_vi_generator.py: generates most of the SOAP type mapping code
#
M
Matthias Bolte 已提交
6
# Copyright (C) 2010-2012 Matthias Bolte <matthias.bolte@googlemail.com>
7
# Copyright (C) 2013 Ata E Husain Bohra <ata.husain@hotmail.com>
8 9 10 11 12 13 14 15 16 17 18 19
#
# This library is free software; you can redistribute it and/or
# modify it under the terms of the GNU Lesser General Public
# License as published by the Free Software Foundation; either
# version 2.1 of the License, or (at your option) any later version.
#
# This library is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# Lesser General Public License for more details.
#
# You should have received a copy of the GNU Lesser General Public
20
# License along with this library.  If not, see
21
# <http://www.gnu.org/licenses/>.
22 23 24 25 26 27 28 29
#

import sys
import os
import os.path



30 31 32 33 34
OCCURRENCE__REQUIRED_ITEM = "r"
OCCURRENCE__REQUIRED_LIST = "rl"
OCCURRENCE__OPTIONAL_ITEM = "o"
OCCURRENCE__OPTIONAL_LIST = "ol"
OCCURRENCE__IGNORED = "i"
35

36 37 38 39 40
valid_occurrences = [OCCURRENCE__REQUIRED_ITEM,
                     OCCURRENCE__REQUIRED_LIST,
                     OCCURRENCE__OPTIONAL_ITEM,
                     OCCURRENCE__OPTIONAL_LIST,
                     OCCURRENCE__IGNORED]
41

42
autobind_names = set()
43

44 45
separator = "/* " + ("* " * 37) + "*\n"

46 47


48 49
def aligned(left, right, length=59):
    while len(left) < length:
M
Matthias Bolte 已提交
50 51 52
        left += " "

    return left + right
53 54 55



56 57
class Member:
    def __init__(self, type, occurrence):
58 59 60 61 62 63 64 65
        self.type = type
        self.occurrence = occurrence


    def is_enum(self):
        return self.type in predefined_enums or self.type in enums_by_name


66 67 68 69 70 71 72 73
    def is_object(self):
        return self.type in predefined_objects or self.type in objects_by_name


    def is_type_generated(self):
        return self.type in enums_by_name or self.type in objects_by_name


74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
    def get_occurrence_comment(self):
        if self.occurrence == OCCURRENCE__REQUIRED_ITEM:
            return "/* required */"
        elif self.occurrence == OCCURRENCE__REQUIRED_LIST:
            return "/* required, list */"
        elif self.occurrence == OCCURRENCE__OPTIONAL_ITEM:
            return "/* optional */"
        elif self.occurrence == OCCURRENCE__OPTIONAL_LIST:
            return "/* optional, list */"

        raise ValueError("unknown occurrence value '%s'" % self.occurrence)



class Parameter(Member):
    def __init__(self, type, name, occurrence):
        Member.__init__(self, type, occurrence)

        if ':' in name and name.startswith("_this"):
93
            self.name, self.autobind_name = name.split(":")
94 95
        else:
            self.name = name
96
            self.autobind_name = None
97 98


99
    def generate_parameter(self, is_last=False, is_header=True, offset=0):
100
        if self.occurrence == OCCURRENCE__IGNORED:
101 102
            raise ValueError("invalid function parameter occurrence value '%s'"
                             % self.occurrence)
103
        elif self.autobind_name is not None:
104 105 106 107 108 109 110 111 112 113 114 115 116 117
            return ""
        else:
            string = "       "
            string += " " * offset
            string += "%s%s" % (self.get_type_string(), self.name)

            if is_last:
                if is_header:
                    string += "); "
                else:
                    string += "), "
            else:
                string += ", "

M
Matthias Bolte 已提交
118
            return aligned(string, self.get_occurrence_comment() + "\n")
119 120 121 122


    def generate_return(self, offset = 0, end_of_line = ";"):
        if self.occurrence == OCCURRENCE__IGNORED:
123 124
            raise ValueError("invalid function parameter occurrence value '%s'"
                             % self.occurrence)
125 126 127
        else:
            string = "       "
            string += " " * offset
128 129
            string += "%s%s)%s" \
                      % (self.get_type_string(True), self.name, end_of_line)
130

M
Matthias Bolte 已提交
131
            return aligned(string, self.get_occurrence_comment() + "\n")
132 133 134 135 136 137 138 139 140 141 142 143 144


    def generate_require_code(self):
        if self.occurrence in [OCCURRENCE__REQUIRED_ITEM,
                               OCCURRENCE__REQUIRED_LIST]:
            return "    ESX_VI__METHOD__PARAMETER__REQUIRE(%s)\n" % self.name
        else:
            return ""


    def generate_serialize_code(self):
        if self.occurrence in [OCCURRENCE__REQUIRED_LIST,
                               OCCURRENCE__OPTIONAL_LIST]:
145 146
            return "    ESX_VI__METHOD__PARAMETER__SERIALIZE_LIST(%s, %s)\n" \
                   % (self.type, self.name)
147
        elif self.type == "String":
148 149
            return "    ESX_VI__METHOD__PARAMETER__SERIALIZE_VALUE(String, %s)\n" \
                   % self.name
150
        else:
151 152
            return "    ESX_VI__METHOD__PARAMETER__SERIALIZE(%s, %s)\n" \
                   % (self.type, self.name)
153 154


155
    def get_type_string(self, as_return_value=False):
156 157
        string = ""

158 159 160
        if self.type == "String" and \
           self.occurrence not in [OCCURRENCE__REQUIRED_LIST,
                                   OCCURRENCE__OPTIONAL_LIST]:
161 162 163 164
            if as_return_value:
                string += "char *"
            else:
                string += "const char *"
165
        elif self.is_enum():
166
            string += "esxVI_%s " % self.type
167
        else:
168 169 170 171 172 173
            string += "esxVI_%s *" % self.type

        if as_return_value:
            string += "*"

        return string
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197


    def get_occurrence_short_enum(self):
        if self.occurrence == OCCURRENCE__REQUIRED_ITEM:
            return "RequiredItem"
        elif self.occurrence == OCCURRENCE__REQUIRED_LIST:
            return "RequiredList"
        elif self.occurrence == OCCURRENCE__OPTIONAL_ITEM:
            return "OptionalItem"
        elif self.occurrence == OCCURRENCE__OPTIONAL_LIST:
            return "OptionalList"

        raise ValueError("unknown occurrence value '%s'" % self.occurrence)



class Method:
    def __init__(self, name, parameters, returns):
        self.name = name
        self.parameters = []
        self.autobind_parameter = None
        self.returns = returns

        for parameter in parameters:
198
            if parameter.autobind_name is None:
199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
                self.parameters.append(parameter)
            else:
                self.autobind_parameter = parameter


    def generate_header(self):
        header = "int esxVI_%s\n" % self.name
        header += "      (esxVI_Context *ctx"

        if len(self.parameters) > 0 or self.returns is not None:
            header += ",\n"

            for parameter in self.parameters[:-1]:
                header += parameter.generate_parameter()

            if self.returns is None:
215
                header += self.parameters[-1].generate_parameter(is_last=True)
216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231
            else:
                header += self.parameters[-1].generate_parameter()
                header += self.returns.generate_return()
        else:
            header += ");\n"

        header += "\n"

        return header


    def generate_source(self):
        source = "/* esxVI_%s */\n" % self.name
        source += "ESX_VI__METHOD(%s," % self.name

        if self.autobind_parameter is not None:
232 233
            autobind_names.add(self.autobind_parameter.autobind_name)
            source += " %s,\n" % self.autobind_parameter.autobind_name
234 235 236 237 238 239 240 241 242
        else:
            source += " /* explicit _this */,\n"

        source += "               (esxVI_Context *ctx"

        if len(self.parameters) > 0 or self.returns is not None:
            source += ",\n"

            for parameter in self.parameters[:-1]:
243 244
                source += parameter.generate_parameter(is_header=False,
                                                       offset=9)
245 246

            if self.returns is None:
247 248 249
                source += self.parameters[-1].generate_parameter(is_last=True,
                                                                 is_header=False,
                                                                 offset=9)
250
            else:
251 252 253 254
                source += self.parameters[-1].generate_parameter(is_header=False,
                                                                 offset=9)
                source += self.returns.generate_return(offset=9,
                                                       end_of_line=",")
255 256 257 258
        else:
            source += "),\n"

        if self.returns is None:
259 260
            source += "               void, /* nothing */, None,\n"
        elif self.returns.type == "String":
261 262
            source += "               String, Value, %s,\n" \
                      % self.returns.get_occurrence_short_enum()
263
        else:
264 265 266
            source += "               %s, /* nothing */, %s,\n" \
                      % (self.returns.type,
                         self.returns.get_occurrence_short_enum())
267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290

        source += "{\n"

        if self.autobind_parameter is not None:
            source += self.autobind_parameter.generate_require_code()

        for parameter in self.parameters:
            source += parameter.generate_require_code()

        source += "},\n"
        source += "{\n"

        if self.autobind_parameter is not None:
            source += self.autobind_parameter.generate_serialize_code()

        for parameter in self.parameters:
            source += parameter.generate_serialize_code()

        source += "})\n\n\n\n"

        return source



291
class Property(Member):
292
    def __init__(self, type, name, occurrence):
293
        Member.__init__(self, type, occurrence)
294

295
        self.name = name
296 297


298
    def generate_struct_member(self):
299
        if self.occurrence == OCCURRENCE__IGNORED:
300 301 302 303
            return "    /* FIXME: %s is currently ignored */\n" % self.name
        else:
            string = "    %s%s; " % (self.get_type_string(), self.name)

M
Matthias Bolte 已提交
304
            return aligned(string, self.get_occurrence_comment() + "\n")
305

306 307 308

    def generate_free_code(self):
        if self.type == "String" and \
309 310 311
           self.occurrence not in [OCCURRENCE__REQUIRED_LIST,
                                   OCCURRENCE__OPTIONAL_LIST,
                                   OCCURRENCE__IGNORED]:
312 313 314 315
            return "    VIR_FREE(item->%s);\n" % self.name
        elif self.is_enum():
            return ""
        else:
316
            if self.occurrence == OCCURRENCE__IGNORED:
317 318 319 320
                return "    /* FIXME: %s is currently ignored */\n" % self.name
            else:
                return "    esxVI_%s_Free(&item->%s);\n" % (self.type, self.name)

321

322 323 324 325 326 327
    def generate_validate_code(self, managed=False):
        if managed:
            macro = "ESX_VI__TEMPLATE__PROPERTY__MANAGED_REQUIRE"
        else:
            macro = "ESX_VI__TEMPLATE__PROPERTY__REQUIRE"

328 329
        if self.occurrence in [OCCURRENCE__REQUIRED_ITEM,
                               OCCURRENCE__REQUIRED_LIST]:
330
            return "    %s(%s)\n" % (macro, self.name)
331
        elif self.occurrence == OCCURRENCE__IGNORED:
332 333 334 335
            return "    /* FIXME: %s is currently ignored */\n" % self.name
        else:
            return ""

336

337
    def generate_deep_copy_code(self):
338
        if self.occurrence == OCCURRENCE__IGNORED:
339
            return "    /* FIXME: %s is currently ignored */\n" % self.name
340 341
        elif self.occurrence in [OCCURRENCE__REQUIRED_LIST,
                                 OCCURRENCE__OPTIONAL_LIST]:
342 343
            return "    ESX_VI__TEMPLATE__PROPERTY__DEEP_COPY_LIST(%s, %s)\n" \
                   % (self.type, self.name)
344
        elif self.type == "String":
345 346
            return "    ESX_VI__TEMPLATE__PROPERTY__DEEP_COPY_VALUE(String, %s)\n" \
                   % self.name
347 348
        elif self.is_enum():
            return "    (*dest)->%s = src->%s;\n" % (self.name, self.name)
349
        else:
350 351
            return "    ESX_VI__TEMPLATE__PROPERTY__DEEP_COPY(%s, %s)\n" \
                   % (self.type, self.name)
352

353

354
    def generate_serialize_code(self):
355
        if self.occurrence == OCCURRENCE__IGNORED:
356
            return "    /* FIXME: %s is currently ignored */\n" % self.name
357 358
        elif self.occurrence in [OCCURRENCE__REQUIRED_LIST,
                                 OCCURRENCE__OPTIONAL_LIST]:
359 360
            return "    ESX_VI__TEMPLATE__PROPERTY__SERIALIZE_LIST(%s, %s)\n" \
                   % (self.type, self.name)
361
        elif self.type == "String":
362 363
            return "    ESX_VI__TEMPLATE__PROPERTY__SERIALIZE_VALUE(String, %s)\n" \
                   % self.name
364
        else:
365 366
            return "    ESX_VI__TEMPLATE__PROPERTY__SERIALIZE(%s, %s)\n" \
                   % (self.type, self.name)
367

368

369
    def generate_deserialize_code(self):
370
        if self.occurrence == OCCURRENCE__IGNORED:
371 372
            return "    ESX_VI__TEMPLATE__PROPERTY__DESERIALIZE_IGNORE(%s) /* FIXME */\n" \
                   % self.name
373 374
        elif self.occurrence in [OCCURRENCE__REQUIRED_LIST,
                                 OCCURRENCE__OPTIONAL_LIST]:
375 376
            return "    ESX_VI__TEMPLATE__PROPERTY__DESERIALIZE_LIST(%s, %s)\n" \
                   % (self.type, self.name)
377
        elif self.type == "String":
378 379
            return "    ESX_VI__TEMPLATE__PROPERTY__DESERIALIZE_VALUE(String, %s)\n" \
                   % self.name
380
        else:
381 382
            return "    ESX_VI__TEMPLATE__PROPERTY__DESERIALIZE(%s, %s)\n" \
                   % (self.type, self.name)
383

384

385 386
    def generate_lookup_code(self):
        if self.occurrence == OCCURRENCE__IGNORED:
387 388
            return "    ESX_VI__TEMPLATE__PROPERTY__CAST_FROM_ANY_TYPE_IGNORE(%s) /* FIXME */\n" \
                   % self.name
389 390
        elif self.occurrence in [OCCURRENCE__REQUIRED_LIST,
                                 OCCURRENCE__OPTIONAL_LIST]:
391 392
            return "    ESX_VI__TEMPLATE__PROPERTY__CAST_LIST_FROM_ANY_TYPE(%s, %s)\n" \
                   % (self.type, self.name)
393
        elif self.type == "String":
394 395
            return "    ESX_VI__TEMPLATE__PROPERTY__CAST_VALUE_FROM_ANY_TYPE(String, %s)\n" \
                   % self.name
396
        else:
397 398
            return "    ESX_VI__TEMPLATE__PROPERTY__CAST_FROM_ANY_TYPE(%s, %s)\n" \
                   % (self.type, self.name)
399 400


401 402
    def get_type_string(self):
        if self.type == "String" and \
403 404
           self.occurrence not in [OCCURRENCE__REQUIRED_LIST,
                                   OCCURRENCE__OPTIONAL_LIST]:
405 406 407 408 409 410 411 412
            return "char *"
        elif self.is_enum():
            return "esxVI_%s " % self.type
        else:
            return "esxVI_%s *" % self.type



413
class Type:
M
Matthias Bolte 已提交
414 415 416 417 418 419
    def __init__(self, kind, name):
        self.kind = kind
        self.name = name


    def generate_typedef(self):
420 421
        return "typedef %s _esxVI_%s esxVI_%s;\n" \
               % (self.kind, self.name, self.name)
M
Matthias Bolte 已提交
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442


    def generate_typeenum(self):
        return "    esxVI_Type_%s,\n" % self.name


    def generate_typetostring(self):
        string = "          case esxVI_Type_%s:\n" % self.name
        string += "            return \"%s\";\n\n" % self.name

        return string


    def generate_typefromstring(self):
        string =  "           else if (STREQ(type, \"%s\")) {\n" % self.name
        string += "               return esxVI_Type_%s;\n" % self.name
        string += "           }\n"

        return string


443
class GenericObject(Type):
444 445 446 447 448 449 450
    FEATURE__DYNAMIC_CAST = (1 << 1)
    FEATURE__LIST         = (1 << 2)
    FEATURE__DEEP_COPY    = (1 << 3)
    FEATURE__ANY_TYPE     = (1 << 4)
    FEATURE__SERIALIZE    = (1 << 5)
    FEATURE__DESERIALIZE  = (1 << 6)

451

452
    def __init__(self, name, category, managed, generic_objects_by_name):
453
        Type.__init__(self, "struct", name)
454 455 456 457 458 459 460 461 462 463 464 465 466 467
        self.category = category
        self.managed = managed
        self.generic_objects_by_name = generic_objects_by_name


    def generate_comment(self):
        comment = separator
        comment += " * %s: %s\n" % (self.category, self.name)

        if self.extends is not None:
            comment += " * %s  extends %s\n" \
                       % (' ' * len(self.category), self.extends)

        first = True
468

469
        if self.extended_by is not None:
470 471 472 473 474 475 476 477 478 479 480 481
            for extended_by in self.extended_by:
                if first:
                    comment += " * %s  extended by %s\n" \
                               % (' ' * len(self.category), extended_by)
                    first = False
                else:
                    comment += " * %s              %s\n" \
                               % (' ' * len(self.category), extended_by)

        comment += " */\n\n"

        return comment
482

483

M
Matthias Bolte 已提交
484
    def generate_struct_members(self, add_banner=False, struct_gap=False):
485 486
        members = ""

M
Matthias Bolte 已提交
487
        if struct_gap:
488 489 490
            members += "\n"

        if self.extends is not None:
491
            members += self.generic_objects_by_name[self.extends] \
492 493
                       .generate_struct_members(add_banner=True,
                                                struct_gap=False) + "\n"
494 495 496 497 498 499 500 501 502 503 504 505

        if self.extends is not None or add_banner:
            members += "    /* %s */\n" % self.name

        for property in self.properties:
            members += property.generate_struct_member()

        if len(self.properties) < 1:
            members += "    /* no properties */\n"

        return members

506

507 508 509 510 511 512 513 514 515 516 517 518 519 520
    def generate_dispatch(self, suffix, is_first=True):
        source = ""

        if self.extended_by is not None:
            if not is_first:
                source += "\n"

            source += "    /* %s */\n" % self.name

            for extended_by in self.extended_by:
                source += "    ESX_VI__TEMPLATE__DISPATCH__%s(%s)\n" \
                          % (suffix, extended_by)

            for extended_by in self.extended_by:
521
                source += self.generic_objects_by_name[extended_by] \
522 523 524 525 526
                          .generate_dispatch(suffix, False)

        return source


M
Matthias Bolte 已提交
527
    def generate_free_code(self, add_banner=False):
528 529 530
        source = ""

        if self.extends is not None:
531
            source += self.generic_objects_by_name[self.extends] \
532
                      .generate_free_code(add_banner=True) + "\n"
533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549 550 551

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            string = ""

            for property in self.properties:
                string += property.generate_free_code()

            if len(string) < 1:
                source += "    /* no properties to be freed */\n"
            else:
                source += string

        return source

552

M
Matthias Bolte 已提交
553
    def generate_validate_code(self, add_banner=False):
554 555 556
        source = ""

        if self.extends is not None:
557
            source += self.generic_objects_by_name[self.extends] \
558
                      .generate_validate_code(add_banner=True) + "\n"
559 560 561 562 563 564 565 566 567 568

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            string = ""

            for property in self.properties:
569
                string += property.generate_validate_code(self.managed)
570 571 572 573 574 575 576 577

            if len(string) < 1:
                source += "    /* no required properties */\n"
            else:
                source += string

        return source

578

579 580 581 582 583 584 585 586 587 588 589 590 591 592

class Object(GenericObject):
    def __init__(self, name, extends, properties, features=0, extended_by=None):
        GenericObject.__init__(self, name, 'VI Object', False, objects_by_name)
        self.extends = extends
        self.features = features
        self.properties = properties
        self.extended_by = extended_by
        self.candidate_for_dynamic_cast = False

        if self.extended_by is not None:
            self.extended_by.sort()


593
    def generate_dynamic_cast_code(self, is_first=True):
594 595 596
        source = ""

        if self.extended_by is not None:
597 598 599
            if not is_first:
                source += "\n"

600 601 602
            source += "    /* %s */\n" % self.name

            for extended_by in self.extended_by:
603 604
                source += "    ESX_VI__TEMPLATE__DYNAMIC_CAST__ACCEPT(%s)\n" \
                          % extended_by
605 606

            for extended_by in self.extended_by:
607 608
                source += objects_by_name[extended_by] \
                          .generate_dynamic_cast_code(False)
609 610 611

        return source

612

613 614 615 616
    def generate_deep_copy_code(self, add_banner = False):
        source = ""

        if self.extends is not None:
617 618
            source += objects_by_name[self.extends] \
                      .generate_deep_copy_code(add_banner=True) + "\n"
619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634 635 636 637

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            string = ""

            for property in self.properties:
                string += property.generate_deep_copy_code()

            if len(string) < 1:
                source += "    /* no properties to be deep copied */\n"
            else:
                source += string

        return source

638

639
    def generate_serialize_code(self, add_banner=False):
640 641 642
        source = ""

        if self.extends is not None:
643 644
            source += objects_by_name[self.extends] \
                      .generate_serialize_code(add_banner=True) + "\n"
645 646 647 648 649 650 651 652 653 654 655 656

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            for property in self.properties:
                source += property.generate_serialize_code()

        return source

657

658
    def generate_deserialize_code(self, add_banner=False):
659 660 661
        source = ""

        if self.extends is not None:
662 663
            source += objects_by_name[self.extends] \
                      .generate_deserialize_code(add_banner=True) + "\n"
664 665 666 667 668 669 670 671 672 673 674 675

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            for property in self.properties:
                source += property.generate_deserialize_code()

        return source

676

677
    def generate_header(self):
678
        header = self.generate_comment()
679 680 681 682 683

        # struct
        header += "struct _esxVI_%s {\n" % self.name

        if self.features & Object.FEATURE__LIST:
684 685
            header += aligned("    esxVI_%s *_next; " % self.name,
                              "/* optional */\n")
686
        else:
687 688
            header += aligned("    esxVI_%s *_unused; " % self.name,
                              "/* optional */\n")
689

M
Matthias Bolte 已提交
690 691
        header += aligned("    esxVI_Type _type; ", "/* required */\n")
        header += self.generate_struct_members(struct_gap=True)
692 693 694
        header += "};\n\n"

        # functions
695 696 697 698 699 700
        header += "int esxVI_%s_Alloc(esxVI_%s **item);\n" \
                  % (self.name, self.name)
        header += "void esxVI_%s_Free(esxVI_%s **item);\n" \
                  % (self.name, self.name)
        header += "int esxVI_%s_Validate(esxVI_%s *item);\n" \
                  % (self.name, self.name)
701 702 703

        if self.features & Object.FEATURE__DYNAMIC_CAST:
            if self.extended_by is not None or self.extends is not None:
704 705
                header += "esxVI_%s *esxVI_%s_DynamicCast(void *item);\n" \
                          % (self.name, self.name)
706 707 708 709
            else:
                report_error("cannot add dynamic cast support for an untyped object")

        if self.features & Object.FEATURE__LIST:
710 711
            header += "int esxVI_%s_AppendToList(esxVI_%s **list, esxVI_%s *item);\n" \
                      % (self.name, self.name, self.name)
712 713

        if self.features & Object.FEATURE__DEEP_COPY:
714 715
            header += "int esxVI_%s_DeepCopy(esxVI_%s **dst, esxVI_%s *src);\n" \
                      % (self.name, self.name, self.name)
716 717

            if self.features & Object.FEATURE__LIST:
718 719 720
                header += ("int esxVI_%s_DeepCopyList(esxVI_%s **dstList, "
                                                     "esxVI_%s *srcList);\n") \
                          % (self.name, self.name, self.name)
721 722

        if self.features & Object.FEATURE__ANY_TYPE:
723 724 725
            header += ("int esxVI_%s_CastFromAnyType(esxVI_AnyType *anyType, "
                                                    "esxVI_%s **item);\n") \
                      % (self.name, self.name)
726 727

            if self.features & Object.FEATURE__LIST:
728 729 730
                header += ("int esxVI_%s_CastListFromAnyType(esxVI_AnyType *anyType, "
                                                            "esxVI_%s **list);\n") \
                          % (self.name, self.name)
731 732

        if self.features & Object.FEATURE__SERIALIZE:
733 734 735 736
            header += ("int esxVI_%s_Serialize(esxVI_%s *item, "
                                              "const char *element, "
                                              "virBufferPtr output);\n") \
                      % (self.name, self.name)
737 738

            if self.features & Object.FEATURE__LIST:
739 740 741 742
                header += ("int esxVI_%s_SerializeList(esxVI_%s *list, "
                                                      "const char *element, "
                                                      "virBufferPtr output);\n") \
                          % (self.name, self.name)
743 744

        if self.features & Object.FEATURE__DESERIALIZE:
745 746
            header += "int esxVI_%s_Deserialize(xmlNodePtr node, esxVI_%s **item);\n" \
                      % (self.name, self.name)
747 748

            if self.features & Object.FEATURE__LIST:
749 750 751
                header += ("int esxVI_%s_DeserializeList(xmlNodePtr node, "
                                                        "esxVI_%s **list);\n") \
                          % (self.name, self.name)
752 753 754 755 756

        header += "\n\n\n"

        return header

757

758
    def generate_source(self):
759
        source = separator
760
        source += " * VI Object: %s\n" % self.name
761 762

        if self.extends is not None:
763
            source += " *            extends %s\n" % self.extends
764 765 766 767 768 769

        first = True

        if self.extended_by is not None:
            for extended_by in self.extended_by:
                if first:
770
                    source += " *            extended by %s\n" % extended_by
771 772
                    first = False
                else:
773
                    source += " *                        %s\n" % extended_by
774 775 776 777 778 779 780 781

        source += " */\n\n"

        # functions
        source += "/* esxVI_%s_Alloc */\n" % self.name
        source += "ESX_VI__TEMPLATE__ALLOC(%s)\n\n" % self.name

        # free
782 783
        source += "/* esxVI_%s_Free */\n" % self.name

784 785 786 787 788
        if self.extended_by is None:
            source += "ESX_VI__TEMPLATE__FREE(%s,\n" % self.name
        else:
            source += "ESX_VI__TEMPLATE__DYNAMIC_FREE(%s,\n" % self.name
            source += "{\n"
789
            source += self.generate_dispatch('FREE')
790 791
            source += "},\n"

792
        source += "{\n"
793

794 795 796 797 798 799 800 801 802 803
        if self.features & Object.FEATURE__LIST:
            if self.extends is not None:
                # avoid "dereferencing type-punned pointer will break
                # strict-aliasing rules" warnings
                source += "    esxVI_%s *next = (esxVI_%s *)item->_next;\n\n" \
                          % (self.extends, self.extends)
                source += "    esxVI_%s_Free(&next);\n" % self.extends
                source += "    item->_next = (esxVI_%s *)next;\n\n" % self.name
            else:
                source += "    esxVI_%s_Free(&item->_next);\n\n" % self.name
804

805 806
        source += self.generate_free_code()
        source += "})\n\n"
807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831

        # validate
        source += "/* esxVI_%s_Validate */\n" % self.name
        source += "ESX_VI__TEMPLATE__VALIDATE(%s,\n" % self.name
        source += "{\n"
        source += self.generate_validate_code()
        source += "})\n\n"

        # dynamic cast
        if self.features & Object.FEATURE__DYNAMIC_CAST:
            if self.extended_by is not None or self.extends is not None:
                source += "/* esxVI_%s_DynamicCast */\n" % self.name
                source += "ESX_VI__TEMPLATE__DYNAMIC_CAST(%s,\n" % self.name
                source += "{\n"
                source += self.generate_dynamic_cast_code()
                source += "})\n\n"
            else:
                report_error("cannot add dynamic cast support for an untyped object")

        # append to list
        if self.features & Object.FEATURE__LIST:
            source += "/* esxVI_%s_AppendToList */\n" % self.name
            source += "ESX_VI__TEMPLATE__LIST__APPEND(%s)\n\n" % self.name

        # deep copy
832 833
        if self.features & Object.FEATURE__DEEP_COPY:
            source += "/* esxVI_%s_DeepCopy */\n" % self.name
834

835 836 837
            if self.extended_by is None:
                source += "ESX_VI__TEMPLATE__DEEP_COPY(%s,\n" % self.name
            else:
838
                source += "ESX_VI__TEMPLATE__DYNAMIC_DEEP_COPY(%s,\n" % self.name
839
                source += "{\n"
840
                source += self.generate_dispatch('DEEP_COPY')
841 842
                source += "},\n"

843 844 845
            source += "{\n"
            source += self.generate_deep_copy_code()
            source += "})\n\n"
846

847 848 849 850
            if self.features & Object.FEATURE__LIST:
                source += "/* esxVI_%s_DeepCopyList */\n" % self.name
                source += "ESX_VI__TEMPLATE__LIST__DEEP_COPY(%s)\n\n" \
                          % self.name
851 852 853 854

        # cast from any type
        if self.features & Object.FEATURE__ANY_TYPE:
            source += "/* esxVI_%s_CastFromAnyType */\n" % self.name
855 856

            if self.extended_by is None:
857 858
                source += "ESX_VI__TEMPLATE__CAST_FROM_ANY_TYPE(%s)\n\n" \
                          % self.name
859
            else:
860 861
                source += "ESX_VI__TEMPLATE__DYNAMIC_CAST_FROM_ANY_TYPE(%s,\n" \
                          % self.name
862
                source += "{\n"
863
                source += self.generate_dispatch('CAST_FROM_ANY_TYPE')
864
                source += "})\n\n"
865 866 867

            if self.features & Object.FEATURE__LIST:
                source += "/* esxVI_%s_CastListFromAnyType */\n" % self.name
868 869
                source += "ESX_VI__TEMPLATE__LIST__CAST_FROM_ANY_TYPE(%s)\n\n" \
                          % self.name
870 871

        # serialize
872 873
        if self.features & Object.FEATURE__SERIALIZE:
            source += "/* esxVI_%s_Serialize */\n" % self.name
874

875 876 877
            if self.extended_by is None:
                source += "ESX_VI__TEMPLATE__SERIALIZE(%s,\n" % self.name
            else:
878 879
                source += "ESX_VI__TEMPLATE__DYNAMIC_SERIALIZE(%s,\n" % self.name
                source += "{\n"
880
                source += self.generate_dispatch('SERIALIZE')
881 882
                source += "},\n"

883 884 885
            source += "{\n"
            source += self.generate_serialize_code()
            source += "})\n\n"
886

887 888 889 890
            if self.features & Object.FEATURE__LIST:
                source += "/* esxVI_%s_SerializeList */\n" % self.name
                source += "ESX_VI__TEMPLATE__LIST__SERIALIZE(%s)\n\n" \
                          % self.name
891

892
        # deserialize
893 894
        if self.features & Object.FEATURE__DESERIALIZE:
            source += "/* esxVI_%s_Deserialize */\n" % self.name
895

896 897 898
            if self.extended_by is None:
                source += "ESX_VI__TEMPLATE__DESERIALIZE(%s,\n" % self.name
            else:
899 900
                source += "ESX_VI__TEMPLATE__DYNAMIC_DESERIALIZE(%s,\n" \
                          % self.name
901
                source += "{\n"
902
                source += self.generate_dispatch('DESERIALIZE')
903 904
                source += "},\n"

905 906 907
            source += "{\n"
            source += self.generate_deserialize_code()
            source += "})\n\n"
908

909 910 911 912
            if self.features & Object.FEATURE__LIST:
                source += "/* esxVI_%s_DeserializeList */\n" % self.name
                source += "ESX_VI__TEMPLATE__LIST__DESERIALIZE(%s)\n\n" \
                          % self.name
913 914 915 916 917 918 919

        source += "\n\n"

        return source



920
class ManagedObject(GenericObject):
921
    def __init__(self, name, extends, properties, features=0, extended_by=None):
922 923
        GenericObject.__init__(self, name, 'VI Managed Object', True,
                               managed_objects_by_name)
924 925 926 927 928 929 930 931 932 933 934 935 936
        self.extends = extends
        self.features = features
        self.properties = properties
        self.extended_by = extended_by

        if self.extended_by is not None:
            self.extended_by.sort()


    def generate_lookup_code1(self, add_banner=False):
        source = ""

        if self.extends is not None:
937 938
            source += managed_objects_by_name[self.extends] \
                      .generate_lookup_code1(add_banner=True) + "\n"
939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957 958 959 960 961 962

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            string = ""

            for property in self.properties:
                string += "    \"%s\\0\"\n" % property.name

            if len(string) < 1:
                source += "    /* no properties */\n"
            else:
                source += string

        return source


    def generate_lookup_code2(self, add_banner=False):
        source = ""

        if self.extends is not None:
963 964
            source += managed_objects_by_name[self.extends] \
                      .generate_lookup_code2(add_banner=True) + "\n"
965 966 967 968 969 970 971 972 973 974 975 976 977 978 979 980 981 982 983 984 985 986 987 988 989 990 991

        if self.extends is not None or add_banner:
            source += "    /* %s */\n" % self.name

        if len(self.properties) < 1:
            source += "    /* no properties */\n"
        else:
            string = ""

            for property in self.properties:
                string += property.generate_lookup_code()

            if len(string) < 1:
                source += "    /* no properties */\n"
            else:
                source += string

        return source


    def generate_header(self):
        header = self.generate_comment()

        # struct
        header += "struct _esxVI_%s {\n" % self.name

        if self.features & Object.FEATURE__LIST:
992 993
            header += aligned("    esxVI_%s *_next; " % self.name,
                              "/* optional */\n")
994
        else:
995 996
            header += aligned("    esxVI_%s *_unused; " % self.name,
                              "/* optional */\n")
997 998

        header += aligned("    esxVI_Type _type; ", "/* required */\n")
999 1000
        header += aligned("    esxVI_ManagedObjectReference *_reference; ",
                          "/* required */\n")
1001 1002 1003 1004 1005 1006 1007
        header += "\n"
        header += self.generate_struct_members()
        header += "};\n\n"

        # functions
        header += "int esxVI_%s_Alloc(esxVI_%s **item);\n" % (self.name, self.name)
        header += "void esxVI_%s_Free(esxVI_%s **item);\n" % (self.name, self.name)
1008 1009 1010
        header += ("int esxVI_%s_Validate(esxVI_%s *item, "
                                         "esxVI_String *selectedPropertyNameList);\n") \
                  % (self.name, self.name)
1011 1012

        if self.features & Object.FEATURE__LIST:
1013 1014
            header += "int esxVI_%s_AppendToList(esxVI_%s **list, esxVI_%s *item);\n" \
                      % (self.name, self.name, self.name)
1015 1016 1017 1018 1019 1020 1021 1022 1023 1024

        header += "\n\n\n"

        return header


    def generate_helper_header(self):
        header = ""

        # functions
1025 1026 1027 1028 1029 1030 1031
        header += ("int esxVI_Lookup%s(esxVI_Context *ctx, "
                                      "const char *name, "
                                      "esxVI_ManagedObjectReference *root, "
                                      "esxVI_String *selectedPropertyNameList, "
                                      "esxVI_%s **item, "
                                      "esxVI_Occurrence occurrence);\n") \
                  % (self.name, self.name)
1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045

        header += "\n"

        return header


    def generate_source(self):
        source = self.generate_comment()

        # functions
        source += "/* esxVI_%s_Alloc */\n" % self.name
        source += "ESX_VI__TEMPLATE__ALLOC(%s)\n\n" % self.name

        # free
1046 1047
        source += "/* esxVI_%s_Free */\n" % self.name

1048 1049 1050 1051 1052
        if self.extended_by is None:
            source += "ESX_VI__TEMPLATE__FREE(%s,\n" % self.name
        else:
            source += "ESX_VI__TEMPLATE__DYNAMIC_FREE(%s,\n" % self.name
            source += "{\n"
1053
            source += self.generate_dispatch('FREE')
1054 1055
            source += "},\n"

1056
        source += "{\n"
1057

1058 1059 1060 1061 1062 1063 1064 1065 1066 1067
        if self.features & ManagedObject.FEATURE__LIST:
            if self.extends is not None:
                # avoid "dereferencing type-punned pointer will break
                # strict-aliasing rules" warnings
                source += "    esxVI_%s *next = (esxVI_%s *)item->_next;\n\n" \
                          % (self.extends, self.extends)
                source += "    esxVI_%s_Free(&next);\n" % self.extends
                source += "    item->_next = (esxVI_%s *)next;\n\n" % self.name
            else:
                source += "    esxVI_%s_Free(&item->_next);\n" % self.name
1068

1069 1070 1071
        source += "    esxVI_ManagedObjectReference_Free(&item->_reference);\n\n"
        source += self.generate_free_code()
        source += "})\n\n"
1072 1073 1074 1075 1076 1077 1078 1079 1080 1081 1082 1083 1084 1085 1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111

        # validate
        source += "/* esxVI_%s_Validate */\n" % self.name
        source += "ESX_VI__TEMPLATE__MANAGED_VALIDATE(%s,\n" % self.name
        source += "{\n"

        source += self.generate_validate_code()

        source += "})\n\n"

        # append to list
        if self.features & ManagedObject.FEATURE__LIST:
            source += "/* esxVI_%s_AppendToList */\n" % self.name
            source += "ESX_VI__TEMPLATE__LIST__APPEND(%s)\n\n" % self.name

        source += "\n\n"

        return source


    def generate_helper_source(self):
        source = ""

        # lookup
        source += "/* esxVI_Lookup%s */\n" % self.name
        source += "ESX_VI__TEMPLATE__LOOKUP(%s,\n" % self.name
        source += "{\n"

        source += self.generate_lookup_code1()

        source += "},\n"
        source += "{\n"

        source += self.generate_lookup_code2()

        source += "})\n\n"

        source += "\n\n"

        return source
1112

1113

1114

1115
class Enum(Type):
M
Matthias Bolte 已提交
1116 1117 1118 1119 1120 1121
    FEATURE__ANY_TYPE = (1 << 1)
    FEATURE__SERIALIZE = (1 << 2)
    FEATURE__DESERIALIZE = (1 << 3)


    def __init__(self, name, values, features=0):
1122
        Type.__init__(self, "enum", name)
M
Matthias Bolte 已提交
1123
        self.values = values
1124
        self.features = features
1125

1126

1127
    def generate_header(self):
1128
        header = separator
1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140
        header += " * VI Enum: %s\n" % self.name
        header += " */\n\n"
        header += "enum _esxVI_%s {\n" % self.name
        header += "    esxVI_%s_Undefined = 0,\n" % self.name

        for value in self.values:
            header += "    esxVI_%s_%s,\n" % (self.name, capitalize_first(value))

        header += "};\n\n"

        # functions
        if self.features & Enum.FEATURE__ANY_TYPE:
1141 1142 1143
            header += ("int esxVI_%s_CastFromAnyType(esxVI_AnyType *anyType, "
                                                    "esxVI_%s *item);\n") \
                      % (self.name, self.name)
1144 1145

        if self.features & Enum.FEATURE__SERIALIZE:
1146 1147 1148
            header += ("int esxVI_%s_Serialize(esxVI_%s item, const char *element, "
                                              "virBufferPtr output);\n") \
                      % (self.name, self.name)
1149 1150

        if self.features & Enum.FEATURE__DESERIALIZE:
1151 1152 1153
            header += ("int esxVI_%s_Deserialize(xmlNodePtr node, "
                                                "esxVI_%s *item);\n") \
                      % (self.name, self.name)
1154 1155 1156 1157 1158

        header += "\n\n\n"

        return header

1159

1160
    def generate_source(self):
1161
        source = separator
1162 1163
        source += " * VI Enum: %s\n" % self.name
        source += " */\n\n"
1164 1165
        source += "static const esxVI_Enumeration _esxVI_%s_Enumeration = {\n" \
                  % self.name
1166 1167 1168
        source += "    esxVI_Type_%s, {\n" % self.name

        for value in self.values:
1169 1170
            source += "        { \"%s\", esxVI_%s_%s },\n" \
                      % (value, self.name, capitalize_first(value))
1171 1172 1173 1174 1175 1176 1177 1178

        source += "        { NULL, -1 },\n"
        source += "    },\n"
        source += "};\n\n"

        # functions
        if self.features & Enum.FEATURE__ANY_TYPE:
            source += "/* esxVI_%s_CastFromAnyType */\n" % self.name
1179 1180
            source += "ESX_VI__TEMPLATE__ENUMERATION__CAST_FROM_ANY_TYPE(%s)\n\n" \
                      % self.name
1181 1182 1183

        if self.features & Enum.FEATURE__SERIALIZE:
            source += "/* esxVI_%s_Serialize */\n" % self.name
1184 1185
            source += "ESX_VI__TEMPLATE__ENUMERATION__SERIALIZE(%s)\n\n" \
                      % self.name
1186 1187 1188

        if self.features & Enum.FEATURE__DESERIALIZE:
            source += "/* esxVI_%s_Deserialize */\n" % self.name
1189 1190
            source += "ESX_VI__TEMPLATE__ENUMERATION__DESERIALIZE(%s)\n\n" \
                      % self.name
1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205 1206 1207 1208 1209

        source += "\n\n"

        return source



def report_error(message):
    print "error: " + message
    sys.exit(1)



def capitalize_first(string):
    return string[:1].upper() + string[1:]



def parse_object(block):
1210
    # expected format: [managed] object <name> [extends <name>]
1211
    header_items = block[0][1].split()
1212 1213 1214 1215 1216
    managed = False

    if header_items[0] == "managed":
        managed = True
        del header_items[0]
1217 1218 1219 1220 1221 1222 1223 1224 1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235 1236 1237 1238 1239 1240

    if len(header_items) < 2:
        report_error("line %d: invalid block header" % (number))

    assert header_items[0] == "object"

    name = header_items[1]
    extends = None

    if len(header_items) > 2:
        if header_items[2] != "extends":
            report_error("line %d: invalid block header" % (number))
        else:
            extends = header_items[3]

    properties = []

    for line in block[1:]:
        # expected format: <type> <name> <occurrence>
        items = line[1].split()

        if len(items) != 3:
            report_error("line %d: invalid property" % line[0])

1241
        if items[2] not in valid_occurrences:
1242 1243
            report_error("line %d: invalid occurrence" % line[0])

M
Matthias Bolte 已提交
1244 1245
        properties.append(Property(type=items[0], name=items[1],
                                   occurrence=items[2]))
1246

1247 1248 1249 1250
    if managed:
        return ManagedObject(name=name, extends=extends, properties=properties)
    else:
        return Object(name=name, extends=extends, properties=properties)
1251 1252 1253 1254 1255 1256 1257 1258 1259 1260 1261 1262 1263 1264 1265 1266 1267 1268 1269 1270



def parse_enum(block):
    # expected format: enum <name>
    header_items = block[0][1].split()

    if len(header_items) < 2:
        report_error("line %d: invalid block header" % (number))

    assert header_items[0] == "enum"

    name = header_items[1]

    values = []

    for line in block[1:]:
        # expected format: <value>
        values.append(line[1])

M
Matthias Bolte 已提交
1271
    return Enum(name=name, values=values)
1272 1273 1274



1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287 1288 1289 1290
def parse_method(block):
    # expected format: method <name> [returns <type> <occurrence>]
    header_items = block[0][1].split()

    if len(header_items) < 2:
        report_error("line %d: invalid block header" % (number))

    assert header_items[0] == "method"

    name = header_items[1]
    returns = None

    if len(header_items) > 2:
        if header_items[2] != "returns":
            report_error("line %d: invalid block header" % (number))
        else:
M
Matthias Bolte 已提交
1291 1292
            returns = Parameter(type=header_items[3], name="output",
                                occurrence=header_items[4])
1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305

    parameters = []

    for line in block[1:]:
        # expected format: <type> <name> <occurrence>
        items = line[1].split()

        if len(items) != 3:
            report_error("line %d: invalid property" % line[0])

        if items[2] not in valid_occurrences:
            report_error("line %d: invalid occurrence" % line[0])

M
Matthias Bolte 已提交
1306 1307
        parameters.append(Parameter(type=items[0], name=items[1],
                                    occurrence=items[2]))
1308

M
Matthias Bolte 已提交
1309
    return Method(name=name, parameters=parameters, returns=returns)
1310 1311 1312



1313 1314 1315 1316
def is_known_type(type):
    return type in predefined_objects or \
           type in predefined_enums or \
           type in objects_by_name or \
1317
           type in managed_objects_by_name or \
1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334
           type in enums_by_name



def open_and_print(filename):
    if filename.startswith("./"):
        print "  GEN    " + filename[2:]
    else:
        print "  GEN    " + filename

    return open(filename, "wb")



predefined_enums = ["Boolean"]

predefined_objects = ["AnyType",
A
Ata E Husain Bohra 已提交
1335
                      "Byte",
1336 1337 1338 1339
                      "Int",
                      "Long",
                      "String",
                      "DateTime",
1340
                      "MethodFault",
1341 1342 1343 1344 1345 1346
                      "ManagedObjectReference"]

additional_enum_features = { "ManagedEntityStatus"      : Enum.FEATURE__ANY_TYPE,
                             "TaskInfoState"            : Enum.FEATURE__ANY_TYPE,
                             "VirtualMachinePowerState" : Enum.FEATURE__ANY_TYPE }

1347
additional_object_features = { "AutoStartDefaults"          : Object.FEATURE__ANY_TYPE,
1348
                               "AutoStartPowerInfo"         : Object.FEATURE__ANY_TYPE,
1349 1350 1351 1352 1353
                               "DatastoreHostMount"         : Object.FEATURE__DEEP_COPY |
                                                              Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
                               "DatastoreInfo"              : Object.FEATURE__ANY_TYPE |
                                                              Object.FEATURE__DYNAMIC_CAST,
1354
                               "HostConfigManager"          : Object.FEATURE__ANY_TYPE,
1355 1356 1357 1358
                               "HostCpuIdInfo"              : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
                               "HostDatastoreBrowserSearchResults" : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
1359 1360 1361 1362 1363 1364 1365 1366 1367 1368 1369 1370 1371 1372 1373
                               "HostHostBusAdapter"         : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
                               "HostInternetScsiHba"        : Object.FEATURE__DYNAMIC_CAST |
                                                              Object.FEATURE__DEEP_COPY,
                               "HostInternetScsiTargetTransport"  : Object.FEATURE__DYNAMIC_CAST,
                               "HostScsiDisk"               : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE |
                                                              Object.FEATURE__DYNAMIC_CAST,
                               "HostScsiTopologyInterface"  : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
                               "HostScsiTopologyLun"        : Object.FEATURE__ANY_TYPE |
                                                              Object.FEATURE__LIST |
                                                              Object.FEATURE__DEEP_COPY,
                               "HostScsiTopologyTarget"     : Object.FEATURE__ANY_TYPE |
                                                              Object.FEATURE__LIST,
M
Matthias Bolte 已提交
1374 1375 1376 1377 1378
                               "HostPortGroup"              : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
                               "HostVirtualSwitch"          : Object.FEATURE__DEEP_COPY |
                                                              Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
1379
                               "ManagedObjectReference"     : Object.FEATURE__ANY_TYPE,
1380
                               "ObjectContent"              : Object.FEATURE__DEEP_COPY,
M
Matthias Bolte 已提交
1381 1382 1383
                               "PhysicalNic"                : Object.FEATURE__DEEP_COPY |
                                                              Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
1384
                               "ResourcePoolResourceUsage"  : Object.FEATURE__ANY_TYPE,
1385 1386 1387 1388
                               "ScsiLun"                    : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE |
                                                              Object.FEATURE__DEEP_COPY,
                               "ScsiLunDurableName"         : Object.FEATURE__LIST,
1389
                               "ServiceContent"             : Object.FEATURE__DESERIALIZE,
1390
                               "SharesInfo"                 : Object.FEATURE__ANY_TYPE,
1391 1392
                               "TaskInfo"                   : Object.FEATURE__LIST |
                                                              Object.FEATURE__ANY_TYPE,
1393
                               "UserSession"                : Object.FEATURE__ANY_TYPE,
1394
                               "VirtualMachineQuestionInfo" : Object.FEATURE__ANY_TYPE,
1395 1396
                               "VirtualMachineSnapshotTree" : Object.FEATURE__DEEP_COPY |
                                                              Object.FEATURE__ANY_TYPE,
1397
                               "VmEventArgument"            : Object.FEATURE__DESERIALIZE }
1398

1399
removed_object_features = {}
1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411



if "srcdir" in os.environ:
    input_filename = os.path.join(os.environ["srcdir"], "esx/esx_vi_generator.input")
    output_dirname = os.path.join(os.environ["srcdir"], "esx")
else:
    input_filename = os.path.join(os.getcwd(), "esx_vi_generator.input")
    output_dirname = os.getcwd()



1412 1413 1414 1415 1416 1417 1418 1419
types_typedef = open_and_print(os.path.join(output_dirname, "esx_vi_types.generated.typedef"))
types_typeenum = open_and_print(os.path.join(output_dirname, "esx_vi_types.generated.typeenum"))
types_typetostring = open_and_print(os.path.join(output_dirname, "esx_vi_types.generated.typetostring"))
types_typefromstring = open_and_print(os.path.join(output_dirname, "esx_vi_types.generated.typefromstring"))
types_header = open_and_print(os.path.join(output_dirname, "esx_vi_types.generated.h"))
types_source = open_and_print(os.path.join(output_dirname, "esx_vi_types.generated.c"))
methods_header = open_and_print(os.path.join(output_dirname, "esx_vi_methods.generated.h"))
methods_source = open_and_print(os.path.join(output_dirname, "esx_vi_methods.generated.c"))
1420
methods_macro = open_and_print(os.path.join(output_dirname, "esx_vi_methods.generated.macro"))
1421 1422
helpers_header = open_and_print(os.path.join(output_dirname, "esx_vi.generated.h"))
helpers_source = open_and_print(os.path.join(output_dirname, "esx_vi.generated.c"))
1423 1424 1425 1426 1427



number = 0
objects_by_name = {}
1428
managed_objects_by_name = {}
1429
enums_by_name = {}
1430
methods_by_name = {}
1431 1432 1433 1434
block = None



1435
# parse input file
1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446
for line in file(input_filename, "rb").readlines():
    number += 1

    if "#" in line:
        line = line[:line.index("#")]

    line = line.lstrip().rstrip()

    if len(line) < 1:
        continue

1447 1448
    if line.startswith("object") or line.startswith("managed object") or \
       line.startswith("enum") or line.startswith("method"):
1449 1450 1451 1452 1453 1454 1455 1456 1457 1458
        if block is not None:
            report_error("line %d: nested block found" % (number))
        else:
            block = []

    if block is not None:
        if line == "end":
            if block[0][1].startswith("object"):
                obj = parse_object(block)
                objects_by_name[obj.name] = obj
1459 1460 1461
            elif block[0][1].startswith("managed object"):
                obj = parse_object(block)
                managed_objects_by_name[obj.name] = obj
1462
            elif block[0][1].startswith("enum"):
1463 1464
                enum = parse_enum(block)
                enums_by_name[enum.name] = enum
1465 1466 1467
            else:
                method = parse_method(block)
                methods_by_name[method.name] = method
1468 1469 1470 1471 1472 1473 1474

            block = None
        else:
            block.append((number, line))



1475 1476 1477 1478 1479 1480 1481 1482 1483 1484
for method in methods_by_name.values():
    # method parameter types must be serializable
    for parameter in method.parameters:
        if not parameter.is_type_generated():
            continue

        if parameter.is_enum():
            enums_by_name[parameter.type].features |= Enum.FEATURE__SERIALIZE
        else:
            objects_by_name[parameter.type].features |= Object.FEATURE__SERIALIZE
1485
            objects_by_name[parameter.type].candidate_for_dynamic_cast = True
1486

1487 1488 1489 1490 1491 1492 1493 1494 1495
        # detect list usage
        if parameter.occurrence == OCCURRENCE__REQUIRED_LIST or \
           parameter.occurrence == OCCURRENCE__OPTIONAL_LIST:
            if parameter.is_enum():
                report_error("unsupported usage of enum '%s' as list in '%s'"
                             % (parameter.type, method.name))
            else:
                objects_by_name[parameter.type].features |= Object.FEATURE__LIST

1496 1497 1498 1499 1500 1501
    # method return types must be deserializable
    if method.returns and method.returns.is_type_generated():
        if method.returns.is_enum():
            enums_by_name[method.returns.type].features |= Enum.FEATURE__DESERIALIZE
        else:
            objects_by_name[method.returns.type].features |= Object.FEATURE__DESERIALIZE
1502
            objects_by_name[method.returns.type].candidate_for_dynamic_cast = True
1503

1504 1505 1506 1507 1508 1509 1510 1511 1512
        # detect list usage
        if method.returns.occurrence == OCCURRENCE__REQUIRED_LIST or \
           method.returns.occurrence == OCCURRENCE__OPTIONAL_LIST:
            if method.returns.is_enum():
                report_error("unsupported usage of enum '%s' as list in '%s'"
                             % (method.returns.type, method.name))
            else:
                objects_by_name[method.returns.type].features |= Object.FEATURE__LIST

1513 1514


1515 1516 1517 1518 1519
for enum in enums_by_name.values():
    # apply additional features
    if enum.name in additional_enum_features:
        enum.features |= additional_enum_features[enum.name]

1520 1521 1522
        if additional_enum_features[enum.name] & Enum.FEATURE__ANY_TYPE:
            enum.features |= Enum.FEATURE__DESERIALIZE

1523 1524 1525 1526


for obj in objects_by_name.values():
    for property in obj.properties:
1527
        if property.occurrence != OCCURRENCE__IGNORED and \
1528
           not is_known_type(property.type):
1529 1530
            report_error("object '%s' contains unknown property type '%s'"
                         % (obj.name, property.type))
1531 1532 1533

    if obj.extends is not None:
        if not is_known_type(obj.extends):
1534 1535
            report_error("object '%s' extends unknown object '%s'"
                         % (obj.name, obj.extends))
1536 1537

    for property in obj.properties:
1538 1539 1540
        if not property.is_type_generated():
            continue

1541 1542 1543 1544 1545 1546
        if property.is_enum():
            enums_by_name[property.type].candidate_for_dynamic_cast = True
        else:
            objects_by_name[property.type].candidate_for_dynamic_cast = True

        # detect list usage
1547 1548 1549 1550 1551 1552 1553
        if property.occurrence == OCCURRENCE__REQUIRED_LIST or \
           property.occurrence == OCCURRENCE__OPTIONAL_LIST:
            if property.is_enum():
                report_error("unsupported usage of enum '%s' as list in '%s'"
                             % (property.type, obj.type))
            else:
                objects_by_name[property.type].features |= Object.FEATURE__LIST
1554 1555 1556 1557 1558

    # apply/remove additional features
    if obj.name in additional_object_features:
        obj.features |= additional_object_features[obj.name]

1559 1560 1561
        if additional_object_features[obj.name] & Object.FEATURE__ANY_TYPE:
            obj.features |= Object.FEATURE__DESERIALIZE

1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572
    if obj.name in removed_object_features:
        obj.features &= ~removed_object_features[obj.name]

    # detect extended_by relation
    if obj.extends is not None:
        extended_obj = objects_by_name[obj.extends]

        if extended_obj.extended_by is None:
            extended_obj.extended_by = [obj.name]
        else:
            extended_obj.extended_by.append(obj.name)
1573
            extended_obj.extended_by.sort()
1574 1575 1576



1577 1578 1579 1580 1581 1582 1583 1584 1585
for obj in objects_by_name.values():
    # if an object is a candidate (it is used directly as parameter or return
    # type or is a member of another object) and it is extended by another
    # object then this type needs the dynamic cast feature
    if obj.candidate_for_dynamic_cast and obj.extended_by:
        obj.features |= Object.FEATURE__DYNAMIC_CAST



1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636 1637 1638 1639 1640 1641 1642 1643 1644 1645 1646 1647 1648 1649 1650 1651 1652 1653 1654 1655 1656 1657 1658 1659
def propagate_feature(obj, feature):
    global features_have_changed

    if not (obj.features & feature):
        return

    for property in obj.properties:
        if property.occurrence == OCCURRENCE__IGNORED or \
           not property.is_type_generated():
            continue

        if property.is_enum():
            if feature == Object.FEATURE__SERIALIZE and \
               not (enums_by_name[property.type].features & Enum.FEATURE__SERIALIZE):
                enums_by_name[property.type].features |= Enum.FEATURE__SERIALIZE
                features_have_changed = True
            elif feature == Object.FEATURE__DESERIALIZE and \
               not (enums_by_name[property.type].features & Enum.FEATURE__DESERIALIZE):
                enums_by_name[property.type].features |= Enum.FEATURE__DESERIALIZE
                features_have_changed = True
        elif property.is_object():
            if not (objects_by_name[property.type].features & feature):
                objects_by_name[property.type].features |= feature
                features_have_changed = True

            if obj.name != property.type:
                propagate_feature(objects_by_name[property.type], feature)



def inherit_features(obj):
    global features_have_changed

    if obj.extended_by is not None:
        for extended_by in obj.extended_by:
            previous = objects_by_name[extended_by].features
            objects_by_name[extended_by].features |= obj.features

            if objects_by_name[extended_by].features != previous:
                features_have_changed = True

    if obj.extends is not None:
        previous = objects_by_name[obj.extends].features
        objects_by_name[obj.extends].features |= obj.features

        if objects_by_name[obj.extends].features != previous:
            features_have_changed = True

    if obj.extended_by is not None:
        for extended_by in obj.extended_by:
            inherit_features(objects_by_name[extended_by])



# there are two directions to spread features:
# 1) up and down the inheritance chain
# 2) from object types to their member property types
# spreading needs to be done alternating on both directions because they can
# affect each other
features_have_changed = True

while features_have_changed:
    features_have_changed = False

    for obj in objects_by_name.values():
        propagate_feature(obj, Object.FEATURE__DEEP_COPY)
        propagate_feature(obj, Object.FEATURE__SERIALIZE)
        propagate_feature(obj, Object.FEATURE__DESERIALIZE)

    for obj in objects_by_name.values():
        inherit_features(obj)



1660 1661 1662 1663
for obj in managed_objects_by_name.values():
    for property in obj.properties:
        if property.occurrence != OCCURRENCE__IGNORED and \
           not is_known_type(property.type):
1664 1665
            report_error("object '%s' contains unknown property type '%s'"
                         % (obj.name, property.type))
1666 1667 1668

    if obj.extends is not None:
        if not is_known_type(obj.extends):
1669 1670
            report_error("object '%s' extends unknown object '%s'"
                         % (obj.name, obj.extends))
1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683

    # detect extended_by relation
    if obj.extends is not None:
        extended_obj = managed_objects_by_name[obj.extends]

        if extended_obj.extended_by is None:
            extended_obj.extended_by = [obj.name]
        else:
            extended_obj.extended_by.append(obj.name)
            extended_obj.extended_by.sort()



1684 1685 1686 1687 1688 1689 1690 1691 1692 1693 1694 1695 1696
notice = "/* Generated by esx_vi_generator.py */\n\n\n\n"

types_typedef.write(notice)
types_typeenum.write(notice)
types_typetostring.write(notice)
types_typefromstring.write(notice)
types_header.write(notice)
types_source.write(notice)
methods_header.write(notice)
methods_source.write(notice)
methods_macro.write(notice)
helpers_header.write(notice)
helpers_source.write(notice)
1697 1698


1699

1700
# output enums
1701
types_typedef.write(separator +
1702 1703
                    " * VI Enums\n" +
                    " */\n\n")
1704 1705 1706 1707 1708

names = enums_by_name.keys()
names.sort()

for name in names:
1709 1710 1711 1712 1713 1714
    types_typedef.write(enums_by_name[name].generate_typedef())
    types_typeenum.write(enums_by_name[name].generate_typeenum())
    types_typetostring.write(enums_by_name[name].generate_typetostring())
    types_typefromstring.write(enums_by_name[name].generate_typefromstring())
    types_header.write(enums_by_name[name].generate_header())
    types_source.write(enums_by_name[name].generate_source())
1715 1716 1717



1718 1719
# output objects
types_typedef.write("\n\n\n" +
1720
                    separator +
1721
                    " * VI Objects\n" +
1722 1723 1724 1725
                    " */\n\n")
types_typeenum.write("\n")
types_typetostring.write("\n")
types_typefromstring.write("\n")
1726

1727 1728
names = objects_by_name.keys()
names.sort()
1729

1730 1731 1732 1733 1734 1735 1736
for name in names:
    types_typedef.write(objects_by_name[name].generate_typedef())
    types_typeenum.write(objects_by_name[name].generate_typeenum())
    types_typetostring.write(objects_by_name[name].generate_typetostring())
    types_typefromstring.write(objects_by_name[name].generate_typefromstring())
    types_header.write(objects_by_name[name].generate_header())
    types_source.write(objects_by_name[name].generate_source())
1737 1738 1739



1740 1741
# output managed objects
types_typedef.write("\n\n\n" +
1742
                    separator +
1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760 1761
                    " * VI Managed Objects\n" +
                    " */\n\n")
types_typeenum.write("\n")
types_typetostring.write("\n")
types_typefromstring.write("\n")

names = managed_objects_by_name.keys()
names.sort()

for name in names:
    types_typedef.write(managed_objects_by_name[name].generate_typedef())
    types_typeenum.write(managed_objects_by_name[name].generate_typeenum())
    types_typetostring.write(managed_objects_by_name[name].generate_typetostring())
    types_typefromstring.write(managed_objects_by_name[name].generate_typefromstring())
    types_header.write(managed_objects_by_name[name].generate_header())
    types_source.write(managed_objects_by_name[name].generate_source())



1762 1763 1764
# output methods
names = methods_by_name.keys()
names.sort()
1765 1766

for name in names:
1767 1768
    methods_header.write(methods_by_name[name].generate_header())
    methods_source.write(methods_by_name[name].generate_source())
1769

1770 1771
names = list(autobind_names)
names.sort()
1772

1773
for name in names:
1774 1775 1776 1777 1778 1779
    string = aligned("#define ESX_VI__METHOD__PARAMETER__THIS__%s " % name, "\\\n", 78)
    string += "    ESX_VI__METHOD__PARAMETER__THIS_FROM_SERVICE(ManagedObjectReference,      \\\n"
    string += aligned("", "%s)\n\n\n\n" % name, 49)

    methods_macro.write(string)

1780 1781 1782 1783 1784 1785 1786 1787 1788


# output helpers
names = managed_objects_by_name.keys()
names.sort()

for name in names:
    helpers_header.write(managed_objects_by_name[name].generate_helper_header())
    helpers_source.write(managed_objects_by_name[name].generate_helper_source())