| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/webdata/web_database.h" | 5 #include "chrome/browser/webdata/web_database.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 #include <limits> | 8 #include <limits> |
| 9 #include <set> | 9 #include <set> |
| 10 #include <string> | 10 #include <string> |
| (...skipping 1531 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1542 return false; | 1542 return false; |
| 1543 } | 1543 } |
| 1544 | 1544 |
| 1545 s.BindInt(0, profile_id); | 1545 s.BindInt(0, profile_id); |
| 1546 if (s.Step()) | 1546 if (s.Step()) |
| 1547 *profile = AutoFillProfileFromStatement(s); | 1547 *profile = AutoFillProfileFromStatement(s); |
| 1548 | 1548 |
| 1549 return s.Succeeded(); | 1549 return s.Succeeded(); |
| 1550 } | 1550 } |
| 1551 | 1551 |
| 1552 static void BindCreditCardToStatement(const CreditCard& creditcard, | 1552 static void BindCreditCardToStatement(const CreditCard& credit_card, |
| 1553 sql::Statement* s) { | 1553 sql::Statement* s) { |
| 1554 s->BindString(0, UTF16ToUTF8(creditcard.Label())); | 1554 s->BindString(0, UTF16ToUTF8(credit_card.Label())); |
| 1555 s->BindInt(1, creditcard.unique_id()); | 1555 s->BindInt(1, credit_card.unique_id()); |
| 1556 | 1556 |
| 1557 string16 text = creditcard.GetFieldText(AutoFillType(CREDIT_CARD_NAME)); | 1557 string16 text = credit_card.GetFieldText(AutoFillType(CREDIT_CARD_NAME)); |
| 1558 s->BindString(2, UTF16ToUTF8(text)); | 1558 s->BindString(2, UTF16ToUTF8(text)); |
| 1559 text = creditcard.GetFieldText(AutoFillType(CREDIT_CARD_TYPE)); | 1559 text = credit_card.GetFieldText(AutoFillType(CREDIT_CARD_TYPE)); |
| 1560 s->BindString(3, UTF16ToUTF8(text)); | 1560 s->BindString(3, UTF16ToUTF8(text)); |
| 1561 text.clear(); // No unencrypted cc info. | 1561 text.clear(); // No unencrypted cc info. |
| 1562 s->BindString(4, UTF16ToUTF8(text)); | 1562 s->BindString(4, UTF16ToUTF8(text)); |
| 1563 text = creditcard.GetFieldText(AutoFillType(CREDIT_CARD_EXP_MONTH)); | 1563 text = credit_card.GetFieldText(AutoFillType(CREDIT_CARD_EXP_MONTH)); |
| 1564 s->BindString(5, UTF16ToUTF8(text)); | 1564 s->BindString(5, UTF16ToUTF8(text)); |
| 1565 text = creditcard.GetFieldText(AutoFillType(CREDIT_CARD_EXP_4_DIGIT_YEAR)); | 1565 text = credit_card.GetFieldText(AutoFillType(CREDIT_CARD_EXP_4_DIGIT_YEAR)); |
| 1566 s->BindString(6, UTF16ToUTF8(text)); | 1566 s->BindString(6, UTF16ToUTF8(text)); |
| 1567 text.clear(); | 1567 text.clear(); |
| 1568 s->BindString(7, UTF16ToUTF8(text)); | 1568 s->BindString(7, UTF16ToUTF8(text)); |
| 1569 s->BindString(8, UTF16ToUTF8(creditcard.billing_address())); | 1569 s->BindString(8, UTF16ToUTF8(credit_card.billing_address())); |
| 1570 s->BindString(9, UTF16ToUTF8(creditcard.shipping_address())); | 1570 s->BindString(9, UTF16ToUTF8(credit_card.shipping_address())); |
| 1571 text = creditcard.GetFieldText(AutoFillType(CREDIT_CARD_NUMBER)); | 1571 text = credit_card.GetFieldText(AutoFillType(CREDIT_CARD_NUMBER)); |
| 1572 std::string encrypted_data; | 1572 std::string encrypted_data; |
| 1573 Encryptor::EncryptString16(text, &encrypted_data); | 1573 Encryptor::EncryptString16(text, &encrypted_data); |
| 1574 s->BindBlob(10, encrypted_data.data(), | 1574 s->BindBlob(10, encrypted_data.data(), |
| 1575 static_cast<int>(encrypted_data.length())); | 1575 static_cast<int>(encrypted_data.length())); |
| 1576 text = creditcard.GetFieldText(AutoFillType(CREDIT_CARD_VERIFICATION_CODE)); | 1576 text = credit_card.GetFieldText(AutoFillType(CREDIT_CARD_VERIFICATION_CODE)); |
| 1577 Encryptor::EncryptString16(text, &encrypted_data); | 1577 Encryptor::EncryptString16(text, &encrypted_data); |
| 1578 s->BindBlob(11, encrypted_data.data(), | 1578 s->BindBlob(11, encrypted_data.data(), |
| 1579 static_cast<int>(encrypted_data.length())); | 1579 static_cast<int>(encrypted_data.length())); |
| 1580 } | 1580 } |
| 1581 | 1581 |
| 1582 bool WebDatabase::AddCreditCard(const CreditCard& creditcard) { | 1582 bool WebDatabase::AddCreditCard(const CreditCard& credit_card) { |
| 1583 sql::Statement s(db_.GetUniqueStatement( | 1583 sql::Statement s(db_.GetUniqueStatement( |
| 1584 "INSERT INTO credit_cards" | 1584 "INSERT INTO credit_cards" |
| 1585 "(label, unique_id, name_on_card, type, card_number," | 1585 "(label, unique_id, name_on_card, type, card_number," |
| 1586 " expiration_month, expiration_year, verification_code, billing_address," | 1586 " expiration_month, expiration_year, verification_code, billing_address," |
| 1587 " shipping_address, card_number_encrypted, verification_code_encrypted)" | 1587 " shipping_address, card_number_encrypted, verification_code_encrypted)" |
| 1588 "VALUES (?,?,?,?,?,?,?,?,?,?,?,?)")); | 1588 "VALUES (?,?,?,?,?,?,?,?,?,?,?,?)")); |
| 1589 if (!s) { | 1589 if (!s) { |
| 1590 NOTREACHED() << "Statement prepare failed"; | 1590 NOTREACHED() << "Statement prepare failed"; |
| 1591 return false; | 1591 return false; |
| 1592 } | 1592 } |
| 1593 | 1593 |
| 1594 BindCreditCardToStatement(creditcard, &s); | 1594 BindCreditCardToStatement(credit_card, &s); |
| 1595 | 1595 |
| 1596 if (!s.Run()) { | 1596 if (!s.Run()) { |
| 1597 NOTREACHED(); | 1597 NOTREACHED(); |
| 1598 return false; | 1598 return false; |
| 1599 } | 1599 } |
| 1600 | 1600 |
| 1601 DCHECK_GT(db_.GetLastChangeCount(), 0); | 1601 DCHECK_GT(db_.GetLastChangeCount(), 0); |
| 1602 return s.Succeeded(); | 1602 return s.Succeeded(); |
| 1603 } | 1603 } |
| 1604 | 1604 |
| 1605 static CreditCard* CreditCardFromStatement(const sql::Statement& s) { | 1605 static CreditCard* CreditCardFromStatement(const sql::Statement& s) { |
| 1606 CreditCard* creditcard = new CreditCard( | 1606 CreditCard* credit_card = new CreditCard( |
| 1607 UTF8ToUTF16(s.ColumnString(0)), s.ColumnInt(1)); | 1607 UTF8ToUTF16(s.ColumnString(0)), s.ColumnInt(1)); |
| 1608 creditcard->SetInfo(AutoFillType(CREDIT_CARD_NAME), | 1608 credit_card->SetInfo(AutoFillType(CREDIT_CARD_NAME), |
| 1609 UTF8ToUTF16(s.ColumnString(2))); | 1609 UTF8ToUTF16(s.ColumnString(2))); |
| 1610 creditcard->SetInfo(AutoFillType(CREDIT_CARD_TYPE), | 1610 credit_card->SetInfo(AutoFillType(CREDIT_CARD_TYPE), |
| 1611 UTF8ToUTF16(s.ColumnString(3))); | 1611 UTF8ToUTF16(s.ColumnString(3))); |
| 1612 string16 credit_card_number = UTF8ToUTF16(s.ColumnString(4)); | 1612 string16 credit_card_number = UTF8ToUTF16(s.ColumnString(4)); |
| 1613 // It could be non-empty prior to version 23. After that it encrypted in | 1613 // It could be non-empty prior to version 23. After that it encrypted in |
| 1614 // the column 10. | 1614 // the column 10. |
| 1615 if (credit_card_number.empty()) { | 1615 if (credit_card_number.empty()) { |
| 1616 int encrypted_cc_len = s.ColumnByteLength(10); | 1616 int encrypted_cc_len = s.ColumnByteLength(10); |
| 1617 std::string encrypted_cc; | 1617 std::string encrypted_cc; |
| 1618 if (encrypted_cc_len) { | 1618 if (encrypted_cc_len) { |
| 1619 encrypted_cc.resize(encrypted_cc_len); | 1619 encrypted_cc.resize(encrypted_cc_len); |
| 1620 memcpy(&encrypted_cc[0], s.ColumnBlob(10), encrypted_cc_len); | 1620 memcpy(&encrypted_cc[0], s.ColumnBlob(10), encrypted_cc_len); |
| 1621 Encryptor::DecryptString16(encrypted_cc, &credit_card_number); | 1621 Encryptor::DecryptString16(encrypted_cc, &credit_card_number); |
| 1622 } | 1622 } |
| 1623 } | 1623 } |
| 1624 creditcard->SetInfo(AutoFillType(CREDIT_CARD_NUMBER), credit_card_number); | 1624 credit_card->SetInfo(AutoFillType(CREDIT_CARD_NUMBER), credit_card_number); |
| 1625 creditcard->SetInfo(AutoFillType(CREDIT_CARD_EXP_MONTH), | 1625 credit_card->SetInfo(AutoFillType(CREDIT_CARD_EXP_MONTH), |
| 1626 UTF8ToUTF16(s.ColumnString(5))); | 1626 UTF8ToUTF16(s.ColumnString(5))); |
| 1627 creditcard->SetInfo(AutoFillType(CREDIT_CARD_EXP_4_DIGIT_YEAR), | 1627 credit_card->SetInfo(AutoFillType(CREDIT_CARD_EXP_4_DIGIT_YEAR), |
| 1628 UTF8ToUTF16(s.ColumnString(6))); | 1628 UTF8ToUTF16(s.ColumnString(6))); |
| 1629 | 1629 |
| 1630 string16 credit_card_verification_code = UTF8ToUTF16(s.ColumnString(7)); | 1630 string16 credit_card_verification_code = UTF8ToUTF16(s.ColumnString(7)); |
| 1631 // It could be non-empty prior to version 23. After that it encrypted in | 1631 // It could be non-empty prior to version 23. After that it encrypted in |
| 1632 // the column 11. | 1632 // the column 11. |
| 1633 if (credit_card_verification_code.empty()) { | 1633 if (credit_card_verification_code.empty()) { |
| 1634 int encrypted_cc_len = s.ColumnByteLength(11); | 1634 int encrypted_cc_len = s.ColumnByteLength(11); |
| 1635 std::string encrypted_cc; | 1635 std::string encrypted_cc; |
| 1636 if (encrypted_cc_len) { | 1636 if (encrypted_cc_len) { |
| 1637 encrypted_cc.resize(encrypted_cc_len); | 1637 encrypted_cc.resize(encrypted_cc_len); |
| 1638 memcpy(&encrypted_cc[0], s.ColumnBlob(11), encrypted_cc_len); | 1638 memcpy(&encrypted_cc[0], s.ColumnBlob(11), encrypted_cc_len); |
| 1639 Encryptor::DecryptString16(encrypted_cc, &credit_card_verification_code); | 1639 Encryptor::DecryptString16(encrypted_cc, &credit_card_verification_code); |
| 1640 } | 1640 } |
| 1641 } | 1641 } |
| 1642 creditcard->SetInfo(AutoFillType(CREDIT_CARD_VERIFICATION_CODE), | 1642 credit_card->SetInfo(AutoFillType(CREDIT_CARD_VERIFICATION_CODE), |
| 1643 credit_card_verification_code); | 1643 credit_card_verification_code); |
| 1644 creditcard->set_billing_address(UTF8ToUTF16(s.ColumnString(8))); | 1644 credit_card->set_billing_address(UTF8ToUTF16(s.ColumnString(8))); |
| 1645 creditcard->set_shipping_address(UTF8ToUTF16(s.ColumnString(9))); | 1645 credit_card->set_shipping_address(UTF8ToUTF16(s.ColumnString(9))); |
| 1646 // Column 10 is processed above. | 1646 // Column 10 is processed above. |
| 1647 // Column 11 is processed above. | 1647 // Column 11 is processed above. |
| 1648 | 1648 |
| 1649 return creditcard; | 1649 return credit_card; |
| 1650 } | 1650 } |
| 1651 | 1651 |
| 1652 bool WebDatabase::GetCreditCardForLabel(const string16& label, | 1652 bool WebDatabase::GetCreditCardForLabel(const string16& label, |
| 1653 CreditCard** creditcard) { | 1653 CreditCard** credit_card) { |
| 1654 DCHECK(creditcard); | 1654 DCHECK(credit_card); |
| 1655 sql::Statement s(db_.GetUniqueStatement( | 1655 sql::Statement s(db_.GetUniqueStatement( |
| 1656 "SELECT * FROM credit_cards " | 1656 "SELECT * FROM credit_cards " |
| 1657 "WHERE label = ?")); | 1657 "WHERE label = ?")); |
| 1658 if (!s) { | 1658 if (!s) { |
| 1659 NOTREACHED() << "Statement prepare failed"; | 1659 NOTREACHED() << "Statement prepare failed"; |
| 1660 return false; | 1660 return false; |
| 1661 } | 1661 } |
| 1662 | 1662 |
| 1663 s.BindString(0, UTF16ToUTF8(label)); | 1663 s.BindString(0, UTF16ToUTF8(label)); |
| 1664 if (!s.Step()) | 1664 if (!s.Step()) |
| 1665 return false; | 1665 return false; |
| 1666 | 1666 |
| 1667 *creditcard = CreditCardFromStatement(s); | 1667 *credit_card = CreditCardFromStatement(s); |
| 1668 | 1668 |
| 1669 return s.Succeeded(); | 1669 return s.Succeeded(); |
| 1670 } | 1670 } |
| 1671 | 1671 |
| 1672 bool WebDatabase::GetCreditCardForID(int card_id, CreditCard** card) { | 1672 bool WebDatabase::GetCreditCardForID(int credit_card_id, |
| 1673 CreditCard** credit_card) { |
| 1673 sql::Statement s(db_.GetUniqueStatement( | 1674 sql::Statement s(db_.GetUniqueStatement( |
| 1674 "SELECT * FROM credit_cards " | 1675 "SELECT * FROM credit_cards " |
| 1675 "WHERE unique_id = ?")); | 1676 "WHERE unique_id = ?")); |
| 1676 if (!s) { | 1677 if (!s) { |
| 1677 NOTREACHED() << "Statement prepare failed"; | 1678 NOTREACHED() << "Statement prepare failed"; |
| 1678 return false; | 1679 return false; |
| 1679 } | 1680 } |
| 1680 | 1681 |
| 1681 s.BindInt(0, card_id); | 1682 s.BindInt(0, credit_card_id); |
| 1682 if (!s.Step()) | 1683 if (!s.Step()) |
| 1683 return false; | 1684 return false; |
| 1684 | 1685 |
| 1685 *card = CreditCardFromStatement(s); | 1686 *credit_card = CreditCardFromStatement(s); |
| 1686 | 1687 |
| 1687 return s.Succeeded(); | 1688 return s.Succeeded(); |
| 1688 } | 1689 } |
| 1689 | 1690 |
| 1690 bool WebDatabase::GetCreditCards( | 1691 bool WebDatabase::GetCreditCards( |
| 1691 std::vector<CreditCard*>* creditcards) { | 1692 std::vector<CreditCard*>* credit_cards) { |
| 1692 DCHECK(creditcards); | 1693 DCHECK(credit_cards); |
| 1693 creditcards->clear(); | 1694 credit_cards->clear(); |
| 1694 | 1695 |
| 1695 sql::Statement s(db_.GetUniqueStatement("SELECT * FROM credit_cards")); | 1696 sql::Statement s(db_.GetUniqueStatement("SELECT * FROM credit_cards")); |
| 1696 if (!s) { | 1697 if (!s) { |
| 1697 NOTREACHED() << "Statement prepare failed"; | 1698 NOTREACHED() << "Statement prepare failed"; |
| 1698 return false; | 1699 return false; |
| 1699 } | 1700 } |
| 1700 | 1701 |
| 1701 while (s.Step()) | 1702 while (s.Step()) |
| 1702 creditcards->push_back(CreditCardFromStatement(s)); | 1703 credit_cards->push_back(CreditCardFromStatement(s)); |
| 1703 | 1704 |
| 1704 return s.Succeeded(); | 1705 return s.Succeeded(); |
| 1705 } | 1706 } |
| 1706 | 1707 |
| 1707 bool WebDatabase::UpdateCreditCard(const CreditCard& creditcard) { | 1708 bool WebDatabase::UpdateCreditCard(const CreditCard& credit_card) { |
| 1708 DCHECK(creditcard.unique_id()); | 1709 DCHECK(credit_card.unique_id()); |
| 1709 sql::Statement s(db_.GetUniqueStatement( | 1710 sql::Statement s(db_.GetUniqueStatement( |
| 1710 "UPDATE credit_cards " | 1711 "UPDATE credit_cards " |
| 1711 "SET label=?, unique_id=?, name_on_card=?, type=?, card_number=?, " | 1712 "SET label=?, unique_id=?, name_on_card=?, type=?, card_number=?, " |
| 1712 " expiration_month=?, expiration_year=?, verification_code=?, " | 1713 " expiration_month=?, expiration_year=?, verification_code=?, " |
| 1713 " billing_address=?, shipping_address=?, card_number_encrypted=?, " | 1714 " billing_address=?, shipping_address=?, card_number_encrypted=?, " |
| 1714 " verification_code_encrypted=? " | 1715 " verification_code_encrypted=? " |
| 1715 "WHERE unique_id=?")); | 1716 "WHERE unique_id=?")); |
| 1716 if (!s) { | 1717 if (!s) { |
| 1717 NOTREACHED() << "Statement prepare failed"; | 1718 NOTREACHED() << "Statement prepare failed"; |
| 1718 return false; | 1719 return false; |
| 1719 } | 1720 } |
| 1720 | 1721 |
| 1721 BindCreditCardToStatement(creditcard, &s); | 1722 BindCreditCardToStatement(credit_card, &s); |
| 1722 s.BindInt(12, creditcard.unique_id()); | 1723 s.BindInt(12, credit_card.unique_id()); |
| 1723 bool result = s.Run(); | 1724 bool result = s.Run(); |
| 1724 DCHECK_GT(db_.GetLastChangeCount(), 0); | 1725 DCHECK_GT(db_.GetLastChangeCount(), 0); |
| 1725 return result; | 1726 return result; |
| 1726 } | 1727 } |
| 1727 | 1728 |
| 1728 bool WebDatabase::RemoveCreditCard(int creditcard_id) { | 1729 bool WebDatabase::RemoveCreditCard(int credit_card_id) { |
| 1729 DCHECK_NE(0, creditcard_id); | 1730 DCHECK_NE(0, credit_card_id); |
| 1730 sql::Statement s(db_.GetUniqueStatement( | 1731 sql::Statement s(db_.GetUniqueStatement( |
| 1731 "DELETE FROM credit_cards WHERE unique_id = ?")); | 1732 "DELETE FROM credit_cards WHERE unique_id = ?")); |
| 1732 if (!s) { | 1733 if (!s) { |
| 1733 NOTREACHED() << "Statement prepare failed"; | 1734 NOTREACHED() << "Statement prepare failed"; |
| 1734 return false; | 1735 return false; |
| 1735 } | 1736 } |
| 1736 | 1737 |
| 1737 s.BindInt(0, creditcard_id); | 1738 s.BindInt(0, credit_card_id); |
| 1738 return s.Run(); | 1739 return s.Run(); |
| 1739 } | 1740 } |
| 1740 | 1741 |
| 1741 bool WebDatabase::AddToCountOfFormElement(int64 pair_id, | 1742 bool WebDatabase::AddToCountOfFormElement(int64 pair_id, |
| 1742 int delta, | 1743 int delta, |
| 1743 bool* was_removed) { | 1744 bool* was_removed) { |
| 1744 DCHECK(was_removed); | 1745 DCHECK(was_removed); |
| 1745 int count = 0; | 1746 int count = 0; |
| 1746 *was_removed = false; | 1747 *was_removed = false; |
| 1747 | 1748 |
| (...skipping 80 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 1828 | 1829 |
| 1829 // Add successive versions here. Each should set the version number and | 1830 // Add successive versions here. Each should set the version number and |
| 1830 // compatible version number as appropriate, then fall through to the next | 1831 // compatible version number as appropriate, then fall through to the next |
| 1831 // case. | 1832 // case. |
| 1832 | 1833 |
| 1833 case kCurrentVersionNumber: | 1834 case kCurrentVersionNumber: |
| 1834 // No migration needed. | 1835 // No migration needed. |
| 1835 return; | 1836 return; |
| 1836 } | 1837 } |
| 1837 } | 1838 } |
| OLD | NEW |